小程序连接webscoket实现即时通讯

小程序连接webscroket实现即时通讯

最近在考虑使用webscroket实现拍卖项目的即时竞拍通讯，需要进入当前页面连接webscroket,退出时回收，于是就想到封装一个简单便捷的公共方法，话不多说直接上代码

// socket已经连接成功
var socketOpen = false
// socket已经调用关闭function
var socketClose = false
// socket发送的消息队列
var socketMsgQueue = []
// 判断心跳变量
var heart = ''
// 心跳失败次数
var heartBeatFailCount = 0
// 终止心跳
var heartBeatTimeOut = null;
// 终止重新连接
var connectSocketTimeOut = null;

var i=1
const url ='';//websocket链接地址，需要在微信后台的wss配置
var webSocket = {
  //开始连接
  connectSocket: function () {
    socketOpen = false
    socketClose = false
    socketMsgQueue = []

    wx.connectSocket({
      url: url,
      // header: { 'ticket': wx.getStorageSync('token')},
      success: function (res) {
        if (res) {
          // 成功回调
          // console.log(res)
          // options.success && options.success(res);
        }
      },
      fail: function (res) {
        if (res) {
          // 失败回调
          // 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}
   */
  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({
        'messageType':1//心跳
      }),
      success: function (res) {
        console.log('socket心跳成功');
        if (heart) {
          heartBeatTimeOut = setTimeout(() => {
            self.heartBeat();
          }, 60000);
        }
      },
      fail: function (res) {
        console.log('socket心跳失败');
        if (heartBeatFailCount > 2) {
          // 重连
          self.connectSocket();
        }
        if (heart) {
          heartBeatTimeOut = setTimeout(() => {
            self.heartBeat();
          }, 60000);
        }
        heartBeatFailCount++;
      },
    });
  }
}
 
// 监听WebSocket连接打开事件。callback 回调函数
wx.onSocketOpen(function (res) {
  console.log('WebSocket连接已打开！')
  // 如果已经调用过关闭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)
  //
  if (res.data =='SUCCESS'||res.data=="连接成功"){
     return;
  }
  //接收服务器返回内容，调用页面回调函数接收
  webSocket.onSocketMessageCallback(res.data)
})
 
// 监听WebSocket关闭。
wx.onSocketClose(function (res) {
  console.log('WebSocket 已关闭!')
 	//主动关闭时，不再连接；被动关闭时，持续连接
  if (!socketClose) {
    let time = Math.pow(2, i)
    i++
    if (time==8){
      i=1
    }
    clearTimeout(connectSocketTimeOut)
    connectSocketTimeOut = setTimeout(() => {
      webSocket.connectSocket();
    }, time*1000);
  }
})
 
module.exports = webSocket;

在需要的页面引入
import webSocket from ‘…/…/…/…/utils/webscoket’;
后连接（将回调函数绑定为当前函数）
webSocket.onSocketMessageCallback = this.onSocketMessageCallback;//绑定回调函数webSocket.connectSocket();//连接webscoket
页面销毁时回收即可
/**

• 生命周期函数–监听页面卸载
  */
  onUnload: function () {
  //websocket
  webSocket.closeSocket();
  }
  运行结果看图
  连接成功
  
  退出成功