快捷導(dǎo)航

記錄一次websocket封裝的過程

更新時間：2020年11月23日 10:54:35 作者：大明的IT筆記

這篇文章主要介紹了記錄一次websocket封裝的過程，幫助大家更好的理解和封裝websocket，感興趣的朋友可以了解下

在一個應(yīng)用中，websocket一般都是以單例形式存在的，即在整個應(yīng)用中，websocket實例始終保持唯一。但有時我們要用到websocket實例的時候，可能websocket還沒實例化，所以要做成異步的形式來獲取實例。

一、封裝。先創(chuàng)建 socket.ts 文件

import EventEmitter from 'events'; // 這里用到了 events 包
const ee = new EventEmitter();
class Ws {
 private wsUrl: string = '';
 private socket: WebSocket | undefined; // socket實例
 private lockReconnect: boolean = false; // 重連鎖
 private timeout: NodeJS.Timeout | undefined;

 // 初始化socket，一般在應(yīng)用啟動時初始化一次就好了，或者需要更換wsUrl
 public init(wsUrl: string) {
  this.wsUrl = wsUrl;
  this.createWebSocket();
 }

 // 獲取socket實例
 public getInstance(): Promise<WebSocket> {
  return new Promise((resolve, reject) => {
   if (this.socket) {
    resolve(this.socket);
   } else {
    ee.on('socket', (state: string) => {
     if (state === 'success') {
      resolve(this.socket);
     } else {
      reject();
     }
    });
   }
  });
 }

 // 創(chuàng)建socket
 private createWebSocket() {
  try {
   console.log('websocket 開始鏈接');
   const socket = new WebSocket(this.wsUrl);
   socket.addEventListener('close', () => {
    console.log('websocket 鏈接關(guān)閉');
    this.socket = undefined;
    this.reconnect();
   });
   socket.addEventListener('error', () => {
    console.log('websocket 發(fā)生異常了');
    this.socket = undefined;
    this.reconnect();
   });
   socket.addEventListener('open', () => {
    // 可在此進行心跳檢測
    // this.heartCheck.start();
    console.log('websocket open');
    this.socket = socket;
    ee.emit('socket', 'success');
   });
   socket.addEventListener('message', (event) => {
    console.log('websocket 接收到消息', event);
   });
  } catch (e) {
   console.log('socket catch error', e);
   this.reconnect();
  }
 }

 // 重連
 private reconnect() {
  if (this.lockReconnect) {
   return;
  }
  console.log('websocket 正在重新連接');
  this.lockReconnect = true;
  //沒連接上會一直重連，設(shè)置延遲避免請求過多
  this.timeout && clearTimeout(this.timeout);
  this.timeout = setTimeout(() => {
   this.createWebSocket();
   this.lockReconnect = false;
  }, 5000);
 }
}

export default new Ws();

二、引入并使用

import socket from '@/utils/ws';

socket
 .getInstance()
 .then((ws) => {
   // 這里的 ws 就是實例化后的 websocket，可以直接使用 websocket 原生 api
  console.log('getInstance ws', ws);
  ws.addEventListener('message', (event) => {
    console.log('ws 接收到消息', event);
   });
 })
 .catch(() => {});

以上就是記錄一次websocket封裝的過程的詳細(xì)內(nèi)容，更多關(guān)于websocket封裝的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: