浏览代码

Revert exception handling

Aleksandr Borzunov 2 年之前
父节点
当前提交
505817f49f
共有 2 个文件被更改,包括 4 次插入2 次删除
  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