vue 实现websocket发送消息并实时接收消息
时间：2021-04-28 08:05:46|栏目：vue|点击： 次
公司做了个数字货币行情的H5，需要用到websocket，刚好自己也不会，可以学习一下，美滋滋。
项目结合vue脚手架和websocket来搭建，主要遇到了两个问题，一个是：断开重连机制要如何处理；另外一个是：如何在页面上随时的发送消息并实时接收返回的数据，断开重连后又如何重新发送消息，接收消息
一、断开重连机制处理 （详细参考这里）
写一个重连的方法（重连方法中必须加一把锁，重连方法执行过程中不再执行重连动作，避免重复连接），然后在websocket的onclose和error中绑定重连方法，这样一般情况下，websocket断开或者链接出错就可以实现重连了。针对断网重连问题，就需要用心跳检测了（主要是利用websocket定时发送消息，当超过一定时间消息还未发送成功，浏览器的websocket会自动触发onclose方法，而此时onclose有绑定了重连函数，于是就触发websocket重新连接），项目这边暂时不考虑这个，所以没做心跳检测。
主要代码：
let socket = null;
let lockReconnet = false; //避免重复连接
const wsUrl = '自己的websocket接口';
let createSocket = url=>{ //创建socket
try{
if('WebSocket' in window){
socket = new WebSocket(url)
}else if('MozWebSocket' in window){
socket = new MozWebSocket(url)
}
initSocket()
}catch(e){
reconnet(url)
}
}
let initSocket = ()=>{ //初始化websocket
socket.onopen = ()=>{
console.log('socket连接成功')
//heartCheck.reset().start() //后端说暂时不需要做心跳检测
}
socket.onmessage = (ev)=>{
console.log(ev,'连接正常')
//heartCheck.reset().start() //后端说暂时不需要做心跳检测
}
socket.onerror = ()=>{
console.log('websocket服务出错了---onerror');
reconnet(wsUrl)
}
socket.onclose = ()=>{
console.log('websocket服务关闭了+++onclose');
reconnet(wsUrl)
}
}
let reconnet = url=>{ //重新连接websock函数
if(lockReconnet)
return false
lockReconnet = true
setTimeout(()=>{
createSocket(url)
lockReconnet = false
},2000)
}
let heartCheck = { //心跳检测
timeout: 60*1000,
timeoutObj: null,
serverTimeoutObj: null,
reset(){
clearTimeout(this.timeoutObj)
clearTimeout(this.serverTimeoutObj)
return this;
},
start(){
let that = this;
this.timeoutObj = setTimeout(()=>{
//发送数据，如果onmessage能接收到数据，表示连接正常,然后在onmessage里面执行reset方法清除定时器
socket.send('heart check')
this.serverTimeoutObj = setTimeout(()=>{
socket.close()
},that.timeout)
},this.timeout)
}
}
二、在页面上随时发送消息并实时接收消息
在上面代码的基础上增加一个发送数据的方法，该方法有两个参数，一个是需要发送的数据；另一个为接收和处理返回数据的回调函数，然后把这个方法暴露出去并挂载到Vue原型上，这样就可以在任意页面或者组件随时的发送消息，并接收消息了。具体代码：
let sendMsg = (data,callback)=>{ //发送数据,接收处理数据
if(socket.readyState===1){
globalCallback = callback; //把接收处理回调函数保存到全局
sendData = data; //把发送数据也保存到全局
data = JSON.stringify(data);
socket.send(data);
}else{
setTimeout(()=>{
console.log(socket,'等待socket链接成功')
sendMsg(data,callback)
},1500)
return false
}
socket.onmessage = ev=>{ //重新监听onmessage，并把数据传给回调函数
callback && callback(ev)
}
}
三、断开重连后如何重新发送消息和接收消息
增加一个保存要发送消息的全局变量，以及一个保存接收处理消息回调函数的全局变量，当重连触发后，重新调用下senMsg方法，并把这两个变量传进去就可以了。
完整的封装代码（mysocket.js）：
//import Vue from 'vue'
let socket = null;
let lockReconnet = false; //避免重复连接
const wsUrl = '自己的websocket接口';
let isReconnet = false;
let globalCallback = null,sendData = null; //把要发送给socket的数据和处理socket返回数据的回调保存起来
let createSocket = url=>{ //创建socket
try{
if('WebSocket' in window){
socket = new WebSocket(url)
}else if('MozWebSocket' in window){
socket = new MozWebSocket(url)
}
//Vue.prototype.socket = socket //需要主动关闭的话就可以直接调用this.socket.close()进行关闭，不需要的话这个可以去掉
initSocket()
}catch(e){
reconnet(url)
}
}
let sendMsg = (data,callback)=>{ //发送数据,接收数据
if(socket.readyState===1){
globalCallback = callback;
sendData = data;
data = JSON.stringify(data);
socket.send(data);
}else{
setTimeout(()=>{
console.log(socket,'等待socket链接成功')
sendMsg(data,callback)
},1500)
return false
}
socket.onmessage = ev=>{
callback && callback(ev)
}
}
let initSocket = ()=>{ //初始化websocket
socket.onopen = ()=>{
console.log('socket连接成功')
//heartCheck.reset().start() //后端说暂时不需要做心跳检测
if(isReconnet){//执行全局回调函数
//console.log('websocket重新连接了')
sendMsg(sendData,globalCallback)
isReconnet = false
}
}
socket.onmessage = (ev)=>{
console.log(ev,'连接正常')
//heartCheck.reset().start() //后端说暂时不需要做心跳检测
}
socket.onerror = ()=>{
console.log('websocket服务出错了---onerror');
reconnet(wsUrl)
}
socket.onclose = ()=>{
console.log('websocket服务关闭了+++onclose');
reconnet(wsUrl)
}
}
let reconnet = url=>{ //重新连接websock函数
if(lockReconnet)
return false
isReconnet = true;
lockReconnet = true
setTimeout(()=>{
createSocket(url)
lockReconnet = false
},2000)
}
let heartCheck = { //心跳检测
timeout: 60*1000,
timeoutObj: null,
serverTimeoutObj: null,
reset(){
clearTimeout(this.timeoutObj)
clearTimeout(this.serverTimeoutObj)
return this;
},
start(){
let that = this;
this.timeoutObj = setTimeout(()=>{
//发送数据，如果onmessage能接收到数据，表示连接正常,然后在onmessage里面执行reset方法清除定时器
socket.send('heart check')
this.serverTimeoutObj = setTimeout(()=>{
socket.close()
},that.timeout)
},this.timeout)
}
}
createSocket(wsUrl)
export default {sendMsg}
在main.js里面引入这个文件，并把sendMsg挂载到Vue原型上，就可以再页面上随时发送消息接收消息了。
import socket from './assets/js/mysocket' Vue.prototype.sendMsg = socket.sendMsg
在页面上调用方法：
getSocketData(symbol){
let data = {"event":"subscription","data":"market.kline."+symbol};
this.sendMsg(data,ev=>{
console.log(JSON.parse(ev.data),'K线相关数据')
})
}
效果：
总结
