From 7017df5fa35506b64bef1ab70187447a0d6efdc1 Mon Sep 17 00:00:00 2001 From: Jan Felix Wiebe Date: Fri, 11 Jul 2025 14:56:30 +0200 Subject: [PATCH] added prometheus exporter --- backend/database.py | 4 ++ backend/main.py | 26 +++++++++- backend/prometheus_metrics.py | 96 +++++++++++++++++++++++++++++++++++ backend/requirements.txt | 3 +- nginx/conf.d/default.conf | 12 +++++ 5 files changed, 139 insertions(+), 2 deletions(-) create mode 100644 backend/prometheus_metrics.py diff --git a/backend/database.py b/backend/database.py index 4c7ec8c..0af6faf 100644 --- a/backend/database.py +++ b/backend/database.py @@ -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()) diff --git a/backend/main.py b/backend/main.py index 08d01a2..0c9177f 100644 --- a/backend/main.py +++ b/backend/main.py @@ -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) diff --git a/backend/prometheus_metrics.py b/backend/prometheus_metrics.py new file mode 100644 index 0000000..ac184e0 --- /dev/null +++ b/backend/prometheus_metrics.py @@ -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() \ No newline at end of file diff --git a/backend/requirements.txt b/backend/requirements.txt index ae745c2..09665bc 100644 --- a/backend/requirements.txt +++ b/backend/requirements.txt @@ -6,4 +6,5 @@ requests>=2.32.0 websockets>=12.0 pytest>=7.0.0 pytest-asyncio>=0.23.0 -httpx>=0.25.0 \ No newline at end of file +httpx>=0.25.0 +prometheus-client>=0.19.0 \ No newline at end of file diff --git a/nginx/conf.d/default.conf b/nginx/conf.d/default.conf index 139c9af..ea7e929 100644 --- a/nginx/conf.d/default.conf +++ b/nginx/conf.d/default.conf @@ -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;