閑扯
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:
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和電量。