前言

在16年的時(shí)候因?yàn)轫?xiàng)目接觸到websocket，而后對心跳重連做了一次總結(jié)，寫了篇博客，而后18年對之前github上的demo代碼進(jìn)行了再次開發(fā)和開源，最終封裝成庫。如下：

github: https://github.com/zimv/websocket-heartbeat-js
npm: https://www.npmjs.com/package/websocket-heartbeat-js

在2020年也就是今年初，同事建議說可以考慮兼容一下小程序，心想也挺好的。便有了今天的 websocket-heartbeat-miniprogram，這次基于以前的代碼新建了一個(gè)項(xiàng)目，只做小程序的版本，因?yàn)樯婕暗礁鞣N小程序以及相關(guān)框架的兼容，覺得還是單獨(dú)出一個(gè)包，更專注一點(diǎn)。

介紹

websocket-heartbeat-miniprogram基于小程序的websocket相關(guān)API進(jìn)行封裝，主要目的是保障客戶端websocket與服務(wù)端連接狀態(tài)。該程序有心跳檢測及自動(dòng)重連機(jī)制，當(dāng)網(wǎng)絡(luò)斷開或者后端服務(wù)問題造成客戶端websocket斷開，程序會(huì)自動(dòng)嘗試重新連接直到再次連接成功。兼容市面上大部分小程序微信，百度，支付寶等，只要都是統(tǒng)一的小程序weboscket-API規(guī)范。也支持小程序框架比如Taro等。無論如何，業(yè)務(wù)是需要一層心跳機(jī)制的，否則一些情況下會(huì)丟失連接導(dǎo)致功能無法使用。

用法

安裝

npm install --save websocket-heartbeat-miniprogram

引入使用

import WebsocketHeartbeat from 'websocket-heartbeat-miniprogram';
WebsocketHeartbeat({
  miniprogram: wx,
  connectSocketParams: {
    url: 'ws://xxx'
  }
})
  .then(task => {
    task.onOpen = () => {//鉤子函數(shù)
      console.log('open');
    };
    task.onClose = （） => {//鉤子函數(shù)
      console.log('close');
    };
    task.onError = e => {//鉤子函數(shù)
      console.log('onError：', e);
    };
    task.onMessage = data => {//鉤子函數(shù)
      console.log('onMessage', data);
    };
    task.onReconnect = () => {//鉤子函數(shù)
      console.log('reconnect...');
    };
    task.socketTask.onOpen(data => {//原生實(shí)例注冊函數(shù)，重連后丟失
      console.log('socketTask open');
    });
    task.socketTask.onMessage(data => {//原生實(shí)例注冊函數(shù)，重連后丟失
      console.log('socketTask data');
    });
  })

本程序內(nèi)部總是使用小程序connectSocket方法進(jìn)行socket連接，如果socket斷開，本程序內(nèi)部會(huì)再次執(zhí)行小程序的connectSocket方法，以此來重新建立連接，重連都會(huì)建立新的小程序socket實(shí)例。

WebsocketHeartbeat方法返回一個(gè)promise，返回的task對象是本程序的一個(gè)實(shí)例,提供了onOpen,onClose,onError,onMessage,onReconnect等鉤子函數(shù)。也暴露了小程序本身的實(shí)例（socketTask），task.socketTask就是小程序connectSocket返回的實(shí)例，而task.socketTask是小程序的原生實(shí)例，它們通過onXXX方法傳遞函數(shù)進(jìn)行監(jiān)聽注冊，在socket重連以后，內(nèi)部重新通過connectSocket新建實(shí)例，將會(huì)返回新的小程序原生實(shí)例，因此之前通過socketTask.onXXX注冊的這些函數(shù)將會(huì)丟失。而本程序內(nèi)部提供的鉤子函數(shù)重連上以后依然有效。大部分情況下推薦就使用本程序的鉤子函數(shù)。

支付寶小程序差異

支付寶小程序只允許同時(shí)存在一個(gè)socket連接，原生的API也和其他小程序有一點(diǎn)小差異，本程序已經(jīng)做了兼容，但是還是要注意支付寶只允許建立一個(gè)socket，如果建立多個(gè)socket，前面的socket都會(huì)被系統(tǒng)關(guān)閉，一定要通過本程序?qū)嵗膖ask.close關(guān)閉舊的socket，否則程序會(huì)一直重連，導(dǎo)致所有socket相互沖突無法使用。

約定

1.只能通過前端主動(dòng)關(guān)閉socket連接

如果需要斷開socket，應(yīng)該執(zhí)行task.close()方法。如果后端想要關(guān)閉socket，應(yīng)該下發(fā)一個(gè)消息，前端判斷此消息，前端再調(diào)用task.close()方法關(guān)閉。因?yàn)闊o論是后端調(diào)用close還是因?yàn)槠渌蛟斐傻膕ocket關(guān)閉，前端的socket都會(huì)觸發(fā)onClose事件，程序無法判斷是什么原因?qū)е碌年P(guān)閉。因此本程序會(huì)默認(rèn)嘗試重連。

import WebsocketHeartbeat from 'websocket-heartbeat-miniprogram';
WebsocketHeartbeat({
  miniprogram: wx,
  connectSocketParams: {
    url: 'ws://xxxx'
  }
})
  .then(task => {
    task.onMessage = data => {
      if(data.data == 'close') task.close();//關(guān)閉socket并且，不再重連
    };
  })

2.后端對前端心跳的反饋

前端發(fā)送心跳消息，后端收到后，需要立刻返回響應(yīng)消息，后端響應(yīng)的消息可以是任何值，因?yàn)楸境绦虿⒉惶幚砗团袛囗憫?yīng)的心跳消息，而只是在收到任何消息后，重置心跳，因?yàn)槭盏饺魏蜗⒕驼f明連接是正常的。因此本程序收到任何后端返回的消息都會(huì)重置心跳倒計(jì)時(shí)，以此來減少不必要的請求，減少服務(wù)器壓力。

API

詳情參考： https://github.com/zimv/websocket-heartbeat-miniprogram

結(jié)語

程序已經(jīng)經(jīng)過單元測試，也在百度，支付寶，微信等小程序之中實(shí)際測試和基于Taro測試。程序長期維護(hù)，發(fā)現(xiàn)兼容問題或者程序問題，希望得到issue反饋！

到此這篇關(guān)于小程序websocket心跳庫(websocket-heartbeat-miniprogram)的文章就介紹到這了,更多相關(guān)小程序websocket心跳庫內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論