Merge pull request #217 from dehort/add_timeout_for_waiting_on_hi_response

Add a timeout for waiting on the HI response from the peer node.

Reviewed-by: https://github.com/apps/ansible-zuul

Author: ansible-zuul[bot]

receptor/connection/base.py

@@ -142,7 +142,7 @@ async def drain_buf(self, item):
 
     async def _wait_handshake(self):
         logger.debug("waiting for HI")
-        response = await self.buf.get()  # TODO: deal with timeout
+        response = await self.buf.get(timeout=20.0)
         self.remote_id = response.header["id"]
         await self.register()
         await self.receptor.recalculate_and_send_routes_soon()

receptor/messages/framed.py

@@ -287,8 +287,8 @@ async def finish(self):
         self.to_read = 0
         self.bb = FileBackedBuffer.from_temp()
 
-    async def get(self):
-        return await self.q.get()
+    async def get(self, timeout=None):
+        return await asyncio.wait_for(self.q.get(), timeout)
 
     def get_nowait(self):
         return self.q.get_nowait()