Jelajahi Sumber

Revert exception handling

Aleksandr Borzunov 2 tahun lalu
induk
melakukan
505817f49f
2 mengubah file dengan 4 tambahan dan 2 penghapusan
  1. 2 1
      hivemind/averaging/averager.py
  2. 2 1
      hivemind/dht/dht.py

+ 2 - 1
hivemind/averaging/averager.py

@@ -310,7 +310,8 @@ class DecentralizedAverager(mp.Process, ServicerBase):
                     method, args, kwargs = self._inner_pipe.recv()
                 except (OSError, ConnectionError, RuntimeError) as e:
                     logger.exception(e)
-                    break
+                    await asyncio.sleep(self.request_timeout)
+                    continue
                 task = asyncio.create_task(getattr(self, method)(*args, **kwargs))
                 if method == "_shutdown":
                     await task

+ 2 - 1
hivemind/dht/dht.py

@@ -124,7 +124,8 @@ class DHT(mp.Process):
                     method, args, kwargs = self._inner_pipe.recv()
                 except (OSError, ConnectionError, RuntimeError) as e:
                     logger.exception(e)
-                    break
+                    await asyncio.sleep(self._node.protocol.wait_timeout)
+                    continue
                 task = asyncio.create_task(getattr(self, method)(*args, **kwargs))
                 if method == "_shutdown":
                     await task