閒扯

WebSocket 以前沒用過，之前寫過一篇博客是基於原生socket的（查看）比較復雜，慎入。今天另外一個APP需要接websocket了，然後便找到了facebook的 SocketRocket 框架，然後用了一天時間接上了，完成了掉線自動重連，自動重登錄，心跳等等功能，用法比原生socket簡單（原生socket基於TCP/UDP協議）。

為什麼用 WebSocket

因為APP裡面有個聊天功能，需要服務器主動推數據到APP。HTTP 通信方式只能由客戶端主動拉取，服務器不能主動推給客戶端，如果有實時的消息，要立刻通知客戶端就麻煩了，要麼客戶端每隔幾秒鐘發一次請求，看看有沒有新數據，這種方式想想都知道耗流量電量。還一種方式就是走TCP/UDP協議服務器主動推給你，這種方式省流量。還有就是用websocket，websocket是h5裡面的東西，h5我不太會，反正它比原生socket用法簡單。

用法

用 SocketRocket 框架，記住幾個代理方法就好了，很簡單。

1.創建和設置代理對象

SRWebSocket *socket = [[SRWebSocket alloc] initWithURLRequest:
[NSURLRequest requestWithURL:[NSURL URLWithString:@"http://ip地址:端口"]];

socket.delegate = self;    // 實現這個 SRWebSocketDelegate 協議啊

[socket open];    // open 就是直接連接了

2.連接成功會調用這個代理方法

- (void)webSocketDidOpen:(SRWebSocket *)webSocket {
    NSLog(@"連接成功，可以立刻登錄你公司後台的服務器了，還有開啟心跳");
}

3.連接失敗會調用這個方法，看 NSLog 裡面的東西

- (void)webSocket:(SRWebSocket *)webSocket didFailWithError:(NSError *)error {
    NSLog(@"連接失敗，這裡可以實現掉線自動重連，要注意以下幾點");
    NSLog(@"1.判斷當前網絡環境，如果斷網了就不要連了，等待網絡到來，在發起重連");
    NSLog(@"2.判斷調用層是否需要連接，例如用戶都沒在聊天界面，連接上去浪費流量");
    NSLog(@"3.連接次數限制，如果連接失敗了，重試10次左右就可以了，不然就死循環了。
    或者每隔1，2，4，8，10，10秒重連...f(x) = f(x-1) * 2, (x=5)");
}

4.連接關閉調用這個方法，注意連接關閉不是連接斷開，關閉是 [socket close] 客戶端主動關閉，斷開可能是斷網了，被動斷開的。

- (void)webSocket:(SRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean {
    NSLog(@"連接斷開，清空socket對象，清空該清空的東西，還有關閉心跳！");
}

5.收到服務器發來的數據會調用這個方法

- (void)webSocket:(SRWebSocket *)webSocket didReceiveMessage:(id)message  {
    NSLog(@"收到數據了，注意 message 是 id 類型的，學過C語言的都知道，id 是 (void *)  
        void* 就厲害了，二進制數據都可以指著，不詳細解釋 void* 了");
    NSLog(@"我這後台約定的 message 是 json 格式數據
        收到數據，就按格式解析吧，然後把數據發給調用層");
}

6.向服務器發送數據

發送的時候可能斷網，可能socket還在連接，要判斷一些情況，寫在下面了

發送邏輯是，我有一個 socketQueue 的串行隊列，發送請求會加到這個隊列裡，然後一個一個發出去，如果掉線了，重連連上後繼續發送，對調用層透明，調用層不需要知道網絡斷開了。

- (void)sendData:(id)data {
    WEAKSELF(ws);
    dispatch_async(self.socketQueue, ^{
        if (ws.socket != nil) {
            // 只有 SR_OPEN 開啟狀態才能調 send 方法啊，不然要崩
            if (ws.socket.readyState == SR_OPEN) {
                [ws.socket send:data];    // 發送數據

            } else if (ws.socket.readyState == SR_CONNECTING) {
                NSLog(@"正在連接中，重連後其他方法會去自動同步數據");
                // 每隔2秒檢測一次 socket.readyState 狀態，檢測 10 次左右
                // 只要有一次狀態是 SR_OPEN 的就調用 [ws.socket send:data] 發送數據
                // 如果 10 次都還是沒連上的，那這個發送請求就丟失了，這種情況是服務器的問題了，小概率的
                // 代碼有點長，我就寫個邏輯在這裡好了

            } else if (ws.socket.readyState == SR_CLOSING || ws.socket.readyState == SR_CLOSED) {
                // websocket 斷開了，調用 reConnect 方法重連
                [ws reConnect:^{
                    NSLog(@"重連成功，繼續發送剛剛的數據");
                    [ws.socket send:data];
                }];
            }
        } else {
            NSLog(@"沒網絡，發送失敗，一旦斷網 socket 會被我設置 nil 的");
            NSLog(@"其實最好是發送前判斷一下網絡狀態比較好，我寫的有點晦澀，socket==nil來表示斷網");
        }
    });
}

7.心跳機制

心跳機制就不難了，開個定時器，問下後台要每隔多少秒發送一次心跳請求就好了。然後注意，斷網了或者socket斷開的時候把心跳關一下，省資源，不然都斷網了，還在循環發心跳，浪費CPU和電量。