const WebSocket = require("ws");

// 创建 WebSocket 服务器实例
const wss = new WebSocket.Server({ port: 8080 });

// 客户端超时时间（与服务端要求的 5 秒一致）
const CLIENT_TIMEOUT = 10000; // 5 秒

// 客户端连接池（用于记录所有活跃连接）
const clients = new Map();

wss.on("connection", (ws) => {
  console.log("新客户端连接");

  // 初始化客户端心跳检测
  let timeoutTimer = null;

  // 为每个客户端创建唯一标识（可以用其他方式生成）
  const clientId = Date.now().toString();
  clients.set(clientId, ws);

  // 心跳检测方法
  const heartbeat = () => {
    // 每次收到消息时重置定时器
    if (timeoutTimer) clearTimeout(timeoutTimer);
    // 设置新的超时检测
    timeoutTimer = setTimeout(() => {
      console.log(`客户端 ${clientId} 心跳超时，关闭连接`);
      // ws.close(); // 主动关闭连接
      clients.delete(clientId); // 从连接池移除
    }, CLIENT_TIMEOUT);
  };

  // 初始检测
  heartbeat();

  // 收到消息时触发
  ws.on("message", (message) => {
    console.log(`收到来自 ${clientId} 的消息: ${message}`);

    // 重置心跳检测
    heartbeat();

    // // 如果需要响应心跳（示例代码）
    // if (message === "heartbeat") {
    //   ws.send("heartbeat-ack"); // 发送心跳确认
    // }
  });

  // 连接关闭时触发
  ws.on("close", () => {
    console.log(`客户端 ${clientId} 连接关闭`);
    if (timeoutTimer) clearTimeout(timeoutTimer);
    clients.delete(clientId); // 清理连接池
  });

  // 连接错误时触发
  ws.on("error", (error) => {
    console.error(`客户端 ${clientId} 发生错误:`, error);
    if (timeoutTimer) clearTimeout(timeoutTimer);
    clients.delete(clientId); // 清理连接池
  });
});

console.log("WebSocket 服务端已启动，监听端口 8080");

// 示例：当客户端超时时，可以执行自定义逻辑
function onClientTimeout(clientId) {
  console.log(`[自定义处理] 客户端 ${clientId} 已断开`);
  // 这里可以添加你的业务逻辑，例如：
  // 1. 通知其他客户端
  // 2. 更新数据库状态
  // 3. 触发重新连接等
}

// 服务器关闭时的清理
process.on("SIGINT", () => {
  console.log("\n正在关闭服务器...");
  wss.close(() => {
    console.log("服务器已关闭");
    process.exit();
  });
});