From cea21eb7e91b4fba791f7cc489d652bf76d6e03b Mon Sep 17 00:00:00 2001 From: Pavel Kirilin Date: Mon, 4 May 2026 15:45:21 +0200 Subject: [PATCH] Fixed prometheus collectors. --- taskiq/middlewares/prometheus_middleware.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/taskiq/middlewares/prometheus_middleware.py b/taskiq/middlewares/prometheus_middleware.py index 56837cf3..01f14867 100644 --- a/taskiq/middlewares/prometheus_middleware.py +++ b/taskiq/middlewares/prometheus_middleware.py @@ -84,22 +84,17 @@ def startup(self) -> None: This function starts prometheus server. It starts it only in case if it's a worker process. """ - from prometheus_client import ( # noqa: PLC0415 - CollectorRegistry, - start_http_server, - ) + from prometheus_client import REGISTRY, start_http_server # noqa: PLC0415 from prometheus_client.multiprocess import ( # noqa: PLC0415 MultiProcessCollector, ) if self.broker.is_worker_process: try: - registry = CollectorRegistry() - MultiProcessCollector(registry) + MultiProcessCollector(REGISTRY) start_http_server( port=self.server_port, addr=self.server_addr, - registry=registry, ) except OSError as exc: logger.debug("Cannot start prometheus server: %s", exc)