封装socket.js
export default class Websocket {
constructor({
heartCheck,
isReconnection
}) {
// 是否连接
this._isLogin = false;
// 当前网络状态
this._netWork = true;
// 是否人为退出
this._isClosed = false;
// 心跳检测频率
this._timeout = 3000;
this._timeoutObj = null;
// 当前重连次数
this._connectNum = 0;
// 心跳检测和断线重连开关，true为启用，false为关闭
this._heartCheck = heartCheck;
this._isReconnection = isReconnection;
this._onSocketOpened();
this.wsUrl = ""
}
// 心跳重置
_reset() {
clearTimeout(this._timeoutObj);
return this;
}
// 心跳开始
_start() {
let _this = this;
this._timeoutObj = setInterval(() => {
wx.sendSocketMessage({
// 心跳发送的信息应由前后端商量后决定
data: JSON.stringify({
"key": 'value'
}),
success(res) {
// console.log(res)
// console.log("发送心跳成功");
},
fail(err) {
console.log(err)
_this._reset()
}
});
}, this._timeout);
}
// 监听websocket连接关闭
onSocketClosed(options) {
wx.onSocketClose(err => {
console.log('当前websocket连接已关闭,错误信息为:' + JSON.stringify(err));
// 停止心跳连接
if (this._heartCheck) {
this._reset();
}
// 关闭已登录开关
this._isLogin = false;
// 检测是否是用户自己退出小程序
if (!this._isClosed) {
// 进行重连
if (this._isReconnection) {
this._reConnect(options)
}
}
})
}
// 检测网络变化
onNetworkChange(options) {
wx.onNetworkStatusChange(res => {
console.log('当前网络状态:' + res.isConnected);
if (!this._netWork) {
this._isLogin = false;
// 进行重连
if (this._isReconnection) {
this._reConnect(options)
}
}
})
}
_onSocketOpened(callBack) {
wx.onSocketOpen(res => {
console.log('websocket已打开');
// 打开已登录开关
this._isLogin = true;
// 发送心跳
if (this._heartCheck) {
this._reset()._start();
}
// 发送登录信息
wx.sendSocketMessage({
// 这里是第一次建立连接所发送的信息，应由前后端商量后决定
data: JSON.stringify({
"key": 'value'
})
})
// 打开网络开关
this._netWork = true;
if (typeof callBack == "function") {
callBack(res)
} else {
console.log('参数的类型必须为函数')
}
})
}
// 接收服务器返回的消息
onReceivedMsg(callBack) {
wx.onSocketMessage(msg => {
if (typeof callBack == "function") {
callBack(msg)
} else {
console.log('参数的类型必须为函数')
}
})
}
// 建立websocket连接
initWebSocket(options) {
let _this = this;
this.wsUrl = options.url ? options.url : this.wsUrl
if (this._isLogin) {
console.log("您已经登录了");
} else {
// 检查网络
wx.getNetworkType({
success(result) {
if (result.networkType != 'none') {
// 开始建立连接
wx.connectSocket({
url: _this.wsUrl,
success(res) {
if (typeof options.success == "function") {
options.success(res)
} else {
console.log('参数的类型必须为函数')
}
},
fail(err) {
if (typeof options.fail == "function") {
options.fail(err)
} else {
console.log('参数的类型必须为函数')
}
}
})
} else {
console.log('网络已断开');
_this._netWork = false;
}
}
})
}
}
// 发送websocket消息
sendWebSocketMsg(options) {
// console.log("send参数：", options)
wx.sendSocketMessage({
data: options.data,
success(res) {
if (options.success && typeof options.success == "function") {
options.success(res)
}
},
fail(err) {
if (options.fail && typeof options.fail == "function") {
options.fail(err)
}
}
})
}
// 重连方法，会根据时间频率越来越慢
_reConnect(options) {
let timer, _this = this;
if (this._connectNum < 3) {
timer = setTimeout(() => {
this.initWebSocket(options)
}, 3000)
this._connectNum += 1;
} else if (this._connectNum < 10) {
timer = setTimeout(() => {
this.initWebSocket(options)
}, 10000)
this._connectNum += 1;
} else {
timer = setTimeout(() => {
this.initWebSocket(options)
}, 450000)
this._connectNum += 1;
}
}
// 关闭websocket连接
closeWebSocket() {
wx.closeSocket();
this._isClosed = true;
}
}
app.js 使用
//引入文件
import Websocket from "./socket"
page({
onLaunch() {
this.socketInit()
},
onShow() {
this.linkWebsocket()
},
// 建立连接
linkWebsocket() {
this.websocket.initWebSocket({
url: this.globalData.websocketUrl,
success(res) {
// console.log(res)
},
fail(err) {
console.log("linkWebsocket err", err)
}
})
},
/**
* 创建websocket对象
*/
socketInit() {
// 创建websocket对象
this.websocket = new Websocket({
// true代表启用心跳检测和断线重连
heartCheck: false,
isReconnection: true
});
// 建立连接
// 监听websocket状态
this.websocket.onSocketClosed({
url: this.globalData.websocketUrl,
success(res) {
console.log(res)
},
fail(err) {
console.log("onSocketClosed err", err)
}
})
// 监听网络变化
this.websocket.onNetworkChange({
url: this.globalData.websocketUrl,
success(res) {
console.log(res)
},
fail(err) {
console.log("onNetworkChange err", err)
}
})
// 监听服务器返回
this.websocket.onReceivedMsg(result => {
console.log('app.js收到服务器内容：' + result.data);
// 要进行的操作
})
},
// 向其他页面暴露当前websocket连接
getSocket() {
return this.websocket;
},
globalData: {
websocketUrl: "你的wss链接地址"
}
})
