From 15134b37af2e5f3c6aad7fc0d845c571e8b321bf Mon Sep 17 00:00:00 2001
From: Quentin Gliech <quenting@element.io>
Date: Wed, 20 Sep 2023 20:28:40 +0200
Subject: [PATCH] Add missing tracing to app session storage

---
 crates/storage-pg/src/app_session.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/crates/storage-pg/src/app_session.rs b/crates/storage-pg/src/app_session.rs
index ae10495e..422fdf2e 100644
--- a/crates/storage-pg/src/app_session.rs
+++ b/crates/storage-pg/src/app_session.rs
@@ -201,6 +201,14 @@ impl<'c> AppSessionRepository for PgAppSessionRepository<'c> {
     type Error = DatabaseError;
 
     #[allow(clippy::too_many_lines)]
+    #[tracing::instrument(
+        name = "db.app_session.list",
+        fields(
+            db.statement,
+        ),
+        skip_all,
+        err,
+    )]
     async fn list(
         &mut self,
         filter: AppSessionFilter<'_>,
@@ -346,6 +354,14 @@ impl<'c> AppSessionRepository for PgAppSessionRepository<'c> {
         Ok(page)
     }
 
+    #[tracing::instrument(
+        name = "db.app_session.count",
+        fields(
+            db.statement,
+        ),
+        skip_all,
+        err,
+    )]
     async fn count(&mut self, filter: AppSessionFilter<'_>) -> Result<usize, Self::Error> {
         let mut oauth2_session_select = Query::select()
             .expr(Expr::cust("1"))