<template>
  <div class="test">
  </div>
</template>
<script>
  export default {
    name : 'test',
    data() {
      return {
        websock: null,
      }
    },
    created() {
      this.initWebSocket();
    },
    destroyed() {
      this.websock.close() //離開路由之后斷開websocket連接
    },
    methods: {
      initWebSocket(){ //初始化weosocket
        const wsuri = "ws://127.0.0.1:8080";
        this.websock = new WebSocket(wsuri);
        this.websock.onmessage = this.websocketonmessage;
        this.websock.onopen = this.websocketonopen;
        this.websock.onerror = this.websocketonerror;
        this.websock.onclose = this.websocketclose;
      },
      websocketonopen(){ //連接建立之后執(zhí)行send方法發(fā)送數(shù)據(jù)
        let actions = {"test":"12345"};
        this.websocketsend(JSON.stringify(actions));
      },
      websocketonerror(){//連接建立失敗重連
        this.initWebSocket();
      },
      websocketonmessage(e){ //數(shù)據(jù)接收
        const redata = JSON.parse(e.data);
      },
      websocketsend(Data){//數(shù)據(jù)發(fā)送
        this.websock.send(Data);
      },
      websocketclose(e){  //關(guān)閉
        console.log('斷開連接',e);
      },
    },
  }
</script>
<style lang='less'>
</style>

let Socket = ''
let setIntervalWesocketPush = null
/**
 * 建立websocket連接
 * @param {string} url ws地址
 */
export const 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已連接')
  }
}
/**打開WS之后發(fā)送心跳 */
const onopenWS = () => {
  sendPing()
}
/**連接失敗重連 */
const onerrorWS = () => {
  Socket.close()
  clearInterval(setIntervalWesocketPush)
  console.log('連接失敗重連中')
  if (Socket.readyState !== 3) {
    Socket = null
    createSocket()
  }
}
/**WS數(shù)據(jù)接收統(tǒng)一處理 */
const onmessageWS = e => {
  window.dispatchEvent(new CustomEvent('onmessageWS', {
    detail: {
      data: e.data
    }
  }))
}
/**
 * 發(fā)送數(shù)據(jù)但連接未建立時進(jìn)行處理等待重發(fā)
 * @param {any} message 需要發(fā)送的數(shù)據(jù)
 */
const connecting = message => {
  setTimeout(() => {
    if (Socket.readyState === 0) {
      connecting(message)
    } else {
      Socket.send(JSON.stringify(message))
    }
  }, 1000)
}
/**
 * 發(fā)送數(shù)據(jù)
 * @param {any} message 需要發(fā)送的數(shù)據(jù)
 */
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已斷開....正在嘗試重連')
  if (Socket.readyState !== 2) {
    Socket = null
    createSocket()
  }
}
/**發(fā)送心跳
 * @param {number} time 心跳間隔毫秒 默認(rèn)5000
 * @param {string} ping 心跳名稱 默認(rèn)字符串ping
 */
export const sendPing = (time = 5000, ping = 'ping') => {
  clearInterval(setIntervalWesocketPush)
  Socket.send(ping)
  setIntervalWesocketPush = setInterval(() => {
    Socket.send(ping)
  }, time)
}

npm install @sven0706/websocket -S

// 在main.js或需要使用的地方引入并建立連接
import { createSocket } from '@sven0706/websocket'
createSocket('wss://api.baidu.com')
// 發(fā)送消息
import { sendWSPush } from '@sven0706/websocket'
sendWSPush(data)
// 接收消息
const getsocketData = e => {  // 創(chuàng)建接收消息函數(shù)
  const data = e && e.detail.data
  console.log(data)
}
// 注冊監(jiān)聽事件
window.addEventListener('onmessageWS', getsocketData)
// 在需要的時候卸載監(jiān)聽事件，比如離開頁面
window.removeEventListener('onmessageWS', getsocketData)

/**
 * 建立websocket連接
 * @param {string} url ws地址
 */
createSocket(url)
/**
 * 發(fā)送數(shù)據(jù)
 * @param {any} message 需要發(fā)送的數(shù)據(jù)
 */
 sendWSPush(message)
/**發(fā)送心跳
 * @param {number} time 心跳間隔毫秒 默認(rèn)5000
 * @param {string} ping 心跳名稱 默認(rèn)字符串ping
 */
 sendPing()