wehub-resource-sync
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
from ray._private.telemetry.open_telemetry_metric_recorder import (
|
|
OpenTelemetryMetricRecorder,
|
|
)
|
|
from ray.dashboard.modules.aggregator.constants import (
|
|
AGGREGATOR_AGENT_METRIC_PREFIX,
|
|
)
|
|
|
|
# OpenTelemetry metrics setup (registered once at import time)
|
|
metric_recorder = OpenTelemetryMetricRecorder()
|
|
|
|
# Counter metrics
|
|
published_counter_name = f"{AGGREGATOR_AGENT_METRIC_PREFIX}_published_events"
|
|
metric_recorder.register_counter_metric(
|
|
published_counter_name,
|
|
"Total number of events successfully published to the destination.",
|
|
)
|
|
|
|
filtered_counter_name = f"{AGGREGATOR_AGENT_METRIC_PREFIX}_filtered_events"
|
|
metric_recorder.register_counter_metric(
|
|
filtered_counter_name,
|
|
"Total number of events filtered out before publishing to the destination.",
|
|
)
|
|
|
|
failed_counter_name = f"{AGGREGATOR_AGENT_METRIC_PREFIX}_publish_failures"
|
|
metric_recorder.register_counter_metric(
|
|
failed_counter_name,
|
|
"Total number of events that failed to publish after retries.",
|
|
)
|
|
|
|
# Histogram metric
|
|
publish_latency_hist_name = f"{AGGREGATOR_AGENT_METRIC_PREFIX}_publish_latency_seconds"
|
|
metric_recorder.register_histogram_metric(
|
|
publish_latency_hist_name,
|
|
"Duration of publish calls in seconds.",
|
|
[0.001, 0.005, 0.01, 0.025, 0.05, 0.1, 0.25, 0.5, 1, 2, 5],
|
|
)
|
|
|
|
# Gauge metrics
|
|
consecutive_failures_gauge_name = (
|
|
f"{AGGREGATOR_AGENT_METRIC_PREFIX}_consecutive_failures_since_last_success"
|
|
)
|
|
metric_recorder.register_gauge_metric(
|
|
consecutive_failures_gauge_name,
|
|
"Number of consecutive failed publish attempts since the last success.",
|
|
)
|
|
|
|
time_since_last_success_gauge_name = (
|
|
f"{AGGREGATOR_AGENT_METRIC_PREFIX}_time_since_last_success_seconds"
|
|
)
|
|
metric_recorder.register_gauge_metric(
|
|
time_since_last_success_gauge_name,
|
|
"Seconds since the last successful publish to the destination.",
|
|
)
|