最近在開發(fā)小程序用到了WebSocket，小程序提供了相應(yīng)的原生API，與H5的API使用方式上有一些區(qū)別，所以流行的H5的一些成熟的類庫(kù)使用起來(lái)有些困難，而原生API又存在一些缺陷，所以就自己實(shí)現(xiàn)了一套心跳重連機(jī)制。

慣例，先簡(jiǎn)單介紹一下Websocket。

Websocket簡(jiǎn)介

Websocket是什么

WebSocket 是一種網(wǎng)絡(luò)通信協(xié)議。RFC6455 定義了它的通信標(biāo)準(zhǔn)。

WebSocket 是 HTML5 開始提供的一種在單個(gè) TCP 連接上進(jìn)行全雙工通訊的協(xié)議。

為什么需要Websocket

HTTP 協(xié)議是一種無(wú)狀態(tài)的、無(wú)連接的、單向的應(yīng)用層協(xié)議。它采用了請(qǐng)求/響應(yīng)模型。通信請(qǐng)求只能由客戶端發(fā)起，服務(wù)端對(duì)請(qǐng)求做出應(yīng)答處理。

所以當(dāng)我們想服務(wù)器主動(dòng)給客戶端發(fā)送消息，HTTP是做不到的，我們只能使用輪詢或者長(zhǎng)輪詢來(lái)實(shí)現(xiàn)類似的功能，這樣的方式效率低并且浪費(fèi)資源，為了解決這樣的問(wèn)題，WebSocket誕生了。

小程序中的WebSocket

小程序WebSocket的API

把你給懶得！自己官網(wǎng)看去!

為什么要做心跳重連

在使用原生WebSocket的時(shí)候，我們經(jīng)常會(huì)感覺(jué)不太穩(wěn)定，服務(wù)端發(fā)送的消息有時(shí)候客戶端接收不到，或者是客戶端發(fā)送的消息服務(wù)端接收不到，雖然WebSocket也提供了onError和onClose的方法，但是經(jīng)常會(huì)有各種未知情況導(dǎo)致斷開連接而并不觸發(fā)Error或Close事件。這樣就導(dǎo)致實(shí)際連接已經(jīng)斷開了，而客戶端和服務(wù)端卻不知道，還在傻傻的等著消息來(lái)。

所以我們要解決的問(wèn)題就很清晰了：

保證連接狀態(tài)，連接斷開時(shí)讓客戶端與服務(wù)端都能知道，進(jìn)而重連。

上代碼

頁(yè)面載入后,我們連接socket先

onLoad(){
  this.linkSocket()
 },
 linkSocket(){
  let that = this
  wx.connectSocket({
   url: app.globalData.wsUrl + 'websocket?' + this.data.taskId + '&' + this.data.userId,
   success() {
    console.log('連接成功')
    that.initEventHandle()
   }
  })
 },

綁定事件

然后調(diào)用initEventHandle來(lái)綁定各種各樣的事件

initEventHandle(){
  let that = this
  wx.onSocketMessage((res) => {
   //收到消息
  })
  wx.onSocketOpen(()=>{
   console.log('WebSocket連接打開')
  })
  wx.onSocketError(function (res) {
   console.log('WebSocket連接打開失敗')
  })
  wx.onSocketClose(function (res) {
   console.log('WebSocket 已關(guān)閉！')
  })
 },

這個(gè)先放在這我們一會(huì)往里填東西

斷線重連

reconnect(){
  if (this.lockReconnect) return;
  this.lockReconnect = true;
  clearTimeout(this.timer)
  if (this.data.limit<12){
   this.timer = setTimeout(() => {
    this.linkSocket();
    this.lockReconnect = false;
   }, 5000);
   this.setData({
    limit: this.data.limit+1
   })
  }
 },

我們?cè)O(shè)置一個(gè)鎖和最大的重連次數(shù)，避免出現(xiàn)無(wú)限重連的情況，為了不給服務(wù)器太大的壓力我這里設(shè)置的是5秒重試一次，最多請(qǐng)求12次。

改造一下initEventHandle這樣我們就可以實(shí)現(xiàn)一般的觸發(fā)Error的斷線重連。

initEventHandle(){
  let that = this
  wx.onSocketMessage((res) => {
   //收到消息
  })
  wx.onSocketOpen(()=>{
   console.log('WebSocket連接打開')
  })
  wx.onSocketError((res)=>{ 
   console.log('WebSocket連接打開失敗')
   this.reconnect()
  })
  wx.onSocketClose((res)=> {
   console.log('WebSocket 已關(guān)閉！')
   this.reconnect()
  })
 },

關(guān)鍵的來(lái)的---心跳對(duì)象

先擼為敬

let heartCheck = {
 timeout: 10000, 
 timeoutObj: null,
 serverTimeoutObj: null,
 reset: function () {
  clearTimeout(this.timeoutObj);
  clearTimeout(this.serverTimeoutObj);
  return this;
 },
 start: function () {
  this.timeoutObj = setTimeout(()=> {
   console.log("發(fā)送ping");
   wx.sendSocketMessage({
    data:"ping",
    // success(){
    //  console.log("發(fā)送ping成功");
    // }
   });
   this.serverTimeoutObj = setTimeout(() =>{
    wx.closeSocket(); 
   }, this.timeout);
  }, this.timeout);
 }
};

心跳對(duì)象內(nèi)timeout為每10秒發(fā)一次心跳,timeoutObj、serverTimeoutObj是清除定時(shí)器用的對(duì)象，reset方法重置定時(shí)器，start發(fā)送心跳。

繼續(xù)改造我們的initEventHandle

initEventHandle(){
  let that = this
  wx.onSocketMessage((res) => {
   //收到消息
   if (res.data == "pong"){
    heartCheck.reset().start()
   }else{
    \\處理數(shù)據(jù)
   }
  })
  wx.onSocketOpen(()=>{
   console.log('WebSocket連接打開')
   heartCheck.reset().start()
  })
  wx.onSocketError((res)=>{ 
   console.log('WebSocket連接打開失敗')
   this.reconnect()
  })
  wx.onSocketClose((res)=> {
   console.log('WebSocket 已關(guān)閉！')
   this.reconnect()
  })
 },

打開連接的時(shí)候調(diào)用start開始心跳,每隔10秒向服務(wù)端發(fā)送消息"ping",服務(wù)端接收到消息后給我們回個(gè)話"pong",就好像微信聊天。

在嗎?

在

在嗎?

在

在嗎?

滾

別看在嗎很煩，至少能確定你們倆還是好友，要不他把你拉黑了你還不知道（有點(diǎn)跑題了）

如果超過(guò)10秒服務(wù)端還沒(méi)回復(fù)“pong”，則認(rèn)為連接斷開的

我擦，敢刪我好友，我這東北爺們的暴脾氣這能忍？我把你也刪了 。。。。。。。。。。。然后趕緊重新申請(qǐng)好友

這是暴脾氣的客戶端直接把連接關(guān)閉，調(diào)用closeSocket，可是一關(guān)閉我們的onSocketClose事件又去重連了（渣男）

至此，心跳重連實(shí)現(xiàn)完畢

總結(jié)

Socket心跳重連在H5中可以有類似的實(shí)現(xiàn)，也有成熟的第三方庫(kù)，比如說(shuō)stomp+websocket+sockjs可以實(shí)現(xiàn)更加完整的更兼容的socket連接方案,可以下次給大家分享(看我心情)。

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

最新評(píng)論