Add timeout

mvdwetering · Mar 30, 2024 · 57e604a · 57e604a
1 parent 357f30a
commit 57e604a
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/ynca/server.py b/ynca/server.py
@@ -149,6 +149,8 @@ class YncaCommandHandler(socketserver.StreamRequestHandler):
     client.
     """
 
+    timeout = 40  # Receiver disconnects after 40 seconds of no traffic
+
     def __init__(self, request, client_address, server: YncaServer):
         self.store = server.store
         self.disconnect_after_receiving_num_commands = (
@@ -283,9 +285,14 @@ def handle(self):
 
         print(f"--- Client connected from: {self.client_address[0]}")
         while True:
-            bytes_line = self.rfile.readline()
-            if bytes_line == b"":
-                print("--- Client disconnected")
+            try:
+                bytes_line = self.rfile.readline()
+                if bytes_line == b"":
+                    print("--- Client disconnected")
+                    print("--- Waiting for connections")
+                    return
+            except TimeoutError:
+                print("--- Disconnecting client because of timeout")
                 print("--- Waiting for connections")
                 return