Expose comm_nbytes in get_metrics and prometheus

dask · Aug 31, 2022 · 8ae02e1 · 8ae02e1
1 parent 3e3cf21
commit 8ae02e1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/distributed/http/worker/prometheus/core.py b/distributed/http/worker/prometheus/core.py
@@ -53,6 +53,12 @@ def collect(self):
             value=self.server.latency,
         )
 
+        yield GaugeMetricFamily(
+            self.build_name("comm_reserved_bytes"),
+            "Number of bytes currently reserved for incoming/outgoing data transfers.",
+            value=self.server.state.comm_nbytes,
+        )
+
         # all metrics using digests require crick to be installed
         # the following metrics will export NaN, if the corresponding digests are None
         if self.crick_available:

diff --git a/distributed/worker.py b/distributed/worker.py
@@ -983,6 +983,7 @@ async def get_metrics(self) -> dict:
                 "memory": spilled_memory,
                 "disk": spilled_disk,
             },
+            comm_reserved_bytes=self.state.comm_nbytes,
             event_loop_interval=self._tick_interval_observed,
         )
         out.update(self.monitor.recent())