sentry-arroyo 2.20.7__tar.gz → 2.20.8__tar.gz

{sentry_arroyo-2.20.7/sentry_arroyo.egg-info → sentry_arroyo-2.20.8}/PKG-INFO

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

{sentry_arroyo-2.20.7 → sentry_arroyo-2.20.8}/arroyo/processing/strategies/run_task_with_multiprocessing.py

@@ -581,7 +581,7 @@ class RunTaskWithMultiprocessing(
                 self.__metrics.increment("sigchld.detected")
                 raise ChildProcessTerminated(signum)
 
-        signal.signal(signal.SIGCHLD, handle_sigchld)
+        self.original_sigchld = signal.signal(signal.SIGCHLD, handle_sigchld)
 
     def __submit_batch(self, input_block_too_small: bool) -> None:
         assert self.__batch_builder is not None
@@ -839,14 +839,19 @@ class RunTaskWithMultiprocessing(
             # compression.)
             self.__batch_builder.append(message)
 
-    def close(self) -> None:
+    def _do_close(self) -> None:
         self.__closed = True
 
+        signal.signal(signal.SIGCHLD, self.original_sigchld)
+
+    def close(self) -> None:
+        self._do_close()
+
         if self.__batch_builder is not None and len(self.__batch_builder) > 0:
             self.__submit_batch(False)
 
     def terminate(self) -> None:
-        self.__closed = True
+        self._do_close()
 
         logger.info("Terminating %r...", self.__pool)
 

{sentry_arroyo-2.20.7 → sentry_arroyo-2.20.8/sentry_arroyo.egg-info}/PKG-INFO

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

{sentry_arroyo-2.20.7 → sentry_arroyo-2.20.8}/setup.py

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

{sentry_arroyo-2.20.7 → sentry_arroyo-2.20.8}/tests/processing/strategies/test_run_task_with_multiprocessing.py

@@ -1,8 +1,9 @@
 import multiprocessing
+import signal
 import time
 from datetime import datetime
 from multiprocessing.managers import SharedMemoryManager
-from typing import Any
+from typing import Any, Generator
 from unittest.mock import Mock, call
 
 import pytest
@@ -25,6 +26,12 @@ from tests.metrics import TestingMetricsBackend
 from tests.metrics import Timing as TimingCall
 
 
+@pytest.fixture(autouse=True)
+def does_not_leak_sigchild_handler() -> Generator[None, None, None]:
+    yield
+    assert isinstance(signal.getsignal(signal.SIGCHLD), int)
+
+
 def test_message_batch() -> None:
     partition = Partition(Topic("test"), 0)