简单封装websocket组件

最新推荐文章于 2025-11-10 21:07:29 发布

原创最新推荐文章于 2025-11-10 21:07:29 发布 · 379 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#javascript #html5 #vue.js #websocket

博客介绍了在项目中使用Websocket的操作，涉及到websocket.js文件，并且说明了在main.js中引入该文件的使用情况，与前端开发技术相关。

websocket.js文件

let Socket = null
let setIntervalWesocketPush = null

/**
 * 建立websocket连接
 * @param {string} url ws地址
 */
export async function createSocket ( url ) {
  // Socket && Socket.close()
  if (!Socket) {
    console.log('建立websocket连接')
    Socket = new WebSocket(url)
    Socket.onopen = onopenWS
    Socket.onmessage = onmessageWS
    Socket.onerror = onerrorWS
    Socket.onclose = oncloseWS
  } else {
    console.log('websocket已连接')
  }
  return Socket
}

/**打开WS之后发送心跳 */
const onopenWS = () => {
  sendPing()
}

/**连接失败*/
const onerrorWS = () => {
  Socket.close()
  clearInterval(setIntervalWesocketPush)
  console.log('连接失败')
  // if (Socket.readyState !== 3) {
  //   Socket = null
  //   createSocket()
  // }
}

/**WS数据接收统一处理 */
const onmessageWS = e => {
  window.dispatchEvent(new CustomEvent('onmessageWS', {
    detail: {
      data: e.data
    }
  }))
}

/**
 * 发送数据但连接未建立时进行处理等待重发
 * @param {any} message 需要发送的数据
 */
const connecting = message => {
  setTimeout(() => {
    if (Socket.readyState === 0) {
      connecting(message)
    } else {
      Socket.send(JSON.stringify(message))
    }
  }, 1000)
}

/**
 * 发送数据
 * @param {any} message 需要发送的数据
 */
export const sendWSPush = message => {
  if (Socket !== null && Socket.readyState === 3) {
    Socket.close()
    createSocket()
  } else if (Socket.readyState === 1) {
    Socket.send(JSON.stringify(message))
  } else if (Socket.readyState === 0) {
    connecting(message)
  }
}

/**断开 */
const oncloseWS = () => {
  clearInterval(setIntervalWesocketPush)
  console.log('websocket已断开')
  Socket = null
  // if (Socket.readyState !== 2) {
  //   Socket = null
  //   createSocket()
  // }
}
/**发送心跳
 * @param {number} time 心跳间隔毫秒 默认5000
 * @param {string} ping 心跳名称 默认字符串ping
 */
export const sendPing = (time = 5000, ping = 'ping') => {
  clearInterval(setIntervalWesocketPush)
  Socket.send(ping)
  setIntervalWesocketPush = setInterval(() => {
    Socket.send(ping)
  }, time)
}

main.js引入websocket

import { createSocket,sendWSPush } from '@/api/websocket'
Vue.prototype.$createSocket = createSocket//传一个参数 url
Vue.prototype.$sendWSPush  = sendWSPush //传一个参数 message

使用

this.$createSocket(wsUrl).then(
        (res)=>{
          this.Socket=res
        }
      );