增加ack确认消息机制

websocket/index.js

@@ -1,5 +1,6 @@
 const WebSocket = require("ws");
 const url = require("url");
+const db = require("../db/mysql");
 
 const { verifyToken } = require("../utils/jwt");
 
@@ -24,27 +25,22 @@ function initWebSocket(server) {
 
     if (!token) {
       console.log("没有提供token");
-
       ws.close();
       return;
     }
 
     // 验证JWT
     let user;
-
     try {
       user = verifyToken(token);
     } catch (error) {
       console.log("JWT验证失败：", error.message);
-
       ws.close();
       return;
     }
-
     console.log("WebSocket用户：", user);
 
     const username = user.username;
-
     // 保存连接
     addConnection(username, ws);
 
@@ -52,7 +48,6 @@ function initWebSocket(server) {
     ws.on("message", (message) => {
       try {
         const data = JSON.parse(message.toString());
-
         console.log("收到WebSocket消息：", data);
 
         handleMessage(username, data);
@@ -83,6 +78,9 @@ function handleMessage(username, data) {
     case "sendMessage":
       handleSendMessage(username, data);
       break;
+    case "ack":
+      handleAck(data);
+      break;
 
     default:
       console.log("未知消息类型：", data.type);
@@ -91,6 +89,7 @@ function handleMessage(username, data) {
 
 // 处理发送消息
 function handleSendMessage(senderName, data) {
+  const messageId = data.messageId;
   const receiverName = data.receiverName;
   const content = data.content;
 
@@ -98,9 +97,49 @@ function handleSendMessage(senderName, data) {
     `${senderName} → ${receiverName}：${content}`
   );
 
+  const sql = `
+        INSERT INTO messages
+        (message_id, sender_name, receiver_name, content)
+        VALUES (?, ?, ?, ?)
+    `;
+
+  db.query(sql, [
+    messageId,
+    senderName,
+    receiverName,
+    content
+  ], (error, result) => {
+    if (error) {
+      console.error(error);
+
+      const senderWs = getConnection(senderName);
+      if (senderWs) {
+        senderWs.send(
+          JSON.stringify({
+            type: "sendError",
+            messageId: messageId,
+            message: "消息保存失败"
+          })
+        );
+
+      }
+      return;
+    }
+  });
+
+  // 告诉发送者：服务器已经保存成功
+  const senderWs = getConnection(senderName);
+  if (senderWs) {
+    senderWs.send(
+      JSON.stringify({
+        type: "sendSuccess",
+        messageId: messageId
+      })
+    );
+  }
+
   // 找到接收者的WebSocket
   const receiverWs = getConnection(receiverName);
-
   if (!receiverWs) {
     console.log(`${receiverName} 当前不在线`);
     return;
@@ -111,12 +150,46 @@ function handleSendMessage(senderName, data) {
     JSON.stringify({
       type: "newMessage",
       message: {
+        messageId,
         senderName,
         receiverName,
-        content,
+        content
       },
     })
   );
 }
 
+function handleAck(receiverName, data) {
+  const messageId = data.messageId;
+  const senderName = data.senderName;
+
+  console.log(
+    `${receiverName} 收到消息 ${messageId}，发送 ACK`
+  );
+
+  // 找到原发送者
+  const senderWs = getConnection(senderName);
+
+  if (!senderWs) {
+    console.log(
+      `${senderName} 当前不在线`
+    );
+    return;
+  }
+
+  // 把 ACK 转给原发送者
+  senderWs.send(
+    JSON.stringify({
+      type: "ack",
+      messageId:
+        messageId
+    })
+  );
+
+  console.log(
+    `ACK ${messageId} 已发送给 ${senderName}`
+  );
+}
+
+
 module.exports = initWebSocket;