avoid making extensive queries by overriding queryset cache

Author: pawelusermoc

elasticapm/contrib/django/client.py

@@ -34,6 +34,7 @@
 import django
 from django.conf import settings as django_settings
 from django.db import DatabaseError
+from django.db.models.query import QuerySet
 from django.http import HttpRequest
 
 try:
@@ -195,6 +196,14 @@ def capture(self, event_type, request=None, **kwargs):
 
         return result
 
+    @staticmethod
+    def _disable_querysets_evaluation(var):
+        """Overriding queryset result cache to avoid making any queries"""
+        if isinstance(var, QuerySet):
+            var._result_cache = []
+        return var
+
+
     def _get_stack_info_for_trace(
         self,
         frames,
@@ -205,6 +214,13 @@ def _get_stack_info_for_trace(
     ):
         """If the stacktrace originates within the elasticapm module, it will skip
         frames until some other module comes up."""
+
+        # Overriding processor func for django to avoid making queries to db through QuerySet objects
+        if locals_processor_func is not None:
+            locals_processor_func = lambda v: self._disable_querysets_evaluation(locals_processor_func(v))
+        else:
+            locals_processor_func = lambda v: self._disable_querysets_evaluation(v)
+
         return list(
             iterate_with_template_sources(
                 frames,