bundle analysis: fix slow trend query (#812)

codecov · Sep 10, 2024 · e8d8cd0 · e8d8cd0
1 parent 4055e66
commit e8d8cd0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/graphql_api/actions/measurements.py b/graphql_api/actions/measurements.py
@@ -49,15 +49,17 @@ def measurements_by_ids(
 
 
 def measurements_last_uploaded_before_start_date(
+    owner_id: int,
     repo_id: int,
     measurable_name: str,
-    measurable_ids: List[int],
+    measurable_id: int,
     start_date: datetime,
     branch: Optional[str] = None,
 ) -> QuerySet:
     queryset = Measurement.objects.filter(
+        owner_id=owner_id,
         repo_id=repo_id,
-        measurable_id__in=measurable_ids,
+        measurable_id=measurable_id,
         name=measurable_name,
         timestamp__lt=start_date,
     )

diff --git a/services/bundle_analysis.py b/services/bundle_analysis.py
@@ -412,9 +412,10 @@ def _compute_measurements(
         for measurable_id, measurements in all_measurements.items():
             if self.after is not None and measurements[0]["timestamp_bin"] > self.after:
                 carryover_measurement = measurements_last_uploaded_before_start_date(
+                    owner_id=self.repository.author.ownerid,
                     repo_id=self.repository.repoid,
                     measurable_name=measurable_name,
-                    measurable_ids=[measurable_id],
+                    measurable_id=measurable_id,
                     start_date=self.after,
                     branch=self.branch,
                 )