Add opentelmetry span for database query

DiamondLightSource · Apr 18, 2024 · ee054cb · ee054cb
1 parent 3ad4ea7
commit ee054cb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/graph_energy_scan/graphql.py b/src/graph_energy_scan/graphql.py
@@ -81,6 +81,6 @@ async def energy_scans(self) -> list[EnergyScan]:
             stmt = select(models.EnergyScan).where(
                 models.EnergyScan.sessionId == self.id
             )
-            return [
-                EnergyScan.from_model(model) for model in await session.scalars(stmt)
-            ]
+            with tracer.start_as_current_span("query_session_energy_scans"):
+                energy_scans = await session.scalars(stmt)
+            return [EnergyScan.from_model(model) for model in energy_scans]