socket perdoling

transport_tcp.py

@@ -2,6 +2,7 @@ import asyncio
 import logging
 from asyncio.streams import StreamReader, StreamWriter
 from time import time
+import socket
 
 from logger import logger
 
@@ -35,9 +36,10 @@ class TCPTransport:
         self._buf = bytearray()
         self._recv_q = asyncio.Queue()
         try:
-            #sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-            #sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
-            reader, writer = await asyncio.open_connection(self.host, self.port)
+            reader, writer = await asyncio.open_connection(host = self.host, port = self.port)
+            sock = writer.transport.get_extra_info('socket')
+            if sock:
+                sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
             self.socket_start = time()
         except Exception as e:
             self._error = e