Luciabrightcode
dab56e743b
- Scale docker-compose.yml (5 browser, 20 extractor replicas) - Add Prometheus and Grafana monitoring services - Implement persistent Redis TaskWorker in src/orchestrator/worker.py - Implement MetricsCollector in src/core/monitoring.py - Implement SessionRecoveryManager in src/core/recovery.py - Update README.md with production usage guide - Update root documentation (implementation_plan.md, walkthrough.md)
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from prometheus_client import Counter, Histogram, Gauge, start_http_server
|
|
import time
|
|
|
|
# Define global metrics
|
|
auth_attempts = Counter('auth_attempts_total', 'Authentication attempts', ['result'])
|
|
session_duration = Histogram('session_duration_seconds', 'Session lifespan')
|
|
challenge_rate = Gauge('challenge_rate', 'Rate of challenges encountered')
|
|
extraction_throughput = Counter('extraction_requests_total', 'API extractions', ['status'])
|
|
|
|
class MetricsCollector:
|
|
_server_started = False
|
|
|
|
@classmethod
|
|
def start_server(cls, port=8000):
|
|
if not cls._server_started:
|
|
start_http_server(port)
|
|
cls._server_started = True
|
|
|
|
@staticmethod
|
|
def record_auth_success():
|
|
auth_attempts.labels(result='success').inc()
|
|
|
|
@staticmethod
|
|
def record_auth_failure(reason: str):
|
|
auth_attempts.labels(result=reason).inc()
|
|
|
|
@staticmethod
|
|
def record_session_lifetime(duration: float):
|
|
session_duration.observe(duration)
|
|
|
|
@staticmethod
|
|
def update_challenge_rate(rate: float):
|
|
challenge_rate.set(rate)
|
|
|
|
@staticmethod
|
|
def record_extraction(status: str):
|
|
extraction_throughput.labels(status=status).inc()
|