improve worker reconnect function

pull/17/head
unknown 2023-07-13 01:19:28 -05:00
parent 79d6a689cd
commit 9d48e2732d
No known key found for this signature in database
GPG Key ID: CA376082283AF69A
2 changed files with 11 additions and 6 deletions

View File

@ -117,11 +117,16 @@ class UI:
def reconnect_remotes(self):
workers = self.world.get_workers()
for worker in workers:
if worker.state == State.UNAVAILABLE and worker.reachable():
logger.info(f"Worker '{worker.uuid}' is now online, marking as available")
worker.state = State.IDLE
for worker in self.world._workers:
logger.debug(f"checking if worker '{worker.uuid}' is now reachable...")
reachable = worker.reachable()
if worker.state == State.UNAVAILABLE:
if reachable:
logger.info(f"worker '{worker.uuid}' is now online, marking as available")
worker.state = State.IDLE
else:
logger.info(f"worker '{worker.uuid}' is still unreachable")
# end handlers

View File

@ -392,7 +392,7 @@ class Worker:
else:
raise InvalidWorkerResponse(e)
except requests.exceptions.ConnectionError:
except requests.RequestException:
self.mark_unreachable()
return