Merge pull request #72 from uc-cdis/update_registry_in_constructors

Pass Registry to Counter and Gauge constructors

Author: nss10

cdispyutils/metrics.py

@@ -109,7 +109,7 @@ def increment_counter(self, name, labels, description=""):
             logger.info(
                 f"Creating counter '{name}' with description '{description}' and labels: {labels}"
             )
-            self.prometheus_metrics[name] = Counter(name, description, [*labels.keys()])
+            self.prometheus_metrics[name] = Counter(name, description, [*labels.keys()], registry=self._registry)
         elif type(self.prometheus_metrics[name]) is not Counter:
             raise ValueError(
                 f"Trying to create counter '{name}' but a {type(self.prometheus_metrics[name])} with this name already exists"
@@ -177,7 +177,7 @@ def _create_gauge_if_not_exist(self, name, labels, value, description):
             logger.info(
                 f"Creating gauge '{name}' with description '{description}' and labels: {labels}"
             )
-            self.prometheus_metrics[name] = Gauge(name, description, [*labels.keys()])
+            self.prometheus_metrics[name] = Gauge(name, description, [*labels.keys()], registry=self._registry)
         elif type(self.prometheus_metrics[name]) is not Gauge:
             raise ValueError(
                 f"Trying to create gauge '{name}' but a {type(self.prometheus_metrics[name])} with this name already exists"

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cdispyutils"
-version = "2.4.0"
+version = "2.4.1"
 description = "This package includes several utility Python tools for the Gen3 stack."
 authors = ["CTDS UChicago <cdis@uchicago.edu>"]
 license = "Apache-2.0"