From e84d36e165507e5209d78ef2161e6042db26fde3 Mon Sep 17 00:00:00 2001 From: Adir111 Date: Wed, 21 Aug 2024 16:07:43 +0300 Subject: [PATCH] Handled error & stopped properly. --- hkube_python_wrapper/wrapper/algorunner.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/hkube_python_wrapper/wrapper/algorunner.py b/hkube_python_wrapper/wrapper/algorunner.py index 77a2abf..da20fbd 100644 --- a/hkube_python_wrapper/wrapper/algorunner.py +++ b/hkube_python_wrapper/wrapper/algorunner.py @@ -53,6 +53,7 @@ def __init__(self): self._redirectLogs = False self._printThread = 0 self._done = True + self._error = False DaemonThread.__init__(self, "WorkerListener") @staticmethod @@ -319,9 +320,8 @@ def _getMethod(self, name): def _aliveSignal(self): def routine(): - while not self._done and not self.stopped: + while not self._done and not self._stopped and not self._error: self._sendCommand(messages.outgoing.alive, None) - print("stopped is: ", self._stopped) time.sleep(5) thread = threading.Thread(target=routine) @@ -575,6 +575,7 @@ def _sendCommand(self, command, data): def sendError(self, error): try: + self._error = True log.error("Sending error to worker " + str(error)) self._wsc.send({ 'command': messages.outgoing.error,