Expose karton metrics via the backend API (#109)

* Expose karton metrics via the backend API * Update the docs Co-authored-by: msm <msm@cert.pl>
CERT-Polska · Mar 17, 2021 · ab6b99d · ab6b99d
1 parent d94ce4e
commit ab6b99d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/karton/core/backend.py b/karton/core/backend.py
@@ -422,6 +422,14 @@ def increment_metrics(self, metric: KartonMetrics, identity: str) -> None:
         """
         self.redis.hincrby(metric.value, identity, 1)
 
+    def get_metrics(self, metric: KartonMetrics) -> Dict[str, int]:
+        """
+        Get a {karton-identity: current-number-of-tasks} mapping for a given metric.
+
+        :param metric: Operation metric type
+        """
+        return {k: int(v) for k, v in self.redis.hgetall(metric.value).items()}
+
     def upload_object(
         self,
         bucket: str,