소스 검색

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