adapta 3.5.12__py3-none-any.whl → 3.5.14__py3-none-any.whl

adapta/_version.py

@@ -1 +1 @@
-__version__ = '3.5.12'
+__version__ = '3.5.14'

adapta/metrics/providers/datadog_provider.py

@@ -25,7 +25,7 @@ from datadog import initialize, statsd, api
 from datadog_api_client.v1.model.metric_metadata import MetricMetadata
 
 from adapta.metrics._base import MetricsProvider
-from adapta.utils import convert_datadog_tags
+from adapta.utils import convert_datadog_tags, doze
 
 
 class EventAlertType(Enum):
@@ -74,19 +74,43 @@ class DatadogMetricsProvider(MetricsProvider):
         )
 
     @classmethod
-    def uds(cls, metric_namespace: str, fixed_tags: dict[str, str] = None, debug=False):
+    def uds(
+        cls,
+        metric_namespace: str,
+        fixed_tags: dict[str, str] = None,
+        debug=False,
+        wait_for_socket_timeout_seconds: int = 0,
+    ):
         """
         Enables sending metrics over UDS (Unix Domain Socket).
         You must have dsdsocket path mounted on your system in order to use this mode.
         Refer to https://docs.datadoghq.com/developers/dogstatsd/unix_socket/?tab=kubernetes
+
+        For most scenarios, wait_for_socket_timeout does not need to be specified. However, if you are not seeing metrics in DD, you might have an issue with socket file not available at the time your application runs.
+        In order to ensure you always start with DSD socket ready, specify wait_for_socket_timeout.
         """
-        return cls(
+        assert wait_for_socket_timeout_seconds >= 0, "Socket file wait timeout must be equal to or greater than 0"
+        socket_path = os.getenv("PROTEUS__DD_STATSD_SOCKET_PATH") or "/var/run/datadog/dsd.socket"
+        wait_time = 0
+        provider = cls(
             metric_namespace=metric_namespace,
             fixed_tags=fixed_tags,
             debug=debug,
-            statsd_socket_path=os.getenv("PROTEUS__DD_STATSD_SOCKET_PATH") or "/var/run/datadog/dsd.socket",
+            statsd_socket_path=socket_path,
         )
 
+        if wait_for_socket_timeout_seconds > 0:
+            while wait_time < wait_for_socket_timeout_seconds:
+                doze(1)
+                wait_time += 1
+                if os.path.exists(socket_path):
+                    return provider
+
+            raise RuntimeError(f"UDS socket not found after {wait_time} seconds")
+
+        # skip DSD await if timeout is set to 0
+        return provider
+
     def update_metric_metadata(self, metric_name: str, metric_metadata: MetricMetadata) -> None:
         """
           Updates metadata of a published metric in DD.
@@ -105,7 +129,7 @@ class DatadogMetricsProvider(MetricsProvider):
         statsd.decrement(metric=metric_name, tags=convert_datadog_tags(tags))
 
     def count(self, metric_name: str, metric_value: int, tags: dict[str, str] | None = None) -> None:
-        raise NotImplementedError
+        statsd.increment(metric=metric_name, value=float(metric_value), tags=convert_datadog_tags(tags))
 
     def gauge(
         self,

{adapta-3.5.12.dist-info → adapta-3.5.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: adapta
-Version: 3.5.12
+Version: 3.5.14
 Summary: Logging, data connectors, monitoring, secret handling and general lifehacks to make data people lives easier.
 License: Apache-2.0
 License-File: LICENSE

{adapta-3.5.12.dist-info → adapta-3.5.14.dist-info}/RECORD

@@ -1,5 +1,5 @@
 adapta/__init__.py,sha256=JSGFwL-Pu--FtbY2HOBYcepwDNOJS10RIR9dL88Iye4,696
-adapta/_version.py,sha256=ehKi8XHLdIUh5ElaqwLkTOx38yWVCVmU_3cAkJ-6CEQ,23
+adapta/_version.py,sha256=SwMwM6zXFsaHCwyOqiN8AMdBGP8q5BsPOeXcl_FlehY,23
 adapta/connectors/__init__.py,sha256=H5mtUGjVkaNkhW-nH7_QElRpfcijduVrW8JsrJHjNnA,631
 adapta/connectors/service_bus/__init__.py,sha256=GDDJ_Y8ua08kZuav4dlrdUzf1ayjAjz2M34eplWD-e8,727
 adapta/connectors/service_bus/_connector.py,sha256=mkvDag9Zd2PYj0fSxStnPtFDjmYcAYOBPr7O62LyDSI,1913
@@ -20,7 +20,7 @@ adapta/metrics/__init__.py,sha256=zSyyuYQfBKAqXXcdG7Tddj2V_KwaskwZnv87Mcb8HQA,70
 adapta/metrics/_base.py,sha256=Pr0r7SO2b0j9QxS3xZRPo-cvZn0xacDX3-Ezh7X0WS0,3214
 adapta/metrics/providers/README.md,sha256=HOJt6BfbUqV8aPv1RPUS2Gm-qaST9geHmVUuVg7sD0A,1930
 adapta/metrics/providers/__init__.py,sha256=H5mtUGjVkaNkhW-nH7_QElRpfcijduVrW8JsrJHjNnA,631
-adapta/metrics/providers/datadog_provider.py,sha256=B-Un1pyUoFZEUSrv8KiQmbUKTdKp6J9c4GvRNE1c0OA,5333
+adapta/metrics/providers/datadog_provider.py,sha256=jsSiPzHD49QvksPOT0tH8H9w7CkKC7aF0cnMdD9ehzI,6401
 adapta/metrics/providers/void_provider.py,sha256=UAIe3zV17r6NwJqFMp5hovFZ3NwX-IJPEutd8ScKZIM,981
 adapta/ml/__init__.py,sha256=2BiAuwbrSVWt-BWSyuHyPaZXBpzMho8RESrG6JR1vDE,686
 adapta/ml/_model.py,sha256=XtFKWrxJfyU-EuBlJm0TcupmEs0mFmPunAkMqrmRGBE,1256
@@ -140,7 +140,7 @@ adapta/utils/decorators/_rate_limit.py,sha256=lzQ81uuaPg-_c5CkvztIetkglbmwdQNb6x
 adapta/utils/metaframe.py,sha256=0MANP7c7jaHcVw3TvOTq4Taz4Pa2u97HqHKp4ml2nxo,5185
 adapta/utils/python_typing/__init__.py,sha256=AhCDfacq0Jwz9rGGkauL4NaozJeYad5IUpCMif056rQ,86
 adapta/utils/python_typing/_functions.py,sha256=YR3YfsvNvB-IX848cMnSty2tLeq1zExARD44AELtfik,541
-adapta-3.5.12.dist-info/METADATA,sha256=L1eq9jiQu861vARPP9pFxAUc0by2XMyoxPBfclljIEQ,3591
-adapta-3.5.12.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-adapta-3.5.12.dist-info/licenses/LICENSE,sha256=0gS6zXsPp8qZhzi1xaGCIYPzb_0e8on7HCeFJe8fOpw,10693
-adapta-3.5.12.dist-info/RECORD,,
+adapta-3.5.14.dist-info/METADATA,sha256=C16RFWPJySbL4H3bSnEp3jm5f7mwcog4gLlTG2Un42A,3591
+adapta-3.5.14.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+adapta-3.5.14.dist-info/licenses/LICENSE,sha256=0gS6zXsPp8qZhzi1xaGCIYPzb_0e8on7HCeFJe8fOpw,10693
+adapta-3.5.14.dist-info/RECORD,,