Print metric and move prompusher context to narrow scope

rucio · Aug 6, 2024 · 9483cf3 · 9483cf3
1 parent c5a6bb7
commit 9483cf3
Showing 1 changed file with 14 additions and 14 deletions.
diff --git a/common/check_expired_dids b/common/check_expired_dids
@@ -34,24 +34,24 @@ OK, WARNING, CRITICAL, UNKNOWN = 0, 1, 2, 3
 if __name__ == "__main__":
     try:
         session = get_session()
-        with PrometheusPusher() as manager:
-            statement = select(
-                func.count()
-            ).select_from(
-                models.DataIdentifier
-            ).where(
-                and_(
-                    models.DataIdentifier.expired_at != null(), 
-                    models.DataIdentifier.expired_at < datetime.utcnow()
-                )
+        statement = select(
+            func.count()
+        ).select_from(
+            models.DataIdentifier
+        ).where(
+            and_(
+                models.DataIdentifier.expired_at != null(), 
+                models.DataIdentifier.expired_at < datetime.utcnow()
             )
-            expired_dids = session.execute(statement).scalar_one()
-
+        )
+        expired_dids = session.execute(statement).scalar_one()
+        print(expired_dids)
+
+        with PrometheusPusher() as manager:
             (manager.gauge(
-                'expired_dids.total',
+                "expired_dids.total",
                 documentation="All expired dids")
             .set(expired_dids))
-
     except Exception:
         print(traceback.format_exc())
         sys.exit(UNKNOWN)