const http = require('http');
// 初始化Express應(yīng)用
const app = express();
const server = http.createServer(app);
// 創(chuàng)建WebScoket服務(wù)器
const io = socketIo(server, {
  cors: {
    origin: "http://192.168.1.3:8080", // 你的前端地址
    origin: '*',
    methods: ['GET', 'POST']
  }
});
// ...
server.listen(3000, async () => {
    console.log(`Server is running on port 3000`);
});

const userSocketMap = new Map(); // 用戶ID到socket.id的映射
const userHeartbeats = new Map(); // 用戶心跳檢測

io.on("connection", async (socket) => {
  // 所有連接邏輯在這里處理
});

// 當客戶端發(fā)送 'login' 事件時，觸發(fā)這個回調(diào)函數(shù)
socket.on('login', ({ userId, csId }) => {
  // 參數(shù)驗證：確保傳入的參數(shù)是字符串類型
  userId = String(userId); // 將 userId 轉(zhuǎn)換為字符串，統(tǒng)一類型
  csId = String(csId); // 將 csId 轉(zhuǎn)換為字符串，表示要聊天的客戶id

  // 存儲關(guān)聯(lián)關(guān)系：將用戶信息與當前 socket 連接關(guān)聯(lián)起來
  socket.userId = userId; // 將 userId 存儲到當前 socket 對象中
  socket.csId = csId; // 將 csId 存儲到當前 socket 對象中
  userSocketMap.set(userId, socket.id); // 在 userSocketMap 中存儲 userId 和 socket.id 的映射關(guān)系

  // 加入房間：根據(jù) csId 創(chuàng)建一個房間，用戶加入該房間
  const room = `room-${csId}`; // 使用 csId 構(gòu)造房間名稱
  socket.join(room); // 讓當前用戶加入這個房間

  // 廣播在線狀態(tài)：通知所有客戶端當前用戶的在線狀態(tài)
  io.emit('user_online', userId); // 發(fā)送 'user_online' 事件，通知用戶上線
  io.emit('Online_user', Array.from(userSocketMap.entries())); // 發(fā)送 'Online_user' 事件，包含所有在線用戶的信息
});

// 當客戶端發(fā)送 'all_member' 事件時，觸發(fā)這個回調(diào)函數(shù)
socket.on('all_member', async () => {
  // 根據(jù)當前用戶的 csId 構(gòu)造房間名稱
  const room = `room-${socket.csId}`;

  // 獲取房間內(nèi)所有用戶的 socket 實例
  const sockets = await io.in(room).fetchSockets();  // 使用 io.in(room).fetchSockets() 獲取房間內(nèi)的所有 socket 實例

  // 提取房間內(nèi)所有用戶的 userId
  const users = sockets.map(s => s.userId);  // 從每個 socket 實例中提取 userId，形成一個用戶 ID 數(shù)組

  // 數(shù)據(jù)庫查詢優(yōu)化：查詢房間內(nèi)用戶的詳細信息及未讀消息數(shù)量
  const [results] = await pool.query(`
    SELECT u.id, u.role, u.username,  // 查詢用戶的基本信息：用戶 ID、角色、用戶名
           COUNT(m.id) AS message_count  // 查詢未讀消息的數(shù)量
    FROM users u
    LEFT JOIN messages m ON u.id = m.sender_id  // 關(guān)聯(lián)消息表，找到發(fā)送給當前用戶的消息
      AND m.receiver_id = ?  // 限定消息的接收者是當前用戶
      AND m.read_at IS NULL  // 限定消息未被閱讀
    WHERE u.id IN (?)  // 限定用戶 ID 在房間內(nèi)用戶列表中
    GROUP BY u.id  // 按用戶 ID 分組，確保每個用戶只返回一條記錄
  `, [socket.userId, users]);  // 查詢參數(shù)：當前用戶的 ID 和房間內(nèi)用戶 ID 列表

  // 將查詢結(jié)果發(fā)送回客戶端
  socket.emit('myUsersList', results);  // 發(fā)送 'myUsersList' 事件，將查詢結(jié)果傳遞給客戶端
});

// 當客戶端發(fā)送 'private_message' 事件時，觸發(fā)這個回調(diào)函數(shù)
socket.on("private_message", async (data) => {
  // 獲取接收者的 socket.id
  const receiverSocketId = userSocketMap.get(String(data.receiverId)); // 從 userSocketMap 中根據(jù)接收者的 userId 獲取對應(yīng)的 socket.id

  // 實時消息推送：將消息發(fā)送給接收者
  if (receiverSocketId) { // 如果接收者在線（存在對應(yīng)的 socket.id）
    io.to(receiverSocketId).emit('new_private_message', { // 向接收者的 socket 發(fā)送 'new_private_message' 事件
      senderId: data.senderId, // 發(fā)送者的 ID
      content: data.content, // 消息內(nèi)容
      timestamp: new Date() // 消息發(fā)送的時間戳
    });
  }

  // 消息持久化：將消息存儲到數(shù)據(jù)庫中
  await pool.execute( // 使用數(shù)據(jù)庫連接池執(zhí)行 SQL 插入語句
    'INSERT INTO messages VALUES (?, ?, ?, ?)', // 插入消息到 messages 表
    [data.senderId, data.receiverId, data.content, new Date()] // 插入的值：發(fā)送者 ID、接收者 ID、消息內(nèi)容、消息發(fā)送時間
  );
});

socket.on('disconnect', () => {
  userSocketMap.delete(socket.userId);
  io.emit('user_offline', socket.userId);
  io.emit('update_member_list');
});

// 心跳接收：客戶端發(fā)送心跳信號時，更新用戶的心跳時間
socket.on('heartbeat', () => {
  userHeartbeats.set(socket.userId, Date.now()); // 將當前用戶的心跳時間更新為當前時間戳
});

// 定時檢測：每隔一段時間檢查用戶是否離線
setInterval(() => {
  const now = Date.now(); // 獲取當前時間戳
  for (const [userId, lastTime] of userHeartbeats) { // 遍歷 userHeartbeats 中的每個用戶及其最后心跳時間
    if (now - lastTime > 4000) { // 如果當前時間與最后心跳時間的差值超過 4000 毫秒（4 秒）
      // 清理離線用戶
      userSocketMap.delete(userId); // 從 userSocketMap 中刪除該用戶，表示用戶已離線
      io.emit('user_offline', userId); // 廣播 'user_offline' 事件，通知所有客戶端該用戶已離線
    }
  }
}, 2000); // 每隔 2000 毫秒（2 秒）執(zhí)行一次定時檢測

setInterval(() => {
  console.log('\n當前連接狀態(tài):');
  console.log('用戶映射:', Array.from(userSocketMap.entries()));
  
  io.sockets.forEach(socket => {
    console.log(`SocketID: ${socket.id}, User: ${socket.userId}`);
  });
}, 30000);

// 使用Redis存儲映射關(guān)系
const redisClient = require('redis').createClient();
await redisClient.set(`user:${userId}:socket`, socket.id);

// 大消息分片處理
socket.on('message_chunk', (chunk) => {
  // 重組邏輯...
});

# Nginx配置
location /socket.io/ {
  proxy_set_header Upgrade $http_upgrade;
  proxy_set_header Connection "upgrade";
  proxy_pass http://socket_nodes;
}

npm install @socket.io/redis-adapter

const { createAdapter } = require("@socket.io/redis-adapter");
io.adapter(createAdapter(redisClient, redisClient.duplicate()));