Update system monitor when polling Prometheus metrics (#8745)

dask · Jul 3, 2024 · 9d36228 · 9d36228
1 parent d36fc3c
commit 9d36228
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/distributed/http/scheduler/prometheus/core.py b/distributed/http/scheduler/prometheus/core.py
@@ -22,6 +22,8 @@ def __init__(self, server: Scheduler):
         self.subsystem = "scheduler"
 
     def collect(self) -> Iterator[GaugeMetricFamily | CounterMetricFamily]:
+        self.server.monitor.update()
+
         yield GaugeMetricFamily(
             self.build_name("clients"),
             "Number of clients connected",

diff --git a/distributed/http/worker/prometheus/core.py b/distributed/http/worker/prometheus/core.py
@@ -32,6 +32,7 @@ def __init__(self, server: Worker):
             )
 
     def collect(self) -> Iterator[Metric]:
+        self.server.monitor.update()
         ws = self.server.state
 
         tasks = GaugeMetricFamily(