refactor monitoring clients

pghoard/__init__.py

@@ -4,4 +4,3 @@
 Copyright (c) 2016 Ohmu Ltd
 See LICENSE for details
 """
-from . import mapping, monitoring

pghoard/basebackup/delta.py

@@ -154,7 +154,7 @@ def _delta_upload_hexdigest(
         result_hash = hashlib.blake2s()
 
         def progress_callback(n_bytes: int = 1) -> None:
-            self.metrics.increase("pghoard.basebackup_bytes_uploaded", inc_value=n_bytes, tags={"delta": True})
+            self.metrics.increase("pghoard.basebackup_bytes_uploaded", inc_value=n_bytes, tags={"delta": "True"})
 
         with NamedTemporaryFile(dir=temp_dir, prefix=os.path.basename(chunk_path), suffix=".tmp") as raw_output_obj:
             raw_output_file = cast(FileLikeWithName, raw_output_obj)

pghoard/metrics.py

@@ -2,37 +2,60 @@
 Interface for monitoring clients
 
 """
-import pghoard
+import logging
+from dataclasses import dataclass
+from typing import Any, Dict, Optional, Type
 
+from pghoard.monitoring import (PrometheusClient, PushgatewayClient, SentryClient, StatsClient)
 
-class Metrics:
-    def __init__(self, **configs):
-        self.clients = self._init_clients(configs)
+LOG = logging.getLogger(__name__)
+
+
+class MetricsClient:
+    def __init__(self, config: Dict[str, Any]):
+        self.config = config
+
+    def gauge(self, metric: str, value: float, tags: Optional[Dict[str, str]] = None) -> None:
+        pass
+
+    def increase(self, metric: str, inc_value: int = 1, tags: Optional[Dict[str, str]] = None) -> None:
+        pass
 
-    def _init_clients(self, configs):
-        clients = {}
+    def unexpected_exception(self, ex: Exception, where: str, tags: Optional[Dict[str, str]] = None) -> None:
+        pass
 
-        if not isinstance(configs, dict):
-            return clients
 
-        map_client = pghoard.mapping.clients
-        for k, config in configs.items():
-            if isinstance(config, dict) and k in map_client:
-                path, classname = map_client[k]
-                mod = __import__(path, fromlist=[classname])
-                klass = getattr(mod, classname)
-                clients[k] = klass(config)
+@dataclass()
+class AvailableClient:
+    client_class: Type[MetricsClient]
+    config_key: str
+
+
+class Metrics:
+    available_clients = [
+        AvailableClient(StatsClient, "statsd"),
+        AvailableClient(PrometheusClient, "prometheus"),
+        AvailableClient(PushgatewayClient, "pushgateway"),
+        AvailableClient(SentryClient, "sentry"),
+    ]
+
+    def __init__(self, **configs):
+        self.clients = {}
 
-        return clients
+        for client_info in self.available_clients:
+            client_config = configs.get(client_info.config_key)
+            if isinstance(client_config, dict):
+                LOG.info("Initializing monitoring client %s", client_info.config_key)
+                self.clients[client_info.config_key] = client_info.client_class(client_config)
 
-    def gauge(self, metric, value, tags=None):
+    def gauge(self, metric: str, value: float, tags: Optional[Dict[str, str]] = None) -> None:
         for client in self.clients.values():
             client.gauge(metric, value, tags)
 
-    def increase(self, metric, inc_value=1, tags=None):
+    def increase(self, metric: str, inc_value: int = 1, tags: Optional[Dict[str, str]] = None) -> None:
         for client in self.clients.values():
             client.increase(metric, inc_value, tags)
 
-    def unexpected_exception(self, ex, where, tags=None):
+    def unexpected_exception(self, ex: Exception, where: str, tags: Optional[Dict[str, str]] = None) -> None:
         for client in self.clients.values():
             client.unexpected_exception(ex, where, tags)

pghoard/monitoring/__init__.py

@@ -1,5 +1,5 @@
-import pkgutil
-
-__path__ = pkgutil.extend_path(__path__, __name__)  # type: ignore
-for importer, modname, ispkg in pkgutil.walk_packages(path=__path__, prefix=__name__ + "."):
-    __import__(modname)
+# Copyright (c) 2024 Aiven, Helsinki, Finland. https://aiven.io/
+from .prometheus import PrometheusClient
+from .pushgateway import PushgatewayClient
+from .sentry import SentryClient
+from .statsd import StatsClient

pghoard/monitoring/prometheus.py

@@ -5,9 +5,12 @@
 
 import time
 
+from pghoard.metrics import MetricsClient
 
-class PrometheusClient:
+
+class PrometheusClient(MetricsClient):
     def __init__(self, config):
+        super().__init__(config)
         self._tags = config.get("tags", {})
         self.metrics = {}
 

pghoard/monitoring/pushgateway.py

@@ -6,9 +6,12 @@
 
 import requests
 
+from pghoard.metrics import MetricsClient
 
-class PushgatewayClient:
+
+class PushgatewayClient(MetricsClient):
     def __init__(self, config):
+        super().__init__(config)
         self._endpoint = config.get("endpoint", "")
         self._job = config.get("job", "pghoard")
         self._instance = config.get("instance", "")

pghoard/monitoring/sentry.py

@@ -1,10 +1,15 @@
+# Copyright (c) 2024 Aiven, Helsinki, Finland. https://aiven.io/
 import logging
+from typing import Any, Dict, Optional
+
+from pghoard.metrics import MetricsClient
 
 LOG = logging.getLogger(__name__)
 
 
-class SentryClient:
-    def __init__(self, config):
+class SentryClient(MetricsClient):
+    def __init__(self, config: Dict[str, Any]):
+        super().__init__(config)
         self.sentry = None
         if config is None:
             LOG.info("Sentry configuration not found, skipping setup")
@@ -29,13 +34,7 @@ def __init__(self, config):
         for key, value in tags.items():
             sentry_sdk.set_tag(key, value)
 
-    def gauge(self, metric, value, tags=None):
-        pass
-
-    def increase(self, metric, inc_value=1, tags=None):
-        pass
-
-    def unexpected_exception(self, ex, where, tags=None):
+    def unexpected_exception(self, ex: Exception, where: str, tags: Optional[Dict[str, str]] = None) -> None:
         if not self.sentry:
             return
 

pghoard/monitoring/statsd.py

@@ -8,9 +8,12 @@
 """
 import socket
 
+from pghoard.metrics import MetricsClient
 
-class StatsClient:
+
+class StatsClient(MetricsClient):
     def __init__(self, config):
+        super().__init__(config)
         self._dest_addr = (config.get("host", "127.0.0.1"), config.get("port", 8125))
         self._socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
         self._tags = config.get("tags", {})