This is an automated email from the ASF dual-hosted git repository.
jscheffl pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 2a2a487eb83 Removed state metric and added maintenance and connected
metric (#48797)
2a2a487eb83 is described below
commit 2a2a487eb839e2a6885ebc3336ff76b46b6114bc
Author: AutomationDev85 <[email protected]>
AuthorDate: Mon Apr 7 15:51:00 2025 +0200
Removed state metric and added maintenance and connected metric (#48797)
Co-authored-by: AutomationDev85 <AutomationDev85>
---
.../airflow/providers/edge/models/edge_worker.py | 22 +++++++++++++++-------
1 file changed, 15 insertions(+), 7 deletions(-)
diff --git a/providers/edge/src/airflow/providers/edge/models/edge_worker.py
b/providers/edge/src/airflow/providers/edge/models/edge_worker.py
index 982eb07db4a..f8b29a78630 100644
--- a/providers/edge/src/airflow/providers/edge/models/edge_worker.py
+++ b/providers/edge/src/airflow/providers/edge/models/edge_worker.py
@@ -148,15 +148,23 @@ def set_metrics(
) -> None:
"""Set metric of edge worker."""
queues = queues if queues else []
- connected = state not in (EdgeWorkerState.UNKNOWN, EdgeWorkerState.OFFLINE)
-
- Stats.gauge(f"edge_worker.state.{worker_name}", int(connected))
- Stats.gauge(
- "edge_worker.state",
- int(connected),
- tags={"name": worker_name, "state": state},
+ connected = state not in (
+ EdgeWorkerState.UNKNOWN,
+ EdgeWorkerState.OFFLINE,
+ EdgeWorkerState.OFFLINE_MAINTENANCE,
+ )
+ maintenance = state in (
+ EdgeWorkerState.MAINTENANCE_MODE,
+ EdgeWorkerState.MAINTENANCE_EXIT,
+ EdgeWorkerState.OFFLINE_MAINTENANCE,
)
+ Stats.gauge(f"edge_worker.connected.{worker_name}", int(connected))
+ Stats.gauge("edge_worker.connected", int(connected), tags={"worker_name":
worker_name})
+
+ Stats.gauge(f"edge_worker.maintenance.{worker_name}", int(maintenance))
+ Stats.gauge("edge_worker.maintenance", int(maintenance),
tags={"worker_name": worker_name})
+
Stats.gauge(f"edge_worker.jobs_active.{worker_name}", jobs_active)
Stats.gauge("edge_worker.jobs_active", jobs_active, tags={"worker_name":
worker_name})
