Merge pull request #333 from ansforge/converter/add-metrics

Converter/add metrics

Author: Clemsazert

converter/converter/converter.py

@@ -1,5 +1,8 @@
 from flask import Flask, request, jsonify, g
 import logging
+from werkzeug.middleware.dispatcher import DispatcherMiddleware
+from prometheus_client import make_wsgi_app, Histogram
+
 from converter.conversion_strategy.conversion_strategy import conversion_strategy
 from converter.utils import (
     get_sender,
@@ -14,13 +17,24 @@
 app = Flask(__name__)
 logger = logging.getLogger(__name__)
 
+# Add prometheus wsgi middleware to route /metrics requests
+# ignore typing issue with reassigning method
+app.wsgi_app = DispatcherMiddleware(app.wsgi_app, {"/metrics": make_wsgi_app()})  # type: ignore[assignment]
+
+
+convertion_timer = Histogram(
+    "conversion_duration_seconds",
+    "The number of seconds it took to the /convert endpoint to answer",
+)
+
 
 def raise_error(message, code: int = 400):
     logger.error(message)
     return jsonify({"error": message}), code
 
 
 @app.route("/convert", methods=["POST"])
+@convertion_timer.time()
 def convert():
     if not request.is_json:
         return raise_error("Content-Type must be application/json")

converter/pyproject.toml

@@ -8,6 +8,7 @@ dependencies = [
     "flask>=3.1.2",
     "gunicorn>=23.0.0",
     "jsonpath-ng>=1.7.0",
+    "prometheus-client>=0.23.1",
     "pydantic>=2.11.9",
     "python-json-logger>=4.0.0",
     "pyyaml>=6.0.2",