Prevented socket hanging on reload

tne-lab · Sep 12, 2024 · 264de58 · 264de58
1 parent 3ac8565
commit 264de58
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/pybehave/Events/OENetworkLogger.py b/pybehave/Events/OENetworkLogger.py
@@ -27,11 +27,13 @@ class OENetworkLogger(EventLogger):
 
     def __init__(self, name: str, address: str, port: str):
         super().__init__(name)
-        context = zmq.Context()
+        self.context = zmq.Context()
         self.fd = None
         self.event_count = 0
-        self.socket = context.socket(zmq.REQ)
+        self.socket = self.context.socket(zmq.REQ)
         self.socket.set(zmq.REQ_RELAXED, True)
+        self.socket.set(zmq.REQ_CORRELATE, True)
+        self.socket.setsockopt(zmq.LINGER, 5)
         self.socket.connect("tcp://" + address + ":" + str(port))
 
     def send_ttl_event(self, ec: int, ttl_type: str | float) -> None:

diff --git a/pybehave/Events/OEWidget.py b/pybehave/Events/OEWidget.py
@@ -26,6 +26,8 @@ def __init__(self, name: str, address: str, port: str):
         self.event_count = 0
         self.socket = context.socket(zmq.REQ)
         self.socket.set(zmq.REQ_RELAXED, True)
+        self.socket.set(zmq.REQ_CORRELATE, True)
+        self.socket.setsockopt(zmq.LINGER, 5)
         self.socket.connect("tcp://" + address + ":" + str(port))
 
         self.layout = QVBoxLayout(self)

diff --git a/pybehave/Sources/OESource.py b/pybehave/Sources/OESource.py
@@ -36,6 +36,8 @@ def initialize(self):
 
             self.out_socket = self.context.socket(zmq.REQ)
             self.out_socket.set(zmq.REQ_RELAXED, True)
+            self.out_socket.set(zmq.REQ_CORRELATE, True)
+            self.out_socket.setsockopt(zmq.LINGER, 5)
             self.out_socket.connect("tcp://" + self.address + ":" + str(self.out_port))
 
             self.available = True