hldy_app_mini/common/websocket.js

// common/websocket.js
// 带中文日志与消息体打印选项的 WsRequest（替换你当前文件即可）

class WsRequest {
  constructor(url = '', options = {}) {
    this.url = url || '';
    this.options = Object.assign({
      header: {},
      protocols: [],
      debug: true,
      lang: 'zh',               // 'zh' 或 'en'，控制日志语言（默认中文）
      showMessageBody: true,    // 是否在日志中显示收到消息的完整 JSON（默认 true）
      filterPing: true,         // 是否过滤掉 ping/pong 日志（默认 true）
      heartbeatInterval: 30000,
      heartbeatTimeout: 15000,
      pingMessage: { type: 'ping' },
      parseJSON: true,
      maxReconnectAttempts: 10,
      reconnectDelayBase: 1000,
      autoConnect: false,
      bindGlobal: true
    }, options);

    this.socketTask = null;
    this.connected = false;
    this._msgQueue = [];
    this.reconnectAttempts = 0;
    this._lastPongAt = Date.now();
    this._hbTimer = null;
    this._hbTimeoutTimer = null;
    this._subscriptions = new Map();

    if (this.options.bindGlobal) this._bindGlobalSocketEvents();
    if (this.options.autoConnect) setTimeout(() => this.open(), 0);
  }

  // 内部多语言 Label
  _label(key) {
    const zh = {
      open: '打开连接 ->',
      alreadyConnected: '已连接，忽略 open',
      connectNoTask: 'connectSocket 未返回 task，使用全局回调',
      onOpen: '连接已打开',
      onMessage: '收到消息',
      onClose: '连接已关闭',
      onError: '连接错误',
      globalOnOpen: '全局 onSocketOpen',
      globalOnMessage: '全局 onSocketMessage',
      hbTimeout: '心跳超时',
      reconnectScheduled: '计划重连',
      sendFailed: '发送失败',
      closeError: '关闭错误'
    };
    const en = {
      open: 'open ->',
      alreadyConnected: 'already connected, ignore open',
      connectNoTask: 'connectSocket returned no task (using global callbacks if available)',
      onOpen: 'onOpen',
      onMessage: 'onMessage',
      onClose: 'onClose',
      onError: 'onError',
      globalOnOpen: 'global onSocketOpen',
      globalOnMessage: 'global onSocketMessage',
      hbTimeout: 'hb timeout',
      reconnectScheduled: 'reconnect scheduled',
      sendFailed: 'send failed',
      closeError: 'close error'
    };
    return this.options.lang === 'zh' ? zh[key] || key : en[key] || key;
  }

  // 简单统一日志
  log(...args) {
    if (!this.options.debug) return;
    // 如果是对象，格式化输出以便控制台可读
    const out = args.map(a => {
      if (typeof a === 'object') {
        try { return JSON.stringify(a, null, 2); } catch (e) { return String(a); }
      }
      return String(a);
    }).join(' ');
    // console.log('[WsRequest]', out);
  }

  // ---------- 对外控制 ----------
  open() {
    if (!this.url) { this.log(this._label('open'), this._label('noUrl') || 'no url'); return; }
    if (this.connected) { this.log(this._label('alreadyConnected')); return; }
    this.log(this._label('open'), this.url);
    try {
      const task = uni.connectSocket({ url: this.url, header: this.options.header, protocols: this.options.protocols });
      if (task && typeof task.onOpen === 'function') {
        this._bindSocketTask(task);
      } else {
        this.log(this._label('connectNoTask'));
        // 依赖全局回调，_bindGlobalSocketEvents 已经绑定则会处理 onOpen/onMessage
      }
    } catch (e) {
      console.error(e);
    }
  }

  close(code = 1000, reason = 'client close') {
    this._stopHeartbeat();
    try {
      if (this.socketTask && typeof this.socketTask.close === 'function') {
        this.socketTask.close({ code, reason });
      } else if (typeof uni.closeSocket === 'function') {
        uni.closeSocket();
      }
    } catch (e) {
      this.log(this._label('closeError'), e);
    }
    this.connected = false;
  }

  send(payload) {
    const data = typeof payload === 'string' ? payload : JSON.stringify(payload);
    if (this.socketTask && typeof this.socketTask.send === 'function' && this.connected) {
      try { this.socketTask.send({ data }); return; } catch (e) { this.log(this._label('sendFailed'), e); this._msgQueue.push(data); }
    } else {
      try { uni.sendSocketMessage({ data }); return; } catch (e) { this._msgQueue.push(data); }
    }
  }

  _flushQueue() {
    if (!this.connected) return;
    while (this._msgQueue.length) {
      const d = this._msgQueue.shift();
      try {
        if (this.socketTask && typeof this.socketTask.send === 'function') {
          this.socketTask.send({ data: d });
        } else {
          uni.sendSocketMessage({ data: d });
        }
      } catch (e) { this._msgQueue.unshift(d); break; }
    }
  }

  _bindSocketTask(task) {
    this.socketTask = task;
    if (task.__ws_bound__) return;
    task.__ws_bound__ = true;

    task.onOpen(res => {
      this.log(this._label('onOpen'), res);
      this.connected = true;
      this.reconnectAttempts = 0;
      this._lastPongAt = Date.now();
      this._startHeartbeat();
      this._flushQueue();
      this.options.onOpen && this.options.onOpen(res);
    });

    task.onMessage(msg => {
      this._handleIncoming(msg);
    });

    task.onClose(res => {
      this.log(this._label('onClose'), res);
      this.connected = false;
      this._stopHeartbeat();
      this.options.onClose && this.options.onClose(res);
      this._tryReconnect();
    });

    task.onError(err => {
      this.log(this._label('onError'), err);
      this.connected = false;
      this._stopHeartbeat();
      this.options.onError && this.options.onError(err);
      this._tryReconnect();
    });
  }

  // 处理收到的消息（task 或 全局都使用）
  _handleIncoming(msg) {
    this._lastPongAt = Date.now();
    let data = msg && msg.data;
    if (this.options.parseJSON) {
      try { data = JSON.parse(msg.data); } catch (e) { /* keep raw */ }
    }
    const type = (data && data.type) || '__default__';

    // 过滤 ping/pong （若你想看 ping/pong，把 filterPing 设为 false）
    const isPing = data && (data.type === 'ping' || data.type === 'pong' || (typeof data === 'string' && (data === 'ping' || data === 'pong')));
    if (isPing && this.options.filterPing) {
      // 仍更新 lastPong，但不打印消息体，除非 showMessageBody 强制 true
      if (this.options.debug && this.options.showMessageBody === false) {
        this.log(this._label('onMessage'), type);
      }
    } else {
      // 打印消息类型与（可选）格式化消息体
      if (this.options.debug) {
        if (this.options.showMessageBody) {
          let bodyStr;
          try { bodyStr = JSON.stringify(data, null, 2); } catch (e) { bodyStr = String(data); }
          this.log(this.options.lang === 'zh' ? `${this._label('onMessage')} (${type}):` : `${this._label('onMessage')} (${type}):`, '\n' + bodyStr);
        } else {
          this.log(this._label('onMessage'), type);
        }
      }
    }

    // 外部回调与订阅异步派发
    if (this.options.onMessage) setTimeout(() => this.options.onMessage(data, msg), 0);
    setTimeout(() => this._dispatch(type, data, msg), 0);
  }

  _bindGlobalSocketEvents() {
    // 解绑默认全局再绑定，避免重复
    try { uni.offSocketOpen && uni.offSocketOpen(); } catch (e) {}
    try { uni.offSocketMessage && uni.offSocketMessage(); } catch (e) {}
    try { uni.offSocketClose && uni.offSocketClose(); } catch (e) {}
    try { uni.offSocketError && uni.offSocketError(); } catch (e) {}

    try {
      uni.onSocketOpen(res => {
        this.log(this._label('globalOnOpen'), res);
        this.connected = true;
        this.reconnectAttempts = 0;
        this._lastPongAt = Date.now();
        this._startHeartbeat();
        this._flushQueue();
        this.options.onOpen && this.options.onOpen(res);
      });
    } catch (e) {}

    try {
      uni.onSocketMessage(msg => {
        // 全局收到消息也走统一处理
        if (this.options.debug) this.log(this._label('globalOnMessage'));
        this._handleIncoming(msg);
      });
    } catch (e) {}

    try {
      uni.onSocketClose(res => {
        this.log(this._label('globalOnClose'), res);
        this.connected = false;
        this._stopHeartbeat();
        this.options.onClose && this.options.onClose(res);
        this._tryReconnect();
      });
    } catch (e) {}

    try {
      uni.onSocketError(err => {
        this.log(this._label('globalOnError'), err);
        this.connected = false;
        this._stopHeartbeat();
        this.options.onError && this.options.onError(err);
        this._tryReconnect();
      });
    } catch (e) {}
  }

  _startHeartbeat() {
    this._stopHeartbeat();
    if (!this.options.heartbeatInterval) return;
    const pingData = typeof this.options.pingMessage === 'string' ? this.options.pingMessage : JSON.stringify(this.options.pingMessage);
    this._hbTimer = setInterval(() => {
      if (!this.connected) return;
      try { this.send(pingData); } catch (e) { this.log('hb send err', e); }
      clearTimeout(this._hbTimeoutTimer);
      this._hbTimeoutTimer = setTimeout(() => {
        const since = Date.now() - this._lastPongAt;
        if (since > this.options.heartbeatTimeout) {
          this.log(this._label('hbTimeout'), since);
          try { this.close(); } catch (e) { this.log('hb close fail', e); }
        }
      }, this.options.heartbeatTimeout);
    }, this.options.heartbeatInterval);
  }

  _stopHeartbeat() {
    if (this._hbTimer) { clearInterval(this._hbTimer); this._hbTimer = null; }
    if (this._hbTimeoutTimer) { clearTimeout(this._hbTimeoutTimer); this._hbTimeoutTimer = null; }
  }

  _tryReconnect() {
    if (this.reconnectAttempts >= this.options.maxReconnectAttempts) { this.log(this._label('reconnectScheduled'), 'exhausted'); return; }
    this.reconnectAttempts++;
    const delay = Math.min(this.options.reconnectDelayBase * Math.pow(1.5, this.reconnectAttempts - 1), 30000);
    this.log(this._label('reconnectScheduled'), delay + 'ms', 'attempt', this.reconnectAttempts);
    setTimeout(() => {
      this.socketTask = null;
      this.connectIfForeground();
    }, delay);
  }

  connectIfForeground() {
    // 如果你在项目中有前台检测，这里可改为判断前台才 open()
    this.open();
  }

  subscribe(type, handler) {
    if (!this._subscriptions) this._subscriptions = new Map();
    if (!this._subscriptions.has(type)) this._subscriptions.set(type, new Set());
    this._subscriptions.get(type).add(handler);
    return () => this.unsubscribe(type, handler);
  }

  unsubscribe(type, handler) {
    if (!this._subscriptions) return;
    const s = this._subscriptions.get(type); if (!s) return; s.delete(handler);
    if (s.size === 0) this._subscriptions.delete(type);
  }

  _dispatch(type, data, raw) {
    if (!this._subscriptions) return;
    const handlers = this._subscriptions.get(type);
    if (handlers) for (const h of Array.from(handlers)) try { h(data, raw); } catch (e) { this.log('handler err', e); }
    const def = this._subscriptions.get('__default__');
    if (def) for (const h of Array.from(def)) try { h(data, raw); } catch (e) { this.log('default handler err', e); }
  }
}

export default WsRequest;