Vue 封装WebSocket实现聊天功能

最近在做客户端聊天的功能，为了保持访客端和座席端的链接，就使用到了webSocket来创建、保持访客端和座席端的实时通信。

 

关于webSocket的知识，可以看一下w3cSchool：Websocket_w3cschool或者可以移步“不懂一休”的文章WebSocket 入门_不懂一休的博客-优快云博客

一、话不多说，简单的websocket链接代码已发送

initWebSocket() {
	if (typeof WebSocket === 'undefined') {
		return
	}
	this.websock = new WebSocket(config.webSoketUrl)
	this.websock.onmessage = this.webSocketOnMessage
	this.websock.onopen = this.webSocketOnOpen
	this.websock.onerror = this.websocketOnError
	this.websock.onclose = this.websocketOnClose
},

webSocketOnOpen() {
	//建立链接，执行send方法发送数据
},

webSocketSend(data) {
	this.websock.send(data)
},

webSocketOnMessage(e) {
	//数据接受
	const readData = JSON.parse(e.data)
	console.log()
},

websocketOnError() {
	//连接建立失败
	this.initWebSocket()
},

websocketOnClose(e) {
	//关闭
	console.log("链接断开", e)
	this.websock.close()
},

需要在页面销毁的时候断开链接

destroyed() {
	//this.websock.onclose = this.websocketOnClose
	document.removeEventListener("click", this.bodyCloseMenus);
},

二、封装的WebSocket.js

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之后发送心跳 */
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已断开....正在尝试重连')
  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或需要使用的地方引入并建立连接
import { createSocket } from '@/assets/js/websocket'
createSocket('ws://xxxx')


// 发送消息
import { sendWSPush } from '@sven0706/websocket'
sendWSPush(data)

// 接收消息
const getsocketData = e => {  // 创建接收消息函数
  const data = e && e.detail.data
  console.log(data)
}
// 监听事件
window.addEventListener('onmessageWS', getsocketData)

// 离开页面卸载监听事件
window.removeEventListener('onmessageWS', getsocketData)

分享完毕~

后续会更新做客服聊天的心得体会，感谢各位看官