diff --git a/django_prometheus/conf/__init__.py b/django_prometheus/conf/__init__.py index 85393936..11d42ab9 100644 --- a/django_prometheus/conf/__init__.py +++ b/django_prometheus/conf/__init__.py @@ -1,4 +1,5 @@ from django.conf import settings +from django_prometheus.utils import PowersOf NAMESPACE = "" @@ -22,6 +23,9 @@ float("inf"), ) +PROMETHEUS_TOTAL_BYTES_BUCKETS = PowersOf(2, 30) + if settings.configured: NAMESPACE = getattr(settings, "PROMETHEUS_METRIC_NAMESPACE", NAMESPACE) PROMETHEUS_LATENCY_BUCKETS = getattr(settings, "PROMETHEUS_LATENCY_BUCKETS", PROMETHEUS_LATENCY_BUCKETS) + PROMETHEUS_TOTAL_BYTES_BUCKETS = getattr(settings, "PROMETHEUS_TOTAL_BYTES_BUCKETS", PROMETHEUS_TOTAL_BYTES_BUCKETS) diff --git a/django_prometheus/middleware.py b/django_prometheus/middleware.py index 8c6f88d6..4f83c4a0 100644 --- a/django_prometheus/middleware.py +++ b/django_prometheus/middleware.py @@ -1,8 +1,8 @@ from django.utils.deprecation import MiddlewareMixin from prometheus_client import Counter, Histogram -from django_prometheus.conf import NAMESPACE, PROMETHEUS_LATENCY_BUCKETS -from django_prometheus.utils import PowersOf, Time, TimeSince +from django_prometheus.conf import NAMESPACE, PROMETHEUS_LATENCY_BUCKETS, PROMETHEUS_TOTAL_BYTES_BUCKETS +from django_prometheus.utils import Time, TimeSince class Metrics: @@ -96,7 +96,7 @@ def register(self): Histogram, "django_http_requests_body_total_bytes", "Histogram of requests by body size.", - buckets=PowersOf(2, 30), + buckets=PROMETHEUS_TOTAL_BYTES_BUCKETS, namespace=NAMESPACE, ) # Set in process_template_response @@ -126,7 +126,7 @@ def register(self): Histogram, "django_http_responses_body_total_bytes", "Histogram of responses by body size.", - buckets=PowersOf(2, 30), + buckets=PROMETHEUS_TOTAL_BYTES_BUCKETS, namespace=NAMESPACE, ) self.responses_by_charset = self.register_metric(