Specify sum of succeeded task runtime seconds (#18)

Author: draftcode

celerymon/event_watcher.py

@@ -19,6 +19,7 @@ class EventWatcher:
     upper_bounds: list[float]
     task_names: set[str]
     succeeded_task_runtime_sec: list[dict[str, int]]
+    succeeded_task_runtime_sec_sum: dict[str, float]
 
     @classmethod
     def create_started(
@@ -61,6 +62,7 @@ def __init__(
         self.last_received_timestamp_per_task_event = dict()
         self.num_events_per_task_count = defaultdict(int)
         self.succeeded_task_runtime_sec = []
+        self.succeeded_task_runtime_sec_sum = defaultdict(float)
         for _ in range(0, len(self.upper_bounds)):
             self.succeeded_task_runtime_sec.append(defaultdict(int))
 
@@ -84,6 +86,7 @@ def on_event(self, event: dict[str, Any]):
             # Not documented, but looking into the Celery codebase, the runtime
             # looks like seconds.
             runtime_sec = event["runtime"]
+            self.succeeded_task_runtime_sec_sum[task_name] += runtime_sec
             for i, bound in enumerate(self.upper_bounds):
                 if runtime_sec <= bound:
                     self.succeeded_task_runtime_sec[i][task_name] += 1

celerymon/metrics.py

@@ -155,7 +155,7 @@ def event_metrics(watcher: EventWatcher) -> list[prometheus_client.Metric]:
             success_task_runtime_seconds_metric.add_metric(
                 labels=[task_name],
                 buckets=buckets,
-                sum_value=None,
+                sum_value=watcher.succeeded_task_runtime_sec_sum[task_name],
                 timestamp=watcher.last_received_timestamp.timestamp(),
             )
     return [