827 del impl client sdk (#828)

implemented `__del__` for ClientSDK
insight-platform · Aug 5, 2024 · e83569a · e83569a
1 parent d724962
commit e83569a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/savant/client/runner/sink.py b/savant/client/runner/sink.py
@@ -56,6 +56,10 @@ def __init__(
         self._source = self._build_zeromq_source(socket, receive_timeout, receive_hwm)
         self._source.start()
 
+    def __del__(self):
+        logger.info('Terminating ZeroMQ connection')
+        self._source.terminate()
+
     @abstractmethod
     def _build_zeromq_source(self, socket: str, receive_timeout: int, receive_hwm: int):
         pass

diff --git a/savant/client/runner/source.py b/savant/client/runner/source.py
@@ -103,6 +103,10 @@ def __init__(
             self._pipeline.sampling_period = 1
         self._writer.start()
 
+    def __del__(self):
+        logger.info('Terminating ZeroMQ connection')
+        self._writer.shutdown()
+
     def __call__(self, source: Frame, send_eos: bool = True) -> SourceResult:
         """Send a single frame to ZeroMQ socket.
 
@@ -275,6 +279,10 @@ class AsyncSourceRunner(SourceRunner):
 
     _writer: NonBlockingWriter
 
+    def __del__(self):
+        logger.info('Terminating ZeroMQ connection')
+        self._writer.shutdown()
+
     async def __call__(self, source: Frame, send_eos: bool = True) -> SourceResult:
         return await self.send(source, send_eos)