close properly tcp connection on exception

dmachard · Feb 3, 2021 · 74fba91 · 74fba91
1 parent 3c542e6
commit 74fba91
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dnstap_receiver/inputs/input_tcpclient.py b/dnstap_receiver/inputs/input_tcpclient.py
@@ -81,10 +81,10 @@ async def tcp_client(cfg, cfg_input, queues_list, stats, geoip_reader, cache):
     except asyncio.CancelledError:
         clogger.debug(f'Input handler: {peername} - closing connection.')
         writer.close()
-        await writer.wait_closed()
     except asyncio.IncompleteReadError:
         clogger.debug(f'Input handler: {peername} - disconnected')
     finally:
+        writer.close()
         clogger.debug(f'Input handler: {peername} - closed')
 
 async def start_tcpclient(cfg, queues_list, stats, geoip_reader, cache):