optimzed the prometheus endpooint

Author: codeperfectplus

src/routes/prometheus.py

@@ -1,13 +1,12 @@
-from flask import blueprints
-from werkzeug.middleware.dispatcher import DispatcherMiddleware
-from prometheus_client import make_wsgi_app, Counter, Gauge
+from flask import Blueprint, Response
+from prometheus_client import Counter, Gauge, generate_latest
 import threading
 import time
 from src.config import app
 from src.utils import _get_system_info
 
 # Define the Prometheus Blueprint
-prometheus_bp = blueprints.Blueprint('prometheus', __name__)
+prometheus_bp = Blueprint('prometheus', __name__)
 
 # Initialize Prometheus metrics
 cpu_usage_metric = Gauge('cpu_usage_percentage', 'Current CPU usage percentage')
@@ -23,27 +22,30 @@ def collect_metrics():
     Runs in a separate thread and updates metrics every 5 seconds.
     """
     while True:
-        # Gather system information
-        system_info = _get_system_info()
-
-        # Update Prometheus metrics
-        cpu_usage_metric.set(system_info['cpu_percent'])
-        memory_usage_metric.set(system_info['memory_percent'])
-        disk_usage_metric.set(system_info['disk_percent'])
-        network_sent_metric.set(system_info['network_sent'])
-        network_recv_metric.set(system_info['network_received'])
-
-        # Increment HTTP request counter
-        request_count.inc()
+        try:
+            # Gather system information
+            system_info = _get_system_info()
+
+            # Update Prometheus metrics
+            cpu_usage_metric.set(system_info['cpu_percent'])
+            memory_usage_metric.set(system_info['memory_percent'])
+            disk_usage_metric.set(system_info['disk_percent'])
+            network_sent_metric.set(system_info['network_sent'])
+            network_recv_metric.set(system_info['network_received'])
+
+            # Increment HTTP request counter
+            request_count.inc()
+        except Exception as e:
+            print(f"Error collecting metrics: {e}")
 
         # Sleep for 5 seconds before the next collection
-        time.sleep(5)
+        time.sleep(10)
 
 # Start the metrics collection in a background thread
 metrics_thread = threading.Thread(target=collect_metrics, daemon=True)
 metrics_thread.start()
 
-# Expose the /metrics endpoint for Prometheus to scrape metrics
-app.wsgi_app = DispatcherMiddleware(app.wsgi_app, {
-    '/metrics': make_wsgi_app()  # Serve Prometheus metrics at /metrics
-})
+# Define a route to serve Prometheus metrics
+@app.route('/metrics')
+def metrics():
+    return Response(generate_latest(), mimetype='text/plain')