sentry-arroyo 2.31.0__py3-none-any.whl → 2.31.1__py3-none-any.whl

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.dist-info → sentry_arroyo-2.31.1.dist-info}/METADATA

@@ -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.dist-info → sentry_arroyo-2.31.1.dist-info}/RECORD

@@ -9,7 +9,7 @@ arroyo/backends/abstract.py,sha256=NQ5xG9rjchSUW8URl1WtSYSzMFtwRKB6wtJhWmDyR9E,9
 arroyo/backends/kafka/__init__.py,sha256=xgf-AqHbQkJsh73YokO2uoyyHfZf8XwUp6BULtM8stI,445
 arroyo/backends/kafka/commit.py,sha256=LPsjvX5PPXR62DT6sa5GuSF78qk9F_L--Fz4kw7-m-s,3060
 arroyo/backends/kafka/configuration.py,sha256=g1Y-vdu3OT9pqWuYRp9fw29Nmm1KBBExQLr3VhDsp90,8950
-arroyo/backends/kafka/consumer.py,sha256=nuWgURnaL4V5UzrbaVDPuATfG9cJNCiga1Pm1Dej2uc,29955
+arroyo/backends/kafka/consumer.py,sha256=QrhgcTiJbwZ3rf_nOn9PQL9n8L9chyXmtTGPRRiBxoY,30130
 arroyo/backends/local/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arroyo/backends/local/backend.py,sha256=7odjCnzoGgdo8JHLgG1ntaXa-ZR9GteGkquiA2WAWmM,13880
 arroyo/backends/local/storages/__init__.py,sha256=AGYujdAAcn3osoj9jq84IzTywYbkIDv9wRg2rLhLXeg,104
@@ -46,11 +46,11 @@ examples/transform_and_produce/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 examples/transform_and_produce/batched.py,sha256=st2R6qTneAtV0JFbKP30Ti3sJDYj8Jkbmta9JckKdZU,2636
 examples/transform_and_produce/script.py,sha256=8kSMIjQNqGYEVyE0PvrfJh-a_UYCrJSstTp_De7kyyg,2306
 examples/transform_and_produce/simple.py,sha256=H7xqxItjl4tx34wVW5dy6mB9G39QucAtxkJSBzVmjgA,1637
-sentry_arroyo-2.31.0.dist-info/licenses/LICENSE,sha256=0Ng3MFdEcnz0sVD1XvGBBzbavvNp_7OAM5yVObB46jU,10829
+sentry_arroyo-2.31.1.dist-info/licenses/LICENSE,sha256=0Ng3MFdEcnz0sVD1XvGBBzbavvNp_7OAM5yVObB46jU,10829
 tests/backends/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/backends/mixins.py,sha256=sfNyE0VTeiD3GHOnBYl-9urvPuURI2G1BWke0cz7Dvc,20445
 tests/backends/test_commit.py,sha256=iTHfK1qsBxim0XwxgMvNNSMqDUMEHoYkYBDcgxGBFbs,831
-tests/backends/test_confluent_producer.py,sha256=kLICuiqyWuJPpBWXlvR0wMqX34Xj1tHCUzZHbjPXmxk,2675
+tests/backends/test_confluent_producer.py,sha256=zMCRqXdrWSgkwIspZJ7uhzpkQIFZKNbQeMAG1BeQPms,2993
 tests/backends/test_kafka.py,sha256=6W6EA41X-ECTfcOeivhQxURnmV2Y0fYy-UzDCnJgDsU,12830
 tests/backends/test_kafka_producer.py,sha256=LpwkqnstcCDxemlKZ0FpzNKrP-1UuXXY15P7P-spjhE,3912
 tests/backends/test_local.py,sha256=Mfd4DFuWVSVtl1GomQ6TIoWuJNcAliKqKU0BShPlEMY,3363
@@ -74,7 +74,7 @@ tests/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/utils/test_concurrent.py,sha256=Gwdzym2UZ1HO3rhOSGmzxImWcLFygY8P7MXHT3Q0xTE,455
 tests/utils/test_metrics.py,sha256=bI0EtGgPokMQyEqX58i0-8zvLfxRP2nWaWr2wLMaJ_o,917
 tests/utils/test_retries.py,sha256=AxJLkXWeL9AjHv_p1n0pe8CXXJp24ZQIuYBHfNcmiz4,3075
-sentry_arroyo-2.31.0.dist-info/METADATA,sha256=X_NmQu_bEVG--nVwRfvHQOCTRLoZ4kec0-BTPtV1Ii4,2208
-sentry_arroyo-2.31.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sentry_arroyo-2.31.0.dist-info/top_level.txt,sha256=DVdMZKysL_iIxm5aY0sYgZtP5ZXMg9YBaBmGQHVmDXA,22
-sentry_arroyo-2.31.0.dist-info/RECORD,,
+sentry_arroyo-2.31.1.dist-info/METADATA,sha256=JREC1Nm6b51e0nmz9ANO0N86Qh2th772Ri8t-I5FXJc,2208
+sentry_arroyo-2.31.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sentry_arroyo-2.31.1.dist-info/top_level.txt,sha256=DVdMZKysL_iIxm5aY0sYgZtP5ZXMg9YBaBmGQHVmDXA,22
+sentry_arroyo-2.31.1.dist-info/RECORD,,

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