Halt threads if shutdown

This commit is contained in:
vic
2023-11-05 15:58:25 -05:00
parent 307c75eb3e
commit c0783810f0

View File

@@ -47,6 +47,7 @@ class Relay:
self.queue = Queue() self.queue = Queue()
def connect(self, ssl_options: dict = None, proxy: dict = None): def connect(self, ssl_options: dict = None, proxy: dict = None):
if self.shutdown: return
self.ws = WebSocketApp( self.ws = WebSocketApp(
self.url, self.url,
on_open=self._on_open, on_open=self._on_open,
@@ -98,10 +99,12 @@ class Relay:
message = self.queue.get(timeout=1) message = self.queue.get(timeout=1)
self.ws.send(message) self.ws.send(message)
self.num_sent_events += 1 self.num_sent_events += 1
except: except Exception as e:
if shutdown(): if shutdown():
break break
else: else:
if shutdown():
break
time.sleep(0.1) time.sleep(0.1)
def add_subscription(self, id, filters: Filters): def add_subscription(self, id, filters: Filters):