Add timeout to connect call

This commit is contained in:
atse 2023-07-31 00:34:09 +08:00
parent efa814a0d3
commit d52aafd7f6

View File

@ -23,7 +23,7 @@ class Client:
if connected:
return
"""
self._connecting_task = asyncio.create_task(asyncio.open_connection(host, port))
self._connecting_task = asyncio.create_task(asyncio.wait_for(asyncio.open_connection(host, port), timeout))
try:
self._reader, self._writer = await self._connecting_task
except asyncio.CancelledError: