Merge pull request #350 from YuriCat/fix/disconnect_after_timeout

fix: disconnect after socket.timeout
DeNA · Apr 19, 2024 · 6008b1c · 6008b1c
2 parents cce6a40 + 4ac69aa
commit 6008b1c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/handyrl/connection.py b/handyrl/connection.py
@@ -204,6 +204,8 @@ def _send_thread(self):
             conn, send_data = self.output_queue.get()
             try:
                 conn.send(send_data)
+            except TimeoutError:
+                self.disconnect(conn)
             except ConnectionResetError:
                 self.disconnect(conn)
             except BrokenPipeError:
@@ -215,6 +217,9 @@ def _recv_thread(self):
             for conn in conns:
                 try:
                     recv_data = conn.recv()
+                except TimeoutError:
+                    self.disconnect(conn)
+                    continue
                 except ConnectionResetError:
                     self.disconnect(conn)
                     continue