Workaround the occasional delayed stop of a Client instance

delivrance · delivrance · commit 2ba921c84d6f · 2020-03-30T16:59:22.000+02:00
diff --git a/pyrogram/connection/transport/tcp/tcp.py b/pyrogram/connection/transport/tcp/tcp.py
@@ -19,6 +19,7 @@
 import ipaddress
 import logging
 import socket
+import time
 
 try:
     import socks
@@ -72,6 +73,9 @@ def close(self):
         except OSError:
             pass
         finally:
+            # A tiny sleep placed here helps avoiding .recv(n) hanging until the timeout.
+            # This is a workaround that seems to fix the occasional delayed stop of a client.
+            time.sleep(0.001)
             super().close()
 
     def recvall(self, length: int) -> bytes or None:
