接入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和电量。