sentry-arroyo 2.20.9__tar.gz → 2.20.10__tar.gz

{sentry_arroyo-2.20.9/sentry_arroyo.egg-info → sentry_arroyo-2.20.10}/PKG-INFO

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

{sentry_arroyo-2.20.9 → sentry_arroyo-2.20.10}/arroyo/backends/abstract.py

@@ -4,7 +4,16 @@ import logging
 import time
 from abc import ABC, abstractmethod, abstractproperty
 from concurrent.futures import Future
-from typing import Callable, Generic, Mapping, Optional, Sequence, TypeVar, Union
+from typing import (
+    Callable,
+    Generic,
+    Mapping,
+    Optional,
+    Protocol,
+    Sequence,
+    TypeVar,
+    Union,
+)
 
 from arroyo.types import BrokerValue, Partition, Topic, TStrategyPayload
 
@@ -178,11 +187,32 @@ class Consumer(Generic[TStrategyPayload], ABC):
         raise NotImplementedError
 
 
+class ProducerFuture(Protocol, Generic[T]):
+    """
+    An abstract interface for a kind of Future. Stdlib futures are too slow to
+    construct, so we use these.
+    """
+
+    def done(self) -> bool:
+        ...
+
+    def result(self, timeout: float | None = None) -> T:
+        """
+        Return result or raise exception. May block, but does not have to.
+        """
+        ...
+
+    def set_result(self, result: T) -> None:
+        ...
+
+    def set_exception(self, exception: Exception) -> None:
+        ...
+
+
 class SimpleProducerFuture(Generic[T]):
     """
     A stub for concurrent.futures.Future that does not construct any Condition
-    variables, therefore is faster to construct. However, some methods are
-    missing, and result() in particular is not efficient with timeout > 0.
+    variables, therefore is faster to construct.
     """
 
     def __init__(self) -> None:
@@ -202,10 +232,6 @@ class SimpleProducerFuture(Generic[T]):
         # only in tests at most. It is only here for the sake of implementing
         # the contract. If you really need result with timeout>0, you should
         # use the stdlib future.
-        #
-        # If this becomes performance sensitive, we can potentially implement
-        # something more sophisticated such as lazily creating the condition
-        # variable, and synchronizing the creation of that using a global lock.
         while deadline is None or time.time() < deadline:
             if self.result_exception is not None:
                 raise self.result_exception
@@ -222,9 +248,6 @@ class SimpleProducerFuture(Generic[T]):
         self.result_exception = exception
 
 
-ProducerFuture = Union[SimpleProducerFuture[T], Future[T]]
-
-
 class Producer(Generic[TStrategyPayload], ABC):
     @abstractmethod
     def produce(

{sentry_arroyo-2.20.9 → sentry_arroyo-2.20.10/sentry_arroyo.egg-info}/PKG-INFO

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

{sentry_arroyo-2.20.9 → sentry_arroyo-2.20.10}/setup.py

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

{sentry_arroyo-2.20.9 → sentry_arroyo-2.20.10}/tests/backends/test_kafka.py

@@ -167,15 +167,6 @@ class TestKafkaStreams(StreamsTestMixin[KafkaPayload]):
                 else:
                     raise AssertionError("expected EndOfPartition error")
 
-    @pytest.mark.parametrize("use_simple_futures", [True, False])
-    def test_producer_future_behavior(self, use_simple_futures: bool) -> None:
-        with self.get_topic() as topic:
-            with closing(self.get_producer(use_simple_futures)) as producer:
-                future = producer.produce(topic, next(self.get_payloads()))
-                assert not future.done()
-                assert future.result(5.0)
-                assert future.done()
-
     def test_lenient_offset_reset_latest(self) -> None:
         payload = KafkaPayload(b"a", b"0", [])
         with self.get_topic() as topic: