摘要：結合與的發(fā)布訂閱模式實踐本文初衷最近恰好在公司做了一個聊天室系統(tǒng)，所以在系統(tǒng)中做了一下對進行的化改造，所以想寫篇文章總結一下，如果大家有什么更好的方法或者心得感悟，歡迎交流技術?？紤]到對并沒什么本質影響，所以本文就不涉及了業(yè)務場景基于的聊天

最近恰好在公司做了一個聊天室系統(tǒng)，所以在系統(tǒng)中做了一下對websocket進行的promise化改造，所以想寫篇文章總結一下，如果大家有什么更好的方法或者心得感悟，歡迎交流

dva + protobuf
考慮到protobuf對websocket并沒什么本質影響，所以本文就不涉及了

基于websocket的聊天室系統(tǒng)

可能存在按順序觸發(fā)的請求
eg. 刪除req---確認刪除rsp---刷新頁面req---刷新頁面rsp
但由于并非所有的刪除操作后都會刷新頁面，所以考慮是不是可以使用發(fā)布訂閱模式來模擬類似promise的流式操作

存在同一類型請求短時間內多次觸發(fā)時，如何尋找每一條回復信息的發(fā)射源，考慮可以使用promise池+唯一識別碼token來實現(xiàn)

由于dva的異步操作是基于redux-saga的，所以如果可以用promise完成與websocket的互動，那么對于effects中使用yield控制異步流程，也會是一個很好的體驗

首先，這一套系統(tǒng)的一切前提是請求的唯一標識符token，前端發(fā)送給服務器之后，服務器必須要把這個token跟數(shù)據(jù)放在一起發(fā)回來才行

本系統(tǒng)的實現(xiàn)原理是

對websocket的send方法進行封裝
發(fā)送階段
1. send執(zhí)行時，先生成一個promise，及其唯一token
2. 將promise的resolve， reject，token，及其他需要的信息放入一個對象，然后推入一個promise池中
3. 執(zhí)行websocket的send方法
4. return 這個promise

接收階段
1. 收到回復消息時，先從promise池中對token進行匹配
2. 根據(jù)回復的狀態(tài)決定執(zhí)行resolve或reject
3. 其他需要的操作

// 每一個實例都只能open一條socket線路，用鎖機制防止重復open
// 本例中不使用心跳檢測，為了方便，只要close是非主動觸發(fā)且前端能捕捉到的(如瀏覽器主動斷開，服務器主動斷開),都會進行自動重連
export class MyWebSocket {
    constructor(url) {
        this.url = url;
    
        // close來源判斷及后續(xù)操作
        this.closeConfig = {
            resolve: null,
            closing: false
        }
        // promise池
        this.promisePool = [];
    }
    
    tokenCheck(req, rsp) {
    // 此處根據(jù)自己的數(shù)據(jù)結構進行tokenCheck的判斷，返回一個boolean
    }
    
    open() {
        return new Promise((resolve, reject) => {
            if (typeof this._websocket === "undefined") {
                this._websocket = new WebSocket(this.url);
                this._websocket.open = (e) => {
                    resolve({e, ws: this});
                };
                this._websocket.onerror = (e) => {
                    reject(e);
                }
            }
            this._websocket.onclose = (e) => {
                // 非主動close
                if (!this.closeConfig.closing) {
                    console.log("reconnect");     
                    // 對應的重連操作     
                }
                // 若手動close，恢復初始狀態(tài)
                this.closeConfig.closing = false;
            }
            
            this._websocket.onmessage = (e) => {
                this.promisePool = this.promisePool.filter((item) => {
                if (this.tokenCheck(req, rsp) {
                  req.resolve(rsp);
                  return false;
                }
                return true;
              })
            };
        });
    }
    
    close() {
        this.closeConfig.closing = true;
        this._websocket.close();        
    }
    // token包含在content中
    send(name, content) {
        return new Promise((resolve, reject) => {
          this.promisePool.push({
            content,
            resolve,
            reject,
            name
          });
          this._websocket.send({name, content});
    });
}

大概流程就是這樣，具體的樣例如下

*test () {
    const ws = new MyWebSocket("www.mywebsocket.com");
    yield ws.open();
    yield ws.send(.....).then(()=>{...});
    yield ws.send(.....).then(()=>{...});
}

本文呢大概就是這么多了，如果有什么錯誤或者遺漏的地方還請大家多多指教

采取了評論大佬的建議，將promise池從數(shù)組改為對象，直接將token做為key，查詢起來也非常方便

export class MyWebSocket {
    constructor(url) {
        this.url = url;
    
        // close來源判斷及后續(xù)操作
        this.closeConfig = {
            resolve: null,
            closing: false
        }
        // promise池
        this.promisePool = {};
    }
    
    tokenCheck(req, rsp) {
    // 此處根據(jù)自己的數(shù)據(jù)結構進行tokenCheck的判斷，返回一個boolean
    }
    
    open() {
        return new Promise((resolve, reject) => {
            if (typeof this._websocket === "undefined") {
                this._websocket = new WebSocket(this.url);
                this._websocket.open = (e) => {
                    resolve({e, ws: this});
                };
                this._websocket.onerror = (e) => {
                    reject(e);
                }
            }
            this._websocket.onclose = (e) => {
                // 非主動close
                if (!this.closeConfig.closing) {
                    console.log("reconnect");     
                    // 對應的重連操作     
                }
                // 若手動close，恢復初始狀態(tài)
                this.closeConfig.closing = false;
            }
            
            this._websocket.onmessage = (e) => {         
                const key = e.content.token;    
                const req = this.promisePool[key]
                req.resolve(e);
                delete this.promisePool[key];
            };
        });
    }
    
    close() {
        this.closeConfig.closing = true;
        this._websocket.close();
    }
    // token包含在content中
    send(name, content) {
        return new Promise((resolve, reject) => {
          this.promisePool[content.token] = {
            content,
            resolve,
            reject,
            name
          };
          this._websocket.send({name, content});
    });
}