improve worker reconnect function
parent
79d6a689cd
commit
9d48e2732d
|
|
@ -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
|
||||
|
||||
|
|
|
|||
|
|
@ -392,7 +392,7 @@ class Worker:
|
|||
else:
|
||||
raise InvalidWorkerResponse(e)
|
||||
|
||||
except requests.exceptions.ConnectionError:
|
||||
except requests.RequestException:
|
||||
self.mark_unreachable()
|
||||
return
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue