metrics

App/Monitor/monitorRoutes.py

@@ -0,0 +1,39 @@
+from fastapi import APIRouter, Request
+import psutil
+import json
+from App import bot
+
+monitor_router = APIRouter(tags=["monitor"])
+
+
+@monitor_router.get("/metrics")
+def get_metrics():
+    # Get CPU usage
+    cpu_usage = psutil.cpu_percent()
+
+    # Get memory usage
+    memory_usage = psutil.virtual_memory().percent
+
+    # Get disk usage
+    disk_usage = psutil.disk_usage("/").percent
+
+    # Get network statistics
+    network_stats = psutil.net_io_counters()
+
+    # Create a dictionary with the metrics
+    metrics = {
+        "cpu_usage": cpu_usage,
+        "memory_usage": memory_usage,
+        "disk_usage": disk_usage,
+        "network_stats": {
+            "bytes_sent": network_stats.bytes_sent,
+            "bytes_received": network_stats.bytes_recv,
+            "packets_sent": network_stats.packets_sent,
+            "packets_received": network_stats.packets_recv,
+        },
+    }
+
+    # Convert the dictionary to a nicely formatted JSON string
+    json_string = json.dumps(metrics, indent=4)
+
+    return json_string

App/app.py

@@ -4,6 +4,7 @@ from .Users.UserRoutes import user_router
 from .modelInit import models, database
 from .Transcription.TranscriptionRoutes import transcription_router
 from .Streaming.StreamingRoutes import streaming_router
+from .Monitor.monitorRoutes import monitor_router
 from fastapi.middleware.cors import CORSMiddleware
 import logging
 
@@ -54,3 +55,4 @@ async def landing_page():
 app.include_router(user_router)
 app.include_router(transcription_router)
 app.include_router(streaming_router)
+app.include_router(monitor_router)

requirements.txt

@@ -18,6 +18,7 @@ pydantic[email]
 uvicorn==0.21.1
 gunicorn
 ujson
+psutil
 orm[mysql]
 celery
 python-multipart