#WebSocket 心跳机制与断线重连

class ResilientWebSocket {
  constructor(url) {
    this.url = url;
    this.ws = null;
    
    // 断线重连相关配置
    this.isReconnecting = false;  // 防止重连潮汐攻击的锁
    this.reconnectTimer = null;
    this.reconnectAttempts = 0;   // 记录已重连死磕了多少次
    this.maxReconnects = 5;
    
    // 心跳设置配置
    this.pingTimeout = 30000;     // 30s 的心跳安详探视期
    this.pongTimeout = 10000;     // 10s 未收到答复死亡绝斩期
    this.pingTimer = null;
    this.pongTimer = null;

    this.init();
  }

  // 装载实体生命轮回
  init() {
    this.ws = new WebSocket(this.url);

    this.ws.onopen = () => {
      // 一旦顺利搭上桥：清空重试的满身疮痍与锁，且开启心跳心脏仪
      this.reconnectAttempts = 0;
      this.isReconnecting = false;
      this.resetHeartbeat();
      console.log('🔗 WebSocket 连接满血建立！');
    };

    this.ws.onmessage = (event) => {
      // 【核心一】：不管收到的是正常的长消息还是心跳的反馈，都代表对方活着！续命心脏机制。
      this.resetHeartbeat();
      
      if (event.data === 'pong') {
        // 这是心跳的回复，不需要交给真实业务去画屏幕，悄悄消化掉即可
        return;
      }
      
      // ...正式交给外部调用着做 IM 刷新屏幕展示等
    };

    // 监听网络由于 Nginx 强断或者拔除网线
    this.ws.onclose = () => this.reconnect();
    this.ws.onerror = () => this.reconnect();
  }

  // --- 心跳续命引擎 ---
  resetHeartbeat() {
    clearTimeout(this.pingTimer);
    clearTimeout(this.pongTimer);
    
    this.pingTimer = setTimeout(() => {
      if (this.ws.readyState === WebSocket.OPEN) {
        // 探病：发送大吼一声
        this.ws.send('ping');
        
        // 【核心二】：挂上延时炸弹定时雷！过了等待线必死。
        this.pongTimer = setTimeout(() => {
           console.warn('⚠️ 10秒苦等无回应，判定服务端陷入假死状态。强行掐死！');
           // 使用原生方法直接暴力捏死现在的躯壳，会触发后文的 onclose 并联动重连
           this.ws.close(); 
        }, this.pongTimeout);
      }
    }, this.pingTimeout);
  }

  // --- 再度转生引擎 ---
  reconnect() {
    if (this.isReconnecting || this.reconnectAttempts >= this.maxReconnects) {
       console.error('❌ 重连次数已耗尽或者已经在拼命挣扎的路上了，放弃拯救。');
       return;
    }
    this.isReconnecting = true;
    this.reconnectAttempts++;

    // 用指数慢慢后退法来延时抢救时间
    const delay = Math.min(3000 * this.reconnectAttempts, 10000); 
    console.log(`📡 正在准备第 ${this.reconnectAttempts} 次抢救重连，${delay/1000}秒后起搏...`);
    
    this.reconnectTimer = setTimeout(() => {
      this.isReconnecting = false;
      this.init(); // 再次借尸还魂呼叫创建！
    }, delay);
  }
}