Feat/django prometheus

connect/api/prometheus/view.py

@@ -0,0 +1,15 @@
+import os
+
+from django.http import HttpResponseForbidden
+from django_prometheus.exports import ExportToDjangoView
+
+
+def metrics_view(request):
+    auth_token = request.headers.get("Authorization")
+    prometheus_auth_token = os.environ.get("PROMETHEUS_AUTH_TOKEN")
+
+    expected_token = f"Bearer {prometheus_auth_token}"
+    if not auth_token or auth_token != expected_token:
+        return HttpResponseForbidden("Access denied")
+
+    return ExportToDjangoView(request)

connect/settings.py

@@ -153,9 +153,11 @@
     "corsheaders",
     "django_grpc_framework",
     "stripe",
+    "django_prometheus",
 ]
 
 MIDDLEWARE = [
+    "django_prometheus.middleware.PrometheusBeforeMiddleware",
     "elasticapm.contrib.django.middleware.TracingMiddleware",
     "elasticapm.contrib.django.middleware.Catch404Middleware",
     "django.middleware.security.SecurityMiddleware",
@@ -168,6 +170,7 @@
     "django.contrib.auth.middleware.AuthenticationMiddleware",
     "django.contrib.messages.middleware.MessageMiddleware",
     "django.middleware.clickjacking.XFrameOptionsMiddleware",
+    "django_prometheus.middleware.PrometheusAfterMiddleware",
 ]
 
 ROOT_URLCONF = "connect.urls"

connect/urls.py

@@ -30,6 +30,7 @@
 )
 from connect.billing.views import StripeHandler
 from connect.api.v2 import routers as api_v2_urls
+from connect.api.prometheus.view import metrics_view
 
 
 api_v2_urls = [path("", include(api_v2_urls))]
@@ -51,6 +52,7 @@
     path("v1/", include(rookly_api_v1_urls)),
     path("v2/", include(api_v2_urls)),
     url(r"^handlers/stripe/$", StripeHandler.as_view(), name="handlers.stripe_handler"),
+    path("api/prometheus/metrics", metrics_view, name="metrics_view"),
 ]
 
 urlpatterns += staticfiles_urlpatterns()

pyproject.toml

@@ -48,6 +48,7 @@ gevent = "22.10.2"
 setuptools = "*"
 factory-boy = "^3.3.0"
 pyjwt = "^2.9.0"
+django-prometheus = "^2.3.1"
 
 [tool.poetry.dev-dependencies]
 flake8 = "~=3.9.2"