Add prometheus client counter

comworkio · Mar 10, 2024 · 527d9ff · 527d9ff
1 parent 48f8979
commit 527d9ff
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/src/utils/counter.py b/src/utils/counter.py
@@ -1,11 +1,17 @@
+from prometheus_client import Counter
+
 from utils.otel import get_otel_meter
 
 def create_counter(name, description):
-    return get_otel_meter().create_counter(
-        name = name,
-        description = description,
-        unit = "1"
-    )
+    return {
+        'otel': get_otel_meter().create_counter(
+                    name = name,
+                    description = description,
+                    unit = "1"
+                ),
+        'prom': Counter(name, description)
+    }
 
 def increment_counter(counter, tid):
-    counter.add(1, {"tid": tid})
+    counter['otel'].add(1, {"tid": tid})
+    counter['prom'].inc()