From 6a85e67abffe7c2e93a39a7967cdeaf577aeacb3 Mon Sep 17 00:00:00 2001
From: CHAMPION923 <1056156912@qq.com>
Date: Wed, 4 Jun 2025 13:01:52 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0DEBUG=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 lib/presenter/presenter.py | 5 +++--
 lib/tcp/tcp_server.py      | 6 +++++-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/lib/presenter/presenter.py b/lib/presenter/presenter.py
index 7950b43..d30b721 100644
--- a/lib/presenter/presenter.py
+++ b/lib/presenter/presenter.py
@@ -227,7 +227,8 @@ class Presenter:
                         continue
                 except ConnectionResetError:
                     print("Warring: clietn force disconnect!!! ")
-                    break
+                    server.conn_close()
+                    continue
                 except socket.error as e:
                     print(f"Net Error: {e}")
                     break
@@ -271,10 +272,10 @@ class Presenter:
             print("KeyboardInterrupt (Ctrl+C) shutting down")
 
         finally:
+            server.close()
             for key in self.process3d_info.keys():
                 self.process3d_info[key].in_q.put(0)
             for key in self.process2d_info.keys():
                 self.process2d_info[key].in_q.put(0)
             ArenaCamera.shutdown()
             print("关闭连接")
-            server.close()
diff --git a/lib/tcp/tcp_server.py b/lib/tcp/tcp_server.py
index 1011490..b9b9dc6 100644
--- a/lib/tcp/tcp_server.py
+++ b/lib/tcp/tcp_server.py
@@ -1,3 +1,5 @@
+from _typeshed import Self
+from ctypes import set_errno
 import socket
 
 
@@ -25,6 +27,8 @@ class TcpServer:
     def send_data(self, data: bytes):
         self.conn.sendall(data)
 
-    def close(self):
+    def conn_close(self):
         self.conn.close()
+
+    def close(self):
         self.sock.close()