WebSocket封装成工具类,实现心跳监测、断开重连

1. 工具类代码

javascript 复制代码
import { Message } from 'element-ui';

export class WebSocketManager {
  constructor(url, reconnectInterval = 5000, onMessageCallback) {
    this.url = url;
    this.reconnectInterval = reconnectInterval; // 重连等待时间(毫秒)
    this.onMessageCallback = onMessageCallback; // 接收外部回调函数
    this.webSocket = null;
    this.reconnectAttempts = 0; // 重连尝试次数
    this.reconnectTimeout = null; // 重连定时器
    this.heartbeatInterval = null; // 心跳定时器
    this.heartbeatTimeout = 30000; // 心跳超时时间(毫秒)

    this.connect(); // 初始化连接
  }

  connect() {
    if (typeof WebSocket === 'undefined') {
      Message.warning('您的浏览器不支持 WebSocket,无法获取数据');
      return false;
    }

    this.webSocket = new WebSocket(this.url);
    this.webSocket.onopen = this.onWebSocketOpen.bind(this);
    this.webSocket.onmessage = this.onWebSocketMessage.bind(this);
    this.webSocket.onerror = this.onWebSocketError.bind(this);
    this.webSocket.onclose = this.onWebSocketClose.bind(this);
  }

  sendHeartbeat() {
    if (this.webSocket && this.webSocket.readyState === WebSocket.OPEN) {
      // 跟后端协商数据格式,发送心跳消息
      const obj = {
        toUserId: "-1",
        text: "[{}]"
      };
      const jsonString = JSON.stringify(obj);
      // 发送心跳消息
      this.webSocket.send(jsonString);

      // 设置心跳超时检查
      this.heartbeatTimeout = setTimeout(() => {
        if (this.webSocket && this.webSocket.readyState === WebSocket.OPEN) {
          this.webSocket.close(1000, '心跳超时,主动断开连接');
          this.reconnect(); // 心跳超时后尝试重连
        }
      }, this.heartbeatTimeout);
    }
  }

  onWebSocketOpen(event) {
    console.log('WebSocket 已连接:', event);
    this.reconnectAttempts = 0; // 连接成功后重连尝试次数归零

    // 启动定时发送心跳消息
   // this.heartbeatInterval = setInterval(() => this.sendHeartbeat(), 5000);
  }

  onWebSocketMessage(event) {
    console.log('收到 WebSocket 消息:', event.data);

    // 如果是心跳响应,则清除心跳超时定时器
    if (event.data === 'heartbeat') {
      clearTimeout(this.heartbeatTimeout);
    } else if (event.data === '连接成功') {
      return;
    } else {
      // 将收到的数据传递给回调函数
      if (this.onMessageCallback) {
        this.onMessageCallback(event.data);
      }
    }
  }

  onWebSocketError(event) {
    console.error('WebSocket 发生错误:', event);
  }

  onWebSocketClose(event) {
    console.log('WebSocket 已断开:', event);

    // 清理心跳相关的定时器
    clearInterval(this.heartbeatInterval);
    clearTimeout(this.heartbeatTimeout);

    // 如果重连尝试次数超过限制,则不再尝试重连
    if (this.reconnectAttempts >= 20) {
      console.error(`超过重连尝试次数限制20次,放弃重连。`);
    } else {
      this.reconnect(); // 尝试重连
    }
  }

  reconnect() {
    this.reconnectAttempts++; // 重连尝试次数加一
    console.log(`正在进行第 ${this.reconnectAttempts} 次重连尝试`);
    this.webSocket = null; // 断开现有连接
    this.heartbeatTimeout = null; // 清理心跳超时定时器
    this.heartbeatInterval = null; // 清理心跳定时器

    // 等待一段时间后重新连接
    this.reconnectTimeout = setTimeout(() => {
      this.connect();
    }, this.reconnectInterval);
  }

  sendWebSocketMessage(message) {
    if (this.webSocket && this.webSocket.readyState === WebSocket.OPEN) {
      this.webSocket.send(message);
    }
  }

  close() {
    if (this.webSocket) {
      this.webSocket.close(); // 关闭 WebSocket 连接
    }
    clearInterval(this.heartbeatInterval); // 清理心跳定时器
    clearTimeout(this.reconnectTimeout); // 清理重连定时器
    clearTimeout(this.heartbeatTimeout); // 清理心跳超时定时器
  }
}

2. 使用

javascript 复制代码
import { WebSocketManager } from '@/utils/websocket.js'
export default {
  data() {
    return {
      list: [],
      webSocketManager: null // 实时巡检websocket
    }
  },
   mounted() {
  	this.setRealTimeRouteInspection()
  },
  beforeDestroy() {
    // 组件销毁时关闭 WebSocket 连接
    if (this.webSocketManager) {
      this.webSocketManager.close()
    }
  },
  methods: {
  	setRealTimeRouteInspection() {
      if (!this.list.length) return
      const newOnFlyList = this.list.filter((item) => item.xjstatus == '1') // 筛选出执行中的工单
      const ids = newOnFlyList.map((item) => item.tid)
      // 获取实时巡检路线
      // window.DSE.realTrajectoryServerPath换成你自己的ip地址,ids[0]换成你自己的参数
      this.webSocketManager = new WebSocketManager(
        `${window.DSE.realTrajectoryServerPath}/zhxj/api/pushMessage/${ids[0]}`,
        5000,
        this.handleWebSocketMessage
      )
    },
    // 获取并处理websocket数据
    handleWebSocketMessage(rawData) {
      try {
        // 解析数据,需要换成你们自己后端定义的数据格式
        const data = JSON.parse(rawData)
        const tackList = JSON.parse(data.text)
        if (!tackList.length) {
          console.log('暂无实时巡检轨迹')
          return
        }
        const realTimeTracklist = tackList.map((el) => [
          Number(el.lgtd),
          Number(el.lttd)
        ])

        console.log('🚀实时巡检轨迹数据:', realTimeTracklist)
       
      } catch (error) {
        throw new Error(error)
      }
    },
  }
相关推荐
PersistJiao20 分钟前
Spark 分布式计算中网络传输和序列化的关系(一)
大数据·网络·spark
活宝小娜1 小时前
vue不刷新浏览器更新页面的方法
前端·javascript·vue.js
程序视点1 小时前
【Vue3新工具】Pinia.js:提升开发效率,更轻量、更高效的状态管理方案!
前端·javascript·vue.js·typescript·vue·ecmascript
coldriversnow1 小时前
在Vue中,vue document.onkeydown 无效
前端·javascript·vue.js
我开心就好o1 小时前
uniapp点左上角返回键, 重复来回跳转的问题 解决方案
前端·javascript·uni-app
刚刚好ā2 小时前
js作用域超全介绍--全局作用域、局部作用、块级作用域
前端·javascript·vue.js·vue
黑客Ash3 小时前
【D01】网络安全概论
网络·安全·web安全·php
->yjy3 小时前
计算机网络(第一章)
网络·计算机网络·php
yqcoder4 小时前
reactflow 中 useNodesState 模块作用
开发语言·前端·javascript
摘星星ʕ•̫͡•ʔ4 小时前
计算机网络 第三章:数据链路层(关于争用期的超详细内容)
网络·计算机网络