【Web】WebSocket 封装

WebSocketReconnect 工具类:

class WebSocketReconnect {
  private url: string;
  private socket: WebSocket | null = null;
  private reconnectCount = 0;
  private reconnectTimeout: NodeJS.Timeout | null = null;
  private startTime: number | null = null;
  private listerEvent: Map<string, Function[]> = new Map();
  private manuallyClosed = false;

  constructor(
    url: string,
    private maxReconnectAttempts = 3,
    private reconnectInterval = 20000,
    private maxReconnectTime = 180000
  ) {
    if (!this.isValidWebSocketUrl(url)) {
      throw new Error("Invalid WebSocket URL. Please check the URL format.");
    }
    this.url = url;
    this.connect();
  }

  /**
   * 检查 WebSocket URL 是否有效
   * @param url WebSocket URL
   * @returns 是否有效
   */
  private isValidWebSocketUrl(url: string = this.url): boolean {
    return /^wss?:\/\/.+/i.test(url);
  }

  /**
   * 建立 WebSocket 连接
   */
  private connect(): void {
    console.log("Connecting to WebSocket...");

    this.manuallyClosed = false; // 重置手动关闭标志位
    this.socket = new WebSocket(this.url);

    // 绑定事件
    this.socket.onopen = () => {
      console.log("WebSocket Connection Opened!");
      this.triggerEvent("open");
      this.clearReconnectTimeout();
      this.reconnectCount = 0;
    };

    this.socket.onmessage = (event: MessageEvent) => {
      console.log("WebSocket Message Received:", event.data);
      this.triggerEvent("message", event.data);
    };

    this.socket.onclose = (event: CloseEvent) => {
      console.log("WebSocket Connection Closed:", event);
      this.triggerEvent("close", event);
      if (!this.manuallyClosed) {
        this.handleReconnect();
      }
    };

    this.socket.onerror = (error: Event) => {
      console.error("WebSocket Error Occurred:", error);
      this.triggerEvent("error", error);
      if (!this.manuallyClosed) {
        this.handleReconnect();
      }
    };
  }

  /**
   * 处理断线重连逻辑
   */
  private handleReconnect(): void {
    if (
      this.reconnectCount < this.maxReconnectAttempts &&
      (this.startTime === null || Date.now() - this.startTime < this.maxReconnectTime)
    ) {
      this.reconnectCount++;
      console.log(
        `Attempting to reconnect (${this.reconnectCount}/${this.maxReconnectAttempts})...`
      );

      if (!this.startTime) {
        this.startTime = Date.now();
      }

      this.reconnectTimeout = setTimeout(() => {
        this.connect();
      }, this.reconnectInterval);
    } else {
      console.log("Max reconnect attempts or time exceeded. Giving up.");
      this.reconnectCount = 0;
      this.startTime = null;
    }
  }

  /**
   * 清除重连定时器
   */
  private clearReconnectTimeout(): void {
    if (this.reconnectTimeout) {
      clearTimeout(this.reconnectTimeout);
      this.reconnectTimeout = null;
    }
  }

  /**
   * 触发注册的事件
   * @param type 事件类型
   * @param args 事件参数
   */
  private triggerEvent(type: string, ...args: any[]): void {
    const callbacks = this.listerEvent.get(type) || [];
    callbacks.forEach((callback) => callback(...args));
  }

  /**
   * 添加事件监听器
   * @param type 事件类型
   * @param callback 回调函数
   */
  public on(type: string, callback: Function): void {
    if (!this.listerEvent.has(type)) {
      this.listerEvent.set(type, []);
    }
    this.listerEvent.get(type)!.push(callback);
  }

  /**
   * 发送消息
   * @param message 要发送的消息
   */
  public send(message: string | ArrayBuffer | Blob): void {
    if (this.socket && this.socket.readyState === WebSocket.OPEN) {
      this.socket.send(message);
    } else {
      console.error("WebSocket is not open. Unable to send message.");
    }
  }

  /**
   * 手动关闭 WebSocket 连接
   */
  public close(): void {
    this.manuallyClosed = true;
    this.clearReconnectTimeout();
    if (this.socket) {
      this.socket.close();
    }
    this.reconnectCount = 0;
    this.startTime = null;
  }

  /**
   * 查询 WebSocket 当前状态
   * @returns WebSocket 状态
   */
  public getState(): string {
    if (!this.socket) {
      return "DISCONNECTED";
    }
    switch (this.socket.readyState) {
      case WebSocket.CONNECTING:
        return "CONNECTING";
      case WebSocket.OPEN:
        return "OPEN";
      case WebSocket.CLOSING:
        return "CLOSING";
      case WebSocket.CLOSED:
        return "CLOSED";
      default:
        return "UNKNOWN";
    }
  }

  /**
   * 动态更新 WebSocket URL 并重新连接
   * @param newUrl 新的 WebSocket URL
   */
  public updateUrl(newUrl: string): void {
    if (!this.isValidWebSocketUrl(newUrl)) {
      throw new Error("Invalid WebSocket URL.");
    }
    this.close();
    this.url = newUrl;
    this.connect();
  }
}

export default WebSocketReconnect;

使用示例:

const ws = new WebSocketReconnect("wss://example.com", 5, 5000, 300000);

ws.on("open", () => console.log("WebSocket connected!"));
ws.on("message", (msg) => console.log("Received message:", msg));
ws.on("close", () => console.log("WebSocket closed."));
ws.on("error", (err) => console.error("WebSocket error:", err));

ws.send("Hello, server!");
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,923评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,154评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,775评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,960评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,976评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,972评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,893评论 3 416
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,709评论 0 271
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,159评论 1 308
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,400评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,552评论 1 346
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,265评论 5 341
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,876评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,528评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,701评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,552评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,451评论 2 352

推荐阅读更多精彩内容