来源
websocket连接之后,如果隔一段时间不聊天,运营商会认为你空占网络资源,强行关闭你的websocket连接,
如果想要一直保持连接,就需要监听websocket关闭事件,等关闭时重新连接从而让websocket一直保持连接.
这个过程称为心跳
1、必须要有心跳，如果没有会自动断开连接。
2‘、断开后必须取消setTimeout，不然会继续执行一次。
3、发送消息只有在onSocketOpen后才会发送，所以添加一个消息数组，等连接成功后再发送。
4、如果在还没连接成功时退出连接，会导致无法关闭，所以添加了socketClose来关闭socket
// socket已经连接成功var socketOpen = false// socket已经调用关闭functionvar socketClose = false// socket发送的消息队列var socketMsgQueue = []// 判断心跳变量var heart = ''// 心跳失败次数var heartBeatFailCount = 0// 终止心跳var heartBeatTimeOut = null;// 终止重新连接var connectSocketTimeOut = null; var webSocket = { /** * 创建一个 WebSocket 连接 * @param {options} * url String 是 开发者服务器接口地址，必须是 wss 协议，且域名必须是后台配置的合法域名 * header Object 否 HTTP Header , header 中不能设置 Referer * method String 否 默认是GET，有效值：OPTIONS, GET, HEAD, POST, PUT, DELETE, TRACE, CONNECT * protocols StringArray 否 子协议数组 1.4.0 * success Function 否 接口调用成功的回调函数 * fail Function 否 接口调用失败的回调函数 * complete Function 否 接口调用结束的回调函数（调用成功、失败都会执行） */ connectSocket: function(options) { wx.showLoading({ title: '', mask: true, }) socketOpen = false socketClose = false socketMsgQueue = [] wx.connectSocket({ url: '开发者服务器接口地', success: function(res) { if (options) { // 成功回调 options.success && options.success(res); } }, fail: function(res) { if (options) { // 失败回调 options.fail && options.fail(res); } } }) }, /** * 通过 WebSocket 连接发送数据 * @param {options} * data String / ArrayBuffer 是 需要发送的内容 * success Function 否 接口调用成功的回调函数 * fail Function 否 接口调用失败的回调函数 * complete Function 否 接口调用结束的回调函数（调用成功、失败都会执行） */ sendSocketMessage: function(options) { if (socketOpen) { wx.sendSocketMessage({ data: options.msg, success: function(res) { if (options) { options.success && options.success(res); } }, fail: function(res) { if (options) { options.fail && options.fail(res); } } }) } else { socketMsgQueue.push(options.msg) } }, /** * 关闭 WebSocket 连接。 * @param {options} * code Number 否 一个数字值表示关闭连接的状态号，表示连接被关闭的原因。如果这个参数没有被指定，默认的取值是1000 （表示正常连接关闭） * reason String 否 一个可读的字符串，表示连接被关闭的原因。这个字符串必须是不长于123字节的UTF-8 文本（不是字符） * fail Function 否 接口调用失败的回调函数 * complete Function 否 接口调用结束的回调函数（调用成功、失败都会执行） */ closeSocket: function(options) { if (connectSocketTimeOut) { clearTimeout(connectSocketTimeOut); connectSocketTimeOut = null; } socketClose = true; var self = this; self.stopHeartBeat(); wx.closeSocket({ success: function(res) { console.log('WebSocket 已关闭！'); if (options) { options.success && options.success(res); } }, fail: function(res) { if (options) { options.fail && options.fail(res); } } }) }, // 收到消息回调 onSocketMessageCallback: function(msg) { }, // 开始心跳 startHeartBeat: function() { console.log('socket开始心跳') var self = this; heart = 'heart'; self.heartBeat(); }, // 结束心跳 stopHeartBeat: function() { console.log('socket结束心跳') var self = this; heart = ''; if (heartBeatTimeOut) { clearTimeout(heartBeatTimeOut); heartBeatTimeOut = null; } if (connectSocketTimeOut) { clearTimeout(connectSocketTimeOut); connectSocketTimeOut = null; } }, // 心跳 heartBeat: function() { var self = this; if (!heart) { return; } self.sendSocketMessage({ msg: JSON.stringify({ 'msg_type': 'heart' }), success: function(res) { console.log('socket心跳成功'); if (heart) { heartBeatTimeOut = setTimeout(() => { self.heartBeat(); }, 7000); } }, fail: function(res) { console.log('socket心跳失败'); if (heartBeatFailCount > 2) { // 重连 self.connectSocket(); } if (heart) { heartBeatTimeOut = setTimeout(() => { self.heartBeat(); }, 7000); } heartBeatFailCount++; }, }); }} // 监听WebSocket连接打开事件。callback 回调函数wx.onSocketOpen(function(res) { console.log('WebSocket连接已打开！') wx.hideLoading(); // 如果已经调用过关闭function if (socketClose) { webSocket.closeSocket(); } else { socketOpen = true for (var i = 0; i < socketMsgQueue.length; i++) { webSocket.sendSocketMessage(socketMsgQueue[i]) } socketMsgQueue = [] webSocket.startHeartBeat(); }}) // 监听WebSocket错误。wx.onSocketError(function(res) { console.log('WebSocket连接打开失败，请检查！', res)}) // 监听WebSocket接受到服务器的消息事件。wx.onSocketMessage(function(res) { console.log('收到服务器内容：' + res.data) webSocket.onSocketMessageCallback(res.data)}) // 监听WebSocket关闭。wx.onSocketClose(function(res) { console.log('WebSocket 已关闭！') if (!socketClose) { clearTimeout(connectSocketTimeOut) connectSocketTimeOut = setTimeout(() => { webSocket.connectSocket(); }, 3000); }}) module.exports = webSocket;
