feat: add prometheus metrics

SwissDataScienceCenter · Oct 13, 2023 · 689929c · 689929c
1 parent 80f70f5
commit 689929c
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 6 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -114,6 +114,7 @@ redis = { version = ">=3.5.3,<4.6.0,!=4.5.5", optional = true }
 rq = { version = "==1.15.0", optional = true }
 sentry-sdk = { version = ">=1.5.11,<1.26.0", extras = ["flask"],  optional = true }
 walrus = { version = ">=0.8.2,<0.10.0", optional = true }
+prometheus-flask-exporter = "^0.22.4"
 
 [tool.poetry.group.dev.dependencies]
 black = "==23.1.0"
@@ -301,6 +302,7 @@ module = [
     "pexpect",
     "PIL",
     "pluggy",
+    "prometheus_flask_exporter",
     "psutil",
     "pyld",
     "pyshacl",

diff --git a/renku/ui/service/entrypoint.py b/renku/ui/service/entrypoint.py
@@ -22,6 +22,7 @@
 import sentry_sdk
 from flask import Flask, Response, jsonify, request, url_for
 from jwt import InvalidTokenError
+from prometheus_flask_exporter import PrometheusMetrics
 from sentry_sdk.integrations.flask import FlaskIntegration
 from sentry_sdk.integrations.redis import RedisIntegration
 from sentry_sdk.integrations.rq import RqIntegration
@@ -73,6 +74,8 @@ def create_app(custom_exceptions=True):
     app.config["MAX_CONTENT_LENGTH"] = MAX_CONTENT_LENGTH
 
     app.config["cache"] = cache
+    metrics = PrometheusMetrics.for_app_factory()
+    metrics.init_app(app)
 
     build_routes(app)