fix: retry once on tcp request error

close #31
gera2ld · Apr 10, 2024 · ee92c6a · ee92c6a
1 parent e820664
commit ee92c6a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/async_dns/request/tcp.py b/async_dns/request/tcp.py
@@ -29,7 +29,11 @@ async def request(req, addr, timeout=3.0):
     '''
     Send raw data with a connection pool.
     '''
-    result = await asyncio.wait_for(_request(req, addr), timeout)
+    try:
+        result = await asyncio.wait_for(_request(req, addr), timeout)
+    except asyncio.IncompleteReadError:
+        # Retry once in case the existing connection is dead
+        result = await asyncio.wait_for(_request(req, addr), timeout)
     return result