接入WebSocket

閒扯

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

爲何用 WebSocket

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

用法

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

1.建立和設置代理對象

1 2 3 4 5 6

SRWebSocket *socket = [[SRWebSocket alloc] initWithURLRequest: [NSURLRequest requestWithURL:[NSURL URLWithString:@ "http://ip地址:端口" ]];   socket.delegate = self;     // 實現這個 SRWebSocketDelegate 協議啊   [socket open];     // open 就是直接鏈接了

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

1 2 3

- ( void )webSocketDidOpen:(SRWebSocket *)webSocket {      NSLog(@ "鏈接成功，能夠馬上登陸你公司後臺的服務器了，還有開啓心跳" ); }

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

1 2 3 4 5 6 7

- ( 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] 客戶端主動關閉，斷開多是斷網了，被動斷開的。

1 2 3

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

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

1 2 3 4 5 6

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

6.向服務器發送數據

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

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

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28

- ( 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和電量。