sentry-arroyo 2.31.0__tar.gz → 2.31.1__tar.gz

{sentry_arroyo-2.31.0/sentry_arroyo.egg-info → sentry_arroyo-2.31.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sentry-arroyo
-Version: 2.31.0
+Version: 2.31.1
 Summary: Arroyo is a Python library for working with streaming data.
 Home-page: https://github.com/getsentry/arroyo
 Author: Sentry

{sentry_arroyo-2.31.0 → sentry_arroyo-2.31.1}/arroyo/backends/kafka/consumer.py

@@ -757,6 +757,7 @@ class ConfluentProducer(ConfluentKafkaProducer):  # type: ignore[misc]
     def __init__(self, configuration: Mapping[str, Any]) -> None:
         super().__init__(configuration)
         self.__metrics = get_metrics()
+        self.producer_name = configuration.get("client.id") or None
 
     def __metrics_delivery_callback(
         self,
@@ -768,10 +769,14 @@ class ConfluentProducer(ConfluentKafkaProducer):  # type: ignore[misc]
         else:
             status = "success"
 
+        tags = {"status": status}
+        if self.producer_name:
+            tags["producer_name"] = self.producer_name
+
         self.__metrics.increment(
             "arroyo.producer.produce_status",
             1,
-            tags={"status": status},
+            tags=tags,
         )
 
     def __delivery_callback(

{sentry_arroyo-2.31.0 → sentry_arroyo-2.31.1/sentry_arroyo.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sentry-arroyo
-Version: 2.31.0
+Version: 2.31.1
 Summary: Arroyo is a Python library for working with streaming data.
 Home-page: https://github.com/getsentry/arroyo
 Author: Sentry

{sentry_arroyo-2.31.0 → sentry_arroyo-2.31.1}/setup.py

@@ -10,7 +10,7 @@ def get_requirements() -> Sequence[str]:
 
 setup(
     name="sentry-arroyo",
-    version="2.31.0",
+    version="2.31.1",
     author="Sentry",
     author_email="oss@sentry.io",
     license="Apache-2.0",

{sentry_arroyo-2.31.0 → sentry_arroyo-2.31.1}/tests/backends/test_confluent_producer.py

@@ -24,11 +24,17 @@ class TestConfluentProducer:
 
     def test_metrics_callback_records_success(self) -> None:
         """Test that the metrics callback records success metric"""
-        producer = ConfluentProducer({"bootstrap.servers": "fake:9092"})
+        producer = ConfluentProducer(
+            {"bootstrap.servers": "fake:9092", "client.id": "test-producer-name"}
+        )
         mock_message = mock.Mock(spec=ConfluentMessage)
         producer._ConfluentProducer__metrics_delivery_callback(None, mock_message)
         assert (
-            Increment("arroyo.producer.produce_status", 1, {"status": "success"})
+            Increment(
+                "arroyo.producer.produce_status",
+                1,
+                {"status": "success", "producer_name": "test-producer-name"},
+            )
             in TestingMetricsBackend.calls
         )
 
@@ -45,7 +51,9 @@ class TestConfluentProducer:
 
     def test_delivery_callback_wraps_user_callback(self) -> None:
         """Test that the delivery callback wrapper calls both metrics and user callbacks"""
-        producer = ConfluentProducer({"bootstrap.servers": "fake:9092"})
+        producer = ConfluentProducer(
+            {"bootstrap.servers": "fake:9092", "client.id": "test-producer-name"}
+        )
         user_callback_invoked = []
 
         def user_callback(
@@ -57,7 +65,11 @@ class TestConfluentProducer:
         mock_message = mock.Mock(spec=ConfluentMessage)
         wrapped(None, mock_message)
         assert (
-            Increment("arroyo.producer.produce_status", 1, {"status": "success"})
+            Increment(
+                "arroyo.producer.produce_status",
+                1,
+                {"status": "success", "producer_name": "test-producer-name"},
+            )
             in TestingMetricsBackend.calls
         )
         assert len(user_callback_invoked) == 1