Spaces:
Sleeping
Sleeping
File size: 2,459 Bytes
34e27fb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | from prometheus_client import Counter, Histogram, Gauge
import time
# Event publishing metrics
event_published_counter = Counter(
'task_events_published_total',
'Total number of task events published',
['event_type']
)
event_publish_duration = Histogram(
'task_event_publish_duration_seconds',
'Time spent publishing task events',
['event_type']
)
event_publish_errors = Counter(
'task_event_publish_errors_total',
'Total number of task event publishing errors',
['event_type']
)
# Circuit breaker metrics
circuit_breaker_state = Gauge(
'circuit_breaker_state',
'Current state of the circuit breaker (0=closed, 1=open, 2=half-open)',
['breaker_name']
)
circuit_breaker_failures = Counter(
'circuit_breaker_failures_total',
'Total number of circuit breaker failures',
['breaker_name']
)
# Rate limiting metrics (for T048)
rate_limiter_requests = Counter(
'rate_limiter_requests_total',
'Total number of requests to rate limiter',
['endpoint']
)
rate_limiter_rejections = Counter(
'rate_limiter_rejections_total',
'Total number of requests rejected by rate limiter',
['endpoint']
)
def increment_event_published(event_type: str):
"""Increment the counter for published events."""
event_published_counter.labels(event_type=event_type).inc()
def observe_event_publish_duration(event_type: str, duration: float):
"""Record the duration of an event publishing operation."""
event_publish_duration.labels(event_type=event_type).observe(duration)
def increment_event_publish_error(event_type: str):
"""Increment the counter for event publishing errors."""
event_publish_errors.labels(event_type=event_type).inc()
def set_circuit_breaker_state(breaker_name: str, state_value: int):
"""Set the gauge value for circuit breaker state."""
circuit_breaker_state.labels(breaker_name=breaker_name).set(state_value)
def increment_circuit_breaker_failure(breaker_name: str):
"""Increment the counter for circuit breaker failures."""
circuit_breaker_failures.labels(breaker_name=breaker_name).inc()
def increment_rate_limiter_request(endpoint: str):
"""Increment the counter for rate limiter requests."""
rate_limiter_requests.labels(endpoint=endpoint).inc()
def increment_rate_limiter_rejection(endpoint: str):
"""Increment the counter for rate limiter rejections."""
rate_limiter_rejections.labels(endpoint=endpoint).inc() |