added prometheus exporter
This commit is contained in:
parent
be5ce63add
commit
7017df5fa3
5 changed files with 139 additions and 2 deletions
|
@ -3,6 +3,7 @@ import uuid
|
|||
from datetime import datetime
|
||||
from models import Order, Drink
|
||||
from websocket_manager import websocket_manager
|
||||
from prometheus_metrics import drink_metrics
|
||||
|
||||
|
||||
class OrderDatabase:
|
||||
|
@ -19,6 +20,9 @@ class OrderDatabase:
|
|||
)
|
||||
self.orders[order_id] = order
|
||||
|
||||
# Zeichne verkaufte Getränke für Prometheus-Statistiken auf
|
||||
drink_metrics.record_drinks(drinks)
|
||||
|
||||
# WebSocket-Broadcast für neue Bestellung
|
||||
await websocket_manager.broadcast_order_created(order.model_dump())
|
||||
|
||||
|
|
|
@ -1,13 +1,15 @@
|
|||
from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect, APIRouter
|
||||
from fastapi.responses import Response
|
||||
from typing import List
|
||||
from models import CreateOrderRequest, Order, DrinkType, MateType
|
||||
from database import db
|
||||
from websocket_manager import websocket_manager
|
||||
from prometheus_metrics import drink_metrics, CONTENT_TYPE_LATEST
|
||||
import json
|
||||
|
||||
app = FastAPI(
|
||||
title="Tschunk Order API",
|
||||
description="Eine RESTful API für Tschunk-Bestellungen mit WebSocket-Unterstützung",
|
||||
description="Eine RESTful API für Tschunk-Bestellungen mit WebSocket-Unterstützung und Prometheus-Metriken",
|
||||
version="1.0.0"
|
||||
)
|
||||
|
||||
|
@ -129,6 +131,28 @@ async def get_available_drinks():
|
|||
}
|
||||
|
||||
|
||||
@api_router.get("/stats/drinks")
|
||||
async def get_drink_statistics():
|
||||
"""
|
||||
Gibt die Statistiken der verkauften Getränke zurück.
|
||||
"""
|
||||
return drink_metrics.get_stats()
|
||||
|
||||
|
||||
@app.get("/metrics")
|
||||
async def get_prometheus_metrics():
|
||||
"""
|
||||
Prometheus-Metriken-Endpunkt.
|
||||
|
||||
Gibt die Metriken im Prometheus-Format zurück:
|
||||
- tschunk_drinks_sold_total: Anzahl verkaufter Getränke nach Typ und Mate-Sorte
|
||||
"""
|
||||
return Response(
|
||||
content=drink_metrics.get_metrics(),
|
||||
media_type=CONTENT_TYPE_LATEST
|
||||
)
|
||||
|
||||
|
||||
# API Router zur App hinzufügen
|
||||
app.include_router(api_router)
|
||||
|
||||
|
|
96
backend/prometheus_metrics.py
Normal file
96
backend/prometheus_metrics.py
Normal file
|
@ -0,0 +1,96 @@
|
|||
from prometheus_client import Counter, generate_latest, CONTENT_TYPE_LATEST, CollectorRegistry
|
||||
from typing import Dict, Tuple
|
||||
import json
|
||||
import os
|
||||
from models import DrinkType, MateType
|
||||
|
||||
|
||||
class DrinkSalesMetrics:
|
||||
def __init__(self, stats_file: str = "drink_stats.json"):
|
||||
self.stats_file = stats_file
|
||||
self.stats: Dict[Tuple[str, str], int] = self._load_stats()
|
||||
|
||||
# Erstelle eine eigene Registry ohne Standard-Metriken
|
||||
self.registry = CollectorRegistry()
|
||||
|
||||
# Prometheus Counter für verkaufte Getränke
|
||||
self.drinks_sold_counter = Counter(
|
||||
'tschunk_drinks_sold_total',
|
||||
'Total number of drinks sold',
|
||||
['drink_type', 'mate_type'],
|
||||
registry=self.registry
|
||||
)
|
||||
|
||||
# Initialisiere Counter mit gespeicherten Werten
|
||||
self._initialize_counters()
|
||||
|
||||
def _load_stats(self) -> Dict[Tuple[str, str], int]:
|
||||
"""Lädt die gespeicherten Statistiken aus der JSON-Datei."""
|
||||
if os.path.exists(self.stats_file):
|
||||
try:
|
||||
with open(self.stats_file, 'r', encoding='utf-8') as f:
|
||||
data = json.load(f)
|
||||
# Konvertiere String-Tupel zurück zu Tuple
|
||||
return {tuple(k.split('|')): v for k, v in data.items()}
|
||||
except (json.JSONDecodeError, KeyError, ValueError) as e:
|
||||
print(f"Fehler beim Laden der Statistiken: {e}")
|
||||
return {}
|
||||
return {}
|
||||
|
||||
def _save_stats(self):
|
||||
"""Speichert die Statistiken in eine JSON-Datei."""
|
||||
try:
|
||||
# Konvertiere Tuple zu String für JSON-Serialisierung
|
||||
data = {f"{k[0]}|{k[1]}": v for k, v in self.stats.items()}
|
||||
with open(self.stats_file, 'w', encoding='utf-8') as f:
|
||||
json.dump(data, f, indent=2, ensure_ascii=False)
|
||||
except Exception as e:
|
||||
print(f"Fehler beim Speichern der Statistiken: {e}")
|
||||
|
||||
def _initialize_counters(self):
|
||||
"""Initialisiert die Prometheus-Counter mit den gespeicherten Werten."""
|
||||
for (drink_type, mate_type), count in self.stats.items():
|
||||
# Setze den Counter auf den gespeicherten Wert
|
||||
# Da Counter nur inkrementiert werden können, müssen wir das manuell machen
|
||||
for _ in range(count):
|
||||
self.drinks_sold_counter.labels(
|
||||
drink_type=drink_type,
|
||||
mate_type=mate_type
|
||||
).inc()
|
||||
|
||||
def record_drinks(self, drinks):
|
||||
"""Zeichnet verkaufte Getränke auf und aktualisiert die Statistiken."""
|
||||
for drink in drinks:
|
||||
drink_type = drink.drink_type.value
|
||||
mate_type = drink.mate_type.value
|
||||
quantity = drink.quantity
|
||||
|
||||
# Aktualisiere lokale Statistiken
|
||||
key = (drink_type, mate_type)
|
||||
self.stats[key] = self.stats.get(key, 0) + quantity
|
||||
|
||||
# Aktualisiere Prometheus-Counter
|
||||
self.drinks_sold_counter.labels(
|
||||
drink_type=drink_type,
|
||||
mate_type=mate_type
|
||||
).inc(quantity)
|
||||
|
||||
# Speichere Statistiken persistent
|
||||
self._save_stats()
|
||||
|
||||
def get_stats(self) -> Dict[str, Dict[str, int]]:
|
||||
"""Gibt die aktuellen Statistiken zurück."""
|
||||
result = {}
|
||||
for (drink_type, mate_type), count in self.stats.items():
|
||||
if drink_type not in result:
|
||||
result[drink_type] = {}
|
||||
result[drink_type][mate_type] = count
|
||||
return result
|
||||
|
||||
def get_metrics(self):
|
||||
"""Gibt die Prometheus-Metriken zurück."""
|
||||
return generate_latest(self.registry)
|
||||
|
||||
|
||||
# Globale Instanz der Metriken
|
||||
drink_metrics = DrinkSalesMetrics()
|
|
@ -6,4 +6,5 @@ requests>=2.32.0
|
|||
websockets>=12.0
|
||||
pytest>=7.0.0
|
||||
pytest-asyncio>=0.23.0
|
||||
httpx>=0.25.0
|
||||
httpx>=0.25.0
|
||||
prometheus-client>=0.19.0
|
|
@ -17,6 +17,18 @@ server {
|
|||
add_header X-Content-Type-Options "nosniff" always;
|
||||
add_header Referrer-Policy "no-referrer-when-downgrade" always;
|
||||
|
||||
# Prometheus metrics endpoint - proxy to backend
|
||||
location /metrics {
|
||||
proxy_pass http://backend;
|
||||
proxy_set_header Host $host;
|
||||
proxy_set_header X-Real-IP $remote_addr;
|
||||
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||
proxy_set_header X-Forwarded-Proto $scheme;
|
||||
proxy_connect_timeout 60s;
|
||||
proxy_send_timeout 60s;
|
||||
proxy_read_timeout 60s;
|
||||
}
|
||||
|
||||
# API routes - proxy to backend
|
||||
location /api/ {
|
||||
limit_req zone=api burst=20 nodelay;
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue