socketTask.js
let timer = null; //心跳计时器
let debug = false; //debug输出log
let retimer = null; //重连计时器
let pingCount = 0; //心跳次数
let pingCountMax = 5; //最大心跳次数超过重连
let pingTimeout = 5000; //心跳间隔时间(毫秒)
let connectTimeout = 2000; //重连时间(毫秒)
let socketTask = null; //ws
// 连接WebSocket
export function connectSocket() {
debug && console.log(socketTask);
var token = uni.getStorageSync('Authorization');
if (!token || socketTask) {
console.log('开启socketTask失败')
return
}
socketTask = uni.connectSocket({
url: 'wss://im-api.q3z3.com/ws?Authorization=' + uni.getStorageSync('Authorization'),//WebSocket地址
// url: 'ws://192.168.0.200:8080/ws?Authorization=' + uni.getStorageSync('Authorization'),//WebSocket地址
complete: () => {}
});
socketTask.onOpen(res => {
console.log('WebSocket连接已打开！');
debug && console.log(socketTask);
// 发送心跳
socketTask.send({
data: 'isConnact',
success: res => {
if (res.errMsg == 'sendSocketMessage:ok') {
debug && console.log('WebSocket心跳ping');
}
}
})
// 定时心跳
timer = setInterval(() => {
// ping超过5次未响应则重连
pingCount++
debug && console.log('ping次数'+pingCount+'/'+pingCountMax);
if (pingCount >= pingCountMax) {
clearSocketTask()
reConnectSocket()
return
}
socketTask.send({
data: 'isConnact',
success: res => {
if (res.errMsg == 'sendSocketMessage:ok') {
debug && console.log('WebSocket心跳ping');
}
}
})
}, pingTimeout)
})
// 监听接收
socketTask.onMessage(res => {
if (socketTask && token) {
if(res.data=='ok'){
pingCount = 0;
return
}
var data = JSON.parse(res.data);
console.log('WebSocket接收消息！');
}
})
// 监听关闭
socketTask.onClose((res) => {
debug && console.log(socketTask);
console.log('WebSocket连接已关闭！');
if (!socketTask) {
console.log('无需操作')
return
}
if (socketTask.readyState !== 1) {
console.log('需要重新连接')
clearSocketTask()
reConnectSocket()
}
})
// 监听异常
socketTask.onError(res => {
debug && console.log(socketTask);
console.log('WebSocket连接异常！');
if (!socketTask) {
console.log('无需操作')
return
}
if (socketTask.readyState !== 1) {
console.log('需要重新连接')
clearSocketTask()
reConnectSocket()
}
});
}
// 清理WebSocket
export function clearSocketTask() {
clearInterval(timer)
clearTimeout(retimer)
pingCount = 0;
if (socketTask) {
socketTask.close()
socketTask = null
console.log('主动关闭WebSocket！');
}
}
// 重新连接WebSocket
export function reConnectSocket() {
retimer = setTimeout(() => {
connectSocket()
}, connectTimeout)
}
main.js 全局引入
import * as socketTask from "@/common/socketTask.js";
Vue.prototype.$socketTask = socketTask;
使用 onLaunch
this.$socketTask.connectSocket()
401登录过期时候
import {clearSocketTask} from "@/common/socketTask.js";
使用
clearSocketTask()