sentry-arroyo 2.28.1__tar.gz → 2.28.3__tar.gz

{sentry_arroyo-2.28.1/sentry_arroyo.egg-info → sentry_arroyo-2.28.3}/PKG-INFO

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

{sentry_arroyo-2.28.1 → sentry_arroyo-2.28.3}/arroyo/processing/strategies/produce.py

@@ -1,7 +1,9 @@
 import logging
 import time
 from collections import deque
-from typing import Deque, Optional, Tuple, Union
+from typing import Deque, Optional, Tuple, Union, cast
+
+from confluent_kafka import KafkaError, KafkaException
 
 from arroyo.backends.abstract import Producer, ProducerFuture
 from arroyo.processing.strategies.abstract import MessageRejected, ProcessingStrategy
@@ -78,6 +80,15 @@ class Produce(ProcessingStrategy[Union[FilteredPayload, TStrategyPayload]]):
                     self.__metrics.increment(
                         "arroyo.producer.produce_status", tags={"status": "success"}
                     )
+                except KafkaException as e:
+                    self.__metrics.increment(
+                        "arroyo.producer.produce_status",
+                        tags={
+                            "status": "error",
+                            "code": str(cast(KafkaError, e.args[0]).code),
+                        },
+                    )
+                    raise e
                 except Exception as e:
                     self.__metrics.increment(
                         "arroyo.producer.produce_status", tags={"status": "error"}

{sentry_arroyo-2.28.1 → sentry_arroyo-2.28.3/sentry_arroyo.egg-info}/PKG-INFO

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

{sentry_arroyo-2.28.1 → sentry_arroyo-2.28.3}/setup.py

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

{sentry_arroyo-2.28.1 → sentry_arroyo-2.28.3}/tests/processing/test_processor.py

@@ -3,7 +3,6 @@ from datetime import datetime, timedelta
 from typing import Any, Mapping, Optional, Sequence, cast
 from unittest import mock
 
-import py.path
 import pytest
 
 from arroyo.backends.local.backend import LocalBroker
@@ -583,7 +582,7 @@ def test_dlq() -> None:
     assert dlq_policy.producer.produce.call_count == 1
 
 
-def test_healthcheck(tmpdir: py.path.local) -> None:
+def test_healthcheck(tmp_path: Any) -> None:
     """
     Test healthcheck strategy e2e with StreamProcessor, to ensure the
     combination of both actually touches the file often enough.
@@ -599,7 +598,7 @@ def test_healthcheck(tmpdir: py.path.local) -> None:
     strategy.submit.side_effect = InvalidMessage(partition, 1)
     factory = mock.Mock()
     factory.create_with_partitions.return_value = Healthcheck(
-        healthcheck_file=str(tmpdir.join("health.txt")), next_step=strategy
+        healthcheck_file=str(tmp_path / "health.txt"), next_step=strategy
     )
 
     processor: StreamProcessor[int] = StreamProcessor(
@@ -617,11 +616,11 @@ def test_healthcheck(tmpdir: py.path.local) -> None:
     assignment_callback(offsets)
 
     processor._run_once()
-    health_mtime = tmpdir.join("health.txt").mtime()
+    health_mtime = (tmp_path / "health.txt").stat().st_mtime
     assert health_mtime < time.time() + 1
 
     processor._run_once()
-    assert tmpdir.join("health.txt").mtime() == health_mtime
+    assert (tmp_path / "health.txt").stat().st_mtime == health_mtime
 
 
 def test_processor_pause_with_invalid_message() -> None: