🐛 Calculate query count using session events

bugbakery · Dec 15, 2023 · 0ab0f97 · 0ab0f97
1 parent 26e3e27
commit 0ab0f97
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/backend/transcribee_backend/db/__init__.py b/backend/transcribee_backend/db/__init__.py
@@ -49,14 +49,13 @@ def SessionContextManager(path: str):
 
 @contextmanager
 def query_counter(session: Session, path: Optional[str]):
-    engine = session.connection().engine
     count = 0
 
     def callback(*args, **kwargs):
         nonlocal count
         count += 1
 
-    event.listen(engine, "before_cursor_execute", callback)
+    event.listen(session, "do_orm_execute", callback)
     yield
-    event.remove(engine, "before_cursor_execute", callback)
+    event.remove(session, "do_orm_execute", callback)
     query_histogram.labels(path=path).observe(count)