@sybz-components/utils 0.0.12 → 1.0.0

package/dist/ws.mjs

@@ -1,169 +0,0 @@
-import qs from 'qs';
-
-const reconnectMaxCount = 3;
-const message = "ping";
-const interval = 3e3;
-const timeout = 1e3;
-class WS {
-  url;
-  socket = null;
-  reconnectCount = 0;
-  delay = null;
-  timer = null;
-  autoReconnect;
-  heartbeat;
-  query;
-  /**
-   * 创建并立即连接一个 WebSocket 实例。
-   *
-   * @param url WebSocket 地址，例如 `wss://example.com/ws`。
-   * @param options 连接配置，支持重连、心跳和 query 参数。
-   *
-   * @example
-   * const ws = new WS('wss://example.com/ws', {
-   *   autoReconnect: { reconnectMaxCount: 5 },
-   *   heartbeat: { message: 'ping', interval: 5000 },
-   *   query: { token: 'abc123' },
-   * })
-   */
-  constructor(url, options) {
-    const { autoReconnect = true, query = {}, heartbeat = false } = options || {};
-    this.autoReconnect = autoReconnect;
-    this.heartbeat = heartbeat;
-    this.query = query;
-    this.url = `${url}` + qs.stringify({ ...this.query }, { addQueryPrefix: true });
-    this.connect();
-  }
-  /**
-   * 主动发起连接。
-   *
-   * 调用时会先关闭旧连接，再创建新的 WebSocket 实例。
-   *
-   * @example
-   * const ws = new WS('wss://example.com/ws')
-   * ws.connect()
-   */
-  connect() {
-    this.close();
-    this.socket = new WebSocket(this.url);
-    this.onError();
-    this.onOpen();
-  }
-  /**
-   * 注册连接成功后的处理逻辑，并在需要时开启心跳。
-   *
-   * @example
-   * const ws = new WS('wss://example.com/ws')
-   * ws.onOpen()
-   */
-  onOpen() {
-    if (this.socket) {
-      this.socket.onopen = () => {
-        this.send("ping");
-        this.heartbeat && this.startHeartbeat();
-      };
-    }
-  }
-  /**
-   * 开启心跳发送。
-   *
-   * @example
-   * const ws = new WS('wss://example.com/ws', {
-   *   heartbeat: { message: 'ping', interval: 3000 },
-   * })
-   * ws.startHeartbeat()
-   */
-  startHeartbeat() {
-    const msg = this.heartbeat?.message || message;
-    const int = this.heartbeat?.interval || interval;
-    this.timer = setInterval(() => {
-      this.send(msg);
-    }, int);
-  }
-  /**
-   * 注册错误处理和自动重连逻辑。
-   *
-   * @example
-   * const ws = new WS('wss://example.com/ws', { autoReconnect: true })
-   * ws.onError()
-   */
-  onError() {
-    if (this.socket) {
-      this.socket.onerror = () => {
-        const count = this.autoReconnect?.reconnectMaxCount || reconnectMaxCount;
-        if (this.autoReconnect && this.reconnectCount < count) {
-          this.reconnectCount++;
-          this.connect();
-        }
-      };
-    }
-  }
-  /**
-   * 关闭连接并清理相关定时器。
-   *
-   * @example
-   * ws.close()
-   */
-  close() {
-    this.socket && this.socket.close();
-    this.delay && clearTimeout(this.delay);
-    this.timer && clearInterval(this.timer);
-    this.socket = null;
-  }
-  /**
-   * 监听服务端消息。
-   *
-   * 内部会优先尝试将消息解析为 JSON；解析失败时会回传原始 `MessageEvent`。
-   *
-   * @param callback 接收到消息时触发的回调。
-   *
-   * @example
-   * ws.onMessage((message) => {
-   *   console.log(message)
-   * })
-   */
-  onMessage(callback) {
-    if (this.socket) {
-      this.socket.onmessage = (data) => {
-        try {
-          const res = JSON.parse(data.data);
-          callback(res);
-        } catch {
-          callback(data);
-        }
-      };
-    }
-  }
-  /**
-   * 发送消息。
-   *
-   * - 如果连接已就绪，会立即发送
-   * - 如果正在连接，会延迟发送
-   * - 如果连接已关闭，会先重新连接再发送
-   *
-   * @param data 需要发送的消息内容。
-   *
-   * @example
-   * ws.send('hello')
-   *
-   * @example
-   * ws.send(JSON.stringify({ type: 'ping' }))
-   */
-  send(data) {
-    if (!this.socket) return;
-    if (this.socket.readyState === this.socket.OPEN) {
-      this.socket.send(JSON.stringify(data));
-    } else if (this.socket.readyState === this.socket.CONNECTING) {
-      this.delay = setTimeout(() => {
-        this.socket?.send(data);
-      }, timeout);
-    } else {
-      this.connect();
-      this.delay = setTimeout(() => {
-        this.socket?.send(data);
-      }, timeout);
-    }
-  }
-}
-
-export { WS };