From fa7e908a0869f79da95c6973c6e3d8eb3b137149 Mon Sep 17 00:00:00 2001
From: Seven <1530011482@qq.com>
Date: Tue, 29 Sep 2026 14:13:11 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84websocket=E5=85=B3=E9=97=AD?=
 =?UTF-8?q?=E8=BF=9E=E6=8E=A5=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DESIGN.md                      | 2 +-
 websocket/connectionManager.js | 8 ++++++--
 websocket/index.js             | 2 +-
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/DESIGN.md b/DESIGN.md
index 092d05a..21f2e08 100644
--- a/DESIGN.md
+++ b/DESIGN.md
@@ -190,7 +190,7 @@ WebSocket 挂载在 HTTP Server 上，客户端连接地址的形式为 `ws://<h
 5. **发送确认早于持久化结果**：`sendSuccess` 和接收方推送发生在异步 `db.query` 回调之前。应将成功确认和实时推送放到成功回调中；失败时只返回错误，不推送为已发送消息。（已解决）
 6. **连接管理器广播引用未定义的 `WebSocket`**：`connectionManager.js` 的 `broadcast` 检查 `WebSocket.OPEN`，但文件中没有导入 `ws`。当广播函数被调用时可能抛出 `ReferenceError`。应在此模块引入库常量，或由连接对象状态采用明确且可用的检查方式。（已解决）
 7. **ACK 参数调用不一致**：消息分发通过 `handleAck(data)` 调用，而函数签名是 `handleAck(receiverName, data)`；函数内部随后读取第二个参数，可能因 `data` 未定义而失败。应统一函数参数并以当前连接用户名作为 ACK 接收者身份。(已解决)
-8. **重复登录连接的关闭竞态**：同一用户名的新连接会覆盖 Map 中旧连接；旧连接关闭时无条件按用户名删除，可能把新连接也从 Map 移除。移除连接时应确认 Map 中仍是即将关闭的那个 WebSocket。
+8. **重复登录连接的关闭竞态**：同一用户名的新连接会覆盖 Map 中旧连接；旧连接关闭时无条件按用户名删除，可能把新连接也从 Map 移除。移除连接时应确认 Map 中仍是即将关闭的那个 WebSocket。（已解决）
 9. **分页和输入校验不足**：`contactMessagesInfo` 未验证 `contactName`、`limit`、`offset` 的类型与范围，也未约束最大页大小。建议验证请求参数并设置默认值和上限。
 10. **在线状态及可靠性受单进程限制**：Map 不支持多实例共享、服务重启恢复、离线消息投递或持久化 ACK 状态。若部署多实例，需要共享在线状态/消息协调机制；若要求可靠消息，应定义消息状态、幂等键、重试及离线投递策略。
 11. **配置与错误响应尚未统一**：数据库连接配置没有从 `.env` 读取；HTTP 接口混用 HTTP 状态码和响应体 `code`，空结果有时返回业务码 401。建议集中配置并统一 API 错误语义及日志策略。
diff --git a/websocket/connectionManager.js b/websocket/connectionManager.js
index 056727b..a627b59 100644
--- a/websocket/connectionManager.js
+++ b/websocket/connectionManager.js
@@ -16,9 +16,13 @@ function getConnection(username) {
 }
 
 // 删除用户连接
-function removeConnection(username) {
-  onlineUsers.delete(username);
+function removeConnection(username, ws) {
+  const currentWs = onlineUsers.get(username);
 
+  // 只有 Map 里面存的还是这个连接，才允许删除
+  if (currentWs === ws) {
+    onlineUsers.delete(username);
+  }
   console.log(`用户 ${username} 已断开`);
   console.log("当前在线用户：", [...onlineUsers.keys()]);
 }
diff --git a/websocket/index.js b/websocket/index.js
index 46f9ed1..09e0ac0 100644
--- a/websocket/index.js
+++ b/websocket/index.js
@@ -68,7 +68,7 @@ function initWebSocket(server) {
 
     // 连接关闭
     ws.on("close", () => {
-      removeConnection(username);
+      removeConnection(username, ws);
 
       // 通知所有在线用户当前在线用户列表
       broadcast({