iOS Websocket使用指南

文章原创如需转载，请注明出处”本文首发于陕西毅杰”;

背景介绍

在消控项目中,实时监测,电气监管未来可能会根据业务,实行主动推送数据,APP接收到WSS的推送数据,实时更新页面;

框架介绍

facebook/SocketRocket

Facebook开源的可用于iOS,Mac Os系统的socket库
支持自动管理socket的生命周期;
支持HTTP,HTTPS;
发送 ping,响应 pong 事件;

如何使用

需要跟服务端约定心跳,并且约定发送的数据,即需要每隔几秒告诉服务端,客户端在线,并且有新的数据推给客户端;
销毁心跳,当页面销毁后,销毁心跳,断开Websocket;

tips如果数据量过大,可以采用 Zip 压缩数据;

Pod导入

pod 'SocketRocket'

关键代码

<SRWebSocketDelegate>//遵循协议
{
NSTimer * heartBeat;//心跳包
}
@property (nonatomic,strong) SRWebSocket *socket;//socket

#pragma mark - Socket
- (void)setUpMarkertSocket {
self.socket = [[SRWebSocket alloc] initWithURLRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"wss://api.test.com/ws"]]];
NSLog(@"请求的websocket地址：%@",self.socket.url.absoluteString);
self.socket.delegate = self;
[self.socket open];
}
#pragma mark - SRWebSocketDelegate
- (void)webSocket:(SRWebSocket *)webSocket didReceiveMessage:(id)message {
if (webSocket == self.socket) {
//接收到推送的新数据message,可以是json,可以是data等
}
}
- (void)webSocketDidOpen:(SRWebSocket *)webSocket {
NSLog(@"socket连接成功");
[self initHeartBeat];//初始化心跳
}
- (void)webSocket:(SRWebSocket *)webSocket didFailWithError:(NSError *)error {
NSLog(@"socket连接失败:%@",error);
}
- (void)webSocket:(SRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean {
NSLog(@"socket连接关闭:%@",reason);
}
- (void)webSocket:(SRWebSocket *)webSocket didReceivePong:(NSData *)pongPayload {
NSData * jsonData = [LFCGzipUtility ungzipData:pongPayload];
id resObj = [NSJSONSerialization JSONObjectWithData:jsonData options:NSJSONReadingMutableContainers error:nil];
NSLog(@"socket连接接收pong:%@",resObj);
}
- (void)checkHeartWithPongData:(NSMutableDictionary *)pongData {
[self sendScoketParams:pongData];
}
- (void)sendRequestParams {
//发送心跳 和后台可以约定发送什么内容  一般可以调用ping  我这里根据后台的要求 发送了data给他
//在成功后需要做的操作。。。
NSMutableDictionary* param = @{
@"sub":[NSString stringWithFormat: @"test"],//
@"id": @"id10" //行情
}.mutableCopy;
[self sendScoketParams:param];
}
- (void)sendScoketParams:(NSMutableDictionary *)param {
NSString* dataStr = [ToolUtil dictionaryConvertToJsonData:param];
@weakify(self);
dispatch_queue_t queue =  dispatch_queue_create("zy", NULL);
dispatch_async(queue, ^{
@strongify(self);
if (self.socket != nil) {
// 只有 SR_OPEN 开启状态才能调 send 方法啊，不然要崩
if (self.socket.readyState == SR_OPEN) {
[self.socket send:dataStr];    // 发送数据
} else if (self.socket.readyState == SR_CONNECTING) {
NSLog(@"正在连接中，重连后其他方法会去自动同步数据");
// 每隔2秒检测一次 socket.readyState 状态，检测 10 次左右
// 只要有一次状态是 SR_OPEN 的就调用 [ws.socket send:data] 发送数据
// 如果 10 次都还是没连上的，那这个发送请求就丢失了，这种情况是服务器的问题了，小概率的
// 代码有点长，我就写个逻辑在这里好了
// [self reConnect];
} else if (self.socket.readyState == SR_CLOSING || self.socket.readyState == SR_CLOSED) {
// websocket 断开了，调用 reConnect 方法重连
NSLog(@"重连");
[self setUpMarkertSocket];
}
} else {
NSLog(@"没网络，发送失败，一旦断网 socket 会被我设置 nil 的");
NSLog(@"其实最好是发送前判断一下网络状态比较好，我写的有点晦涩，socket==nil来表示断网");
}
});
}

心跳管理

//初始化心跳
- (void)initHeartBeat {
dispatch_main_async_safe(^{
[self destoryHeartBeat];
[self sendRequestParams];
//心跳设置为3分钟，NAT超时一般为5分钟
heartBeat = [NSTimer timerWithTimeInterval:10 target:self selector:@selector(sendRequestParams) userInfo:nil repeats:YES];
//和服务端约定好发送什么作为心跳标识，尽可能的减小心跳包大小
[[NSRunLoop currentRunLoop] addTimer:heartBeat forMode:NSRunLoopCommonModes];
})
}
//取消心跳
- (void)destoryHeartBeat {
dispatch_main_async_safe(^{
if (heartBeat) {
if ([heartBeat respondsToSelector:@selector(isValid)]){
if ([heartBeat isValid]){
[heartBeat invalidate];
heartBeat = nil;
}
}
}
})
}

使用遇到问题,欢迎留言讨论;

☛兄dei，请我喝杯茶☚

Websocket初见