Profile some of the CLE report's methods

corehq/apps/reports/standard/cases/basic.py

@@ -1,3 +1,5 @@
+import contextlib
+
 from django.utils.translation import gettext as _
 from django.utils.translation import gettext_lazy
 
@@ -13,7 +15,7 @@
 from corehq.apps.reports.generic import ElasticProjectInspectionReport
 from corehq.apps.reports.standard import (
     ProjectReport,
-    ProjectReportParametersMixin,
+    ProjectReportParametersMixin, ESQueryProfilerMixin,
 )
 from corehq.apps.reports.standard.cases.filters import CaseSearchFilter
 from corehq.apps.reports.standard.cases.utils import (
@@ -28,7 +30,7 @@
 from .data_sources import CaseDisplayES
 
 
-class CaseListMixin(ElasticProjectInspectionReport, ProjectReportParametersMixin):
+class CaseListMixin(ESQueryProfilerMixin, ElasticProjectInspectionReport, ProjectReportParametersMixin):
     fields = [
         'corehq.apps.reports.filters.case_list.CaseListFilter',
         'corehq.apps.reports.filters.select.CaseTypeFilter',
@@ -107,7 +109,8 @@ def _build_query(self):
     @memoized
     def es_results(self):
         try:
-            return self._build_query().run().raw
+            with self.profiler.timing_context("ES query") if self.profiler_enabled else contextlib.nullcontext():
+                return self._build_query().run().raw
         except ESError as e:
             original_exception = e.args[0]
             if isinstance(original_exception, TransportError):
@@ -234,3 +237,10 @@ def rows(self):
                 display.modified_on,
                 display.closed_display
             ]
+
+    @property
+    def json_response(self):
+        with self.profiler.timing_context if self.profiler_enabled else contextlib.nullcontext():
+            response = super().json_response
+
+        return response

corehq/apps/reports/standard/cases/case_list_explorer.py

@@ -62,6 +62,9 @@ class CaseListExplorer(CaseListReport, XpathCaseSearchFilterMixin):
     documentation_link = DOCS_LINK_CASE_LIST_EXPLORER
     use_bootstrap5 = False
 
+    profiler_enabled = True
+    profiler_name = "Case List Explorer"
+
     exportable = True
     exportable_all = True
     emailable = True
@@ -93,30 +96,36 @@ def es_results(self):
     def _build_query(self, sort=True):
         query = super(CaseListExplorer, self)._build_query()
         query = self._populate_sort(query, sort)
-        query = self.apply_xpath_case_search_filter(query)
+        query = self._apply_xpath_case_search_filter(query)
         return query
 
     def _populate_sort(self, query, sort):
-        if not sort:
-            # Don't sort on export
-            query = query.set_sorting_block(['_doc'])
-            return query
-
-        num_sort_columns = int(self.request.GET.get('iSortingCols', 0))
-        for col_num in range(num_sort_columns):
-            descending = self.request.GET['sSortDir_{}'.format(col_num)] == 'desc'
-            column_id = int(self.request.GET["iSortCol_{}".format(col_num)])
-            column = self.headers.header[column_id]
-            try:
-                meta_property = INDEXED_METADATA_BY_KEY[column.prop_name]
-                if meta_property.key == '@case_id':
-                    # This condition is added because ES 5 does not allow sorting on _id.
-                    #  When we will have case_id in root of the document, this should be removed.
-                    query.es_query['sort'] = get_case_id_sort_block(descending)
-                    return query
-                query = query.sort(meta_property.es_field_name, desc=descending)
-            except KeyError:
-                query = query.sort_by_case_property(column.prop_name, desc=descending)
+        with self.profiler.timing_context("Populate sort"):
+            if not sort:
+                # Don't sort on export
+                query = query.set_sorting_block(['_doc'])
+                return query
+
+            num_sort_columns = int(self.request.GET.get('iSortingCols', 0))
+            for col_num in range(num_sort_columns):
+                descending = self.request.GET['sSortDir_{}'.format(col_num)] == 'desc'
+                column_id = int(self.request.GET["iSortCol_{}".format(col_num)])
+                column = self.headers.header[column_id]
+                try:
+                    meta_property = INDEXED_METADATA_BY_KEY[column.prop_name]
+                    if meta_property.key == '@case_id':
+                        # This condition is added because ES 5 does not allow sorting on _id.
+                        #  When we will have case_id in root of the document, this should be removed.
+                        query.es_query['sort'] = get_case_id_sort_block(descending)
+                        return query
+                    query = query.sort(meta_property.es_field_name, desc=descending)
+                except KeyError:
+                    query = query.sort_by_case_property(column.prop_name, desc=descending)
+        return query
+
+    def _apply_xpath_case_search_filter(self, query):
+        with self.profiler.timing_context("Apply Xpath case filter"):
+            query = self.apply_xpath_case_search_filter(query)
         return query
 
     @property
@@ -173,8 +182,11 @@ def headers(self):
     @property
     def rows(self):
         self.track_search()
-        data = (wrap_case_search_hit(row) for row in self.es_results['hits'].get('hits', []))
-        return self._get_rows(data)
+
+        with self.profiler.timing_context("Retrieving rows"):
+            data = (wrap_case_search_hit(row) for row in self.es_results['hits'].get('hits', []))
+            with self.profiler.timing_context("Parsing rows"):
+                return self._get_rows(data)
 
     def track_search(self):
         track_workflow(