ai生成设计书文档，并解决文档中指出的部分bug

websocket/index.js

@@ -138,38 +138,35 @@ function handleSendMessage(senderName, data) {
       }
       return;
     }
-  });
+    // 数据库确认写入后，才确认发送并推送给接收者
+    const senderWs = getConnection(senderName);
+    if (senderWs) {
+      senderWs.send(
+        JSON.stringify({
+          type: "sendSuccess",
+          messageId: messageId
+        })
+      );
+    }
 
-  // 告诉发送者：服务器已经保存成功
-  const senderWs = getConnection(senderName);
-  if (senderWs) {
-    senderWs.send(
+    const receiverWs = getConnection(receiverName);
+    if (!receiverWs) {
+      console.log(`${receiverName} 当前不在线`);
+      return;
+    }
+
+    receiverWs.send(
       JSON.stringify({
-        type: "sendSuccess",
-        messageId: messageId
+        type: "newMessage",
+        message: {
+          messageId,
+          senderName,
+          receiverName,
+          content
+        },
       })
     );
-  }
-
-  // 找到接收者的WebSocket
-  const receiverWs = getConnection(receiverName);
-  if (!receiverWs) {
-    console.log(`${receiverName} 当前不在线`);
-    return;
-  }
-
-  // 推送消息
-  receiverWs.send(
-    JSON.stringify({
-      type: "newMessage",
-      message: {
-        messageId,
-        senderName,
-        receiverName,
-        content
-      },
-    })
-  );
+  });
 }
 
 function handleAck(receiverName, data) {