Jetfire是OC版本;Starscream是swift版本。 Jetfire--github地址
当初老大让我把github上的开源项目Rocketchat-swift翻译成OC时可把我这个小菜鸟愁死了，一没文档，二找不到资源。幸好机智如我，去cocopods查了下他们使用的websocket协议的封装三方是Starscream，github上还附加了OC版的(这个真要赞一下)。
下面上干货
先导入类库，可在github上下载拖入，也可cocopods search jetfire。这个不赘述。
JFRWebSocket提供了简单的发送方法以及连接、断开、收消息代理方法,而实际运用的时候业务逻辑要复杂的多，因此再做一层封装。
新建一个Socket类 引入 JFRWebSocket.h
声明socket属性
@property (strong, nonatomic) JFRWebSocket *socket;
声明一个init方法 用于app刚启动时获得url
- (void)initWithToken:(NSString *)token {
if (token == nil) {
//跳转注册页面
} else {
_url = deaultUrl; //发送请求，获得URL
self.socket = [[JFRWebSocket alloc]initWithURL:[NSURL URLWithString:_url] protocols:@[@"chat",@"superchat"]];
self.socket.delegate = self;
[self.socket connect];
}
}
然后是连接方法与断开方法
#pragma mark-----连接方法-----------
- (void)connect {
if (self.isConnected) {
if (_timer != nil) {
[_timer invalidate];//该定时器用于自动重连
}
}else{
[self.socket connect];
}
}
#pragma mark-----断开连接方法-----------
- (void)disconnect {
[self.socket disconnect];
}
设置 自动重连
#pragma mark-----设置自动重连-----------
-(void)setTYSocketAutoConnect:(NSTimeInterval)timeInterval{
if (!_isConnected) {
[self connect];
}
_isAutoConnect = YES;
_timeInterval = timeInterval;
}
然后在连接断开的代理里进行重连
-(void)websocketDidDisconnect:(JFRWebSocket*)socket error:(NSError*)error{
_isConnected = NO;
if ([self.delegate respondsToSelector:@selector(socketDidDisconnect:error:)]) {
[self.delegate socketDidDisconnect:self error:error];
}
if (_isAutoConnect) {
[_timer invalidate];
_timer = [NSTimer scheduledTimerWithTimeInterval:_timeInterval target:self selector:@selector(connect) userInfo:nil repeats:YES];
[_timer fire];
[self connect];
}
}
接下来是发送消息的方法 （ JFRWebSocket提供了发送字符串与发送二进制流的方法 本文中因为业务关系 封装了一个消息类用于通信 ） 为了方便我们得到发送之后的反馈 这里用了一个block
声明一个block
@property (nonatomic, copy) void (^sendBack)();
发送方法
#pragma mark-----Message发送消息-----------
- (void)sendMessage:(Message *)message callBack:(void (^)(NSString* string))callBack{
if (callBack) {
self.sendBack = ^(NSString *string){
callBack(string);
};
}
if (self.socket.isConnected) {
NSDictionary *dict = [message getDictionaryFromMessage:message];
NSLog(@"发送消息%@", dict);
NSData *data = [NSJSONSerialization dataWithJSONObject:dict options:NSJSONWritingPrettyPrinted error:nil];
[self.socket writeData:data];
}
}
}
然后在JFRWebSocket 收消息的代理里 调用callBack
-(void)websocket:(JFRWebSocket*)socket didReceiveData:(NSData*)data {NSDictionary *dictionary = [NSJSONSerialization JSONObjectWithData:data options:NSJSONReadingMutableContainers error:nil];
if (dictionary) {
Message *messageObject = [[Message alloc]init];
[messageObject setValuesForKeysWithDictionary:dictionary];
//这此处判断消息类型来进行不同操作 本文只举一个栗子
if ([messageObject.type isEqualToString:@"SystemMsgOrder"]) {
if ([self.delegate respondsToSelector:@selector(socket:didReceiveMessage:)]) {
[self.delegate socket:self didReceiveOrderMessage:messageObject];
}
}
}else{
NSString *str = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];//本文后台返回的消息是字符串形式
if ([str hasPrefix:@"sendback"]) {//判断是发送回调
if (self.sendBack) {
self.sendBack(str);
}
}
}
}
到此完成了基本业务封装。如有问题欢迎交流