From 9699bbacbcb6c3087a85cfe3d048f7d792cb2f5d Mon Sep 17 00:00:00 2001
From: renzhiyuan <465386466@qq.com>
Date: Wed, 24 Sep 2025 18:08:21 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BB=93=E6=9E=84=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/biz/router.go    | 14 --------------
 internal/services/chat.go |  6 +++++-
 2 files changed, 5 insertions(+), 15 deletions(-)

diff --git a/internal/biz/router.go b/internal/biz/router.go
index 2db6fbf..32ef390 100644
--- a/internal/biz/router.go
+++ b/internal/biz/router.go
@@ -142,20 +142,6 @@ func (r *AiRouterBiz) RouteWithSocket(c *websocket.Conn, req *entitys.ChatSockRe
 	}()
 
 	defer func() {
-		if err != nil {
-			_ = entitys.MsgSend(c, entitys.Response{
-
-				Content: err.Error(),
-				Type:    entitys.ResponseErr,
-			})
-		}
-		_ = entitys.MsgSend(c, entitys.Response{
-
-			Content: "",
-			Type:    entitys.ResponseEnd,
-		})
-
-		//r.channelPool.Put(ch)
 		close(ch)
 	}()
 
diff --git a/internal/services/chat.go b/internal/services/chat.go
index a4efe66..e29d82b 100644
--- a/internal/services/chat.go
+++ b/internal/services/chat.go
@@ -74,6 +74,7 @@ func (h *ChatService) Chat(c *websocket.Conn) {
 	h.Gw.AddClient(client)
 	log.Println("client connected:", clientID)
 	log.Println("客户端已连接")
+
 	// 循环读取客户端消息
 	for {
 		messageType, message, err := c.ReadMessage()
@@ -108,8 +109,11 @@ func (h *ChatService) Chat(c *websocket.Conn) {
 				Content: err.Error(),
 				Type:    entitys.ResponseErr,
 			})
-			continue
 		}
+		_ = entitys.MsgSend(c, entitys.Response{
+			Content: "",
+			Type:    entitys.ResponseEnd,
+		})
 	}
 	h.Gw.RemoveClient(clientID)
 	_ = c.Close()