目前websocket技術(shù)已經(jīng)很成熟，選型Go語言，當然是為了節(jié)省成本以及它強大的高并發(fā)性能。我使用的是第三方開源的websocket庫即gorilla/websocket。
由于我們線上推送的量不小，推送后端需要部署多節(jié)點保持高可用，所以需要自己做集群，具體架構(gòu)方案如圖：
Auth Service：鑒權(quán)服務(wù)，根據(jù)Token驗證用戶權(quán)限。
Collect Service：消息采集服務(wù)，負責(zé)收集業(yè)務(wù)系統(tǒng)消息，存入MongoDB后，發(fā)送給消息分發(fā)服務(wù)。
Dispatch Service：消息分發(fā)服務(wù)，根據(jù)路由規(guī)則分發(fā)至對應(yīng)消息推送服務(wù)節(jié)點上。
Push Service：消息推送服務(wù)，通過websocket將消息推送給用戶。
集群推送的關(guān)鍵點在于，web端與服務(wù)端建立長連接之后，具體跟哪個推送節(jié)點保持長連接的，如果我們能夠找到對應(yīng)的連接節(jié)點，那么我們就可以將消息推送出去。下面講解一下集群的大致流程：
1>. web端用戶登錄之后，帶上token與后端推送服務(wù)（Push Service）保持長連接。
2>. 推送服務(wù)收到連接請求之后，攜帶token去鑒權(quán)服務(wù)（Auth Service）驗證此token權(quán)限，并返回用戶ID。
3>. 把返回的用戶ID與長連接存入本地緩存，保持用戶ID與長連接綁定關(guān)系。
4>. 再將用戶ID與本推送節(jié)點IP存入redis，建立用戶（即長連接）與節(jié)點綁定關(guān)系，并設(shè)置失效時間。
5>. 采集服務(wù)（Collect Service）收集業(yè)務(wù)消息，首先存入mongodb，然后將消息透傳給分發(fā)服務(wù)（Dispatch Service）。
6>. 分發(fā)服務(wù)收到消息之后，根據(jù)消息體中的用戶ID，從redis中獲取對應(yīng)的推送服務(wù)節(jié)點IP，然后轉(zhuǎn)發(fā)給對應(yīng)的推送節(jié)點。
7>. 推送服務(wù)節(jié)點收到消息之后，根據(jù)用戶ID，從本地緩存中取出對應(yīng)的長連接，將消息推送給客戶端。
其他注意事項：
- 如何清理推送服務(wù)節(jié)點的失效長連接？
上述第四步用戶ID與節(jié)點存入redis時設(shè)置了過期時間，我們通過訂閱Redis的鍵事件通知（key-event notification），清理掉過期的長連接。 - 用戶（即長連接）與節(jié)點綁定關(guān)系如何續(xù)期？
客戶端與推送服務(wù)端建立心跳機制（報文帶上token），定期刷新redis中用戶與節(jié)點綁定關(guān)系時間。 - 如何保證長連接一直可用？
心跳重連機制，客戶端定時往服務(wù)端發(fā)送心跳報文以檢測連接是否正常。
