Create middleware/metrics.py

middleware/metrics.py

@@ -0,0 +1,25 @@
+import time
+from functools import wraps
+from flask import request
+
+REQUESTS_TOTAL = {}
+REQUEST_DURATION = {}
+
+def track_request(f):
+    @wraps(f)
+    def wrapper(*args, **kwargs):
+        start = time.time()
+        try:
+            response = f(*args, **kwargs)
+            return response
+        finally:
+            duration = time.time() - start
+            endpoint = request.path
+            if endpoint not in REQUEST_DURATION:
+                REQUEST_DURATION[endpoint] = []
+            REQUEST_DURATION[endpoint].append(duration)
+            REQUESTS_TOTAL[endpoint] = REQUESTS_TOTAL.get(endpoint, 0) + 1
+    return wrapper
+
+def get_metrics():
+    return {"total_requests": REQUESTS_TOTAL, "avg_duration": {k: sum(v)/len(v) for k, v in REQUEST_DURATION.items()}}