iOS通过SocketRocket实现websocket的即时聊天

之前公司的即时聊天用的是常轮循，一直都觉得很不科学，最近后台说配置好了socket服务器，我高兴地准备用asyncsocket，但是告诉我要用websocket，基于HTML5的，HTML5中提出了一种新的双向通信协议--WebSocket，本文尝试采用这种技术来实现以上的实时聊天功能。

在搜索了很多资料后，用square大神的SocketRocket进行实现，会比较简单，同时URL和端口，发送消息参数需要和后台约定好。

首先pod导入SocketRocket

platform :ios, '7.0'
pod 'SocketRocket', '~> 0.5.0'

然后在搭建一个最简单的页面，只有一个输入框和button

在控制器中导入头文件

#import <SocketRocket/SRWebSocket.h>

创建

SRWebSocket *webSocket;

简单点，在viewDidLoad中实例化并且设置代理，链接URL和规定端口号，

webSocket.delegate = nil;

[webSocket close];

webSocket = [[SRWebSocket alloc] initWithURLRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"你的服务器URL和端口号"]]];

webSocket.delegate = self;

NSLog(@"Opening Connection...");

[webSocket open];

记得要遵守协议<SRWebSocketDelegate>，实现delegate方法

#pragma mark - SRWebSocketDelegate

- (void)webSocketDidOpen:(SRWebSocket *)webSocket;{

NSLog(@"Websocket Connected");

NSError *error;

NSData *jsonData = [NSJSONSerialization dataWithJSONObject:@{@"id":@"chat",@"clientid":@"hxz",@"to":@""} options:NSJSONWritingPrettyPrinted error:&error];

NSString *jsonString = [[NSString alloc] initWithData:jsonData encoding:NSUTF8StringEncoding];

[webSocket send:jsonString];

}

- (void)webSocket:(SRWebSocket *)webSocket didFailWithError:(NSError *)error;{

NSLog(@":( Websocket Failed With Error %@", error);

webSocket = nil;

}

- (void)webSocket:(SRWebSocket *)webSocket didReceiveMessage:(id)message;{

NSLog(@"Received \"%@\"", message);

}

- (void)webSocket:(SRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean;{

NSLog(@"WebSocket closed");

webSocket = nil;

}

- (IBAction)sendMessage:(id)sender {

NSError *error;

NSData *jsonData = [NSJSONSerialization dataWithJSONObject:@{@"id":@"chat",@"clientid":@"hxz",@"to":@"mary",@"msg":@{@"type":@"0",@"content":self.textfield.text}} options:NSJSONWritingPrettyPrinted error:&error];

NSString *jsonString = [[NSString alloc] initWithData:jsonData encoding:NSUTF8StringEncoding];

[webSocket send:jsonString];

}

其中webSocketDidOpen是在链接服务器成功后回调的方法，在这里发送一次消息，把id 名字发送到服务器，告知服务器，

在send方法中有两个选择：

// Send a UTF8 String or Data.

- (void)send:(id)data;

// Send Data (can be nil) in a ping message.

- (void)sendPing:(NSData *)data;

第一个是需要发送JSON字符串格式的Data，必须把对象转换成JSON字符串格式，否则报错，第二种是发送NSData类型，而且根据注释可以为nil

在文本框输入消息，发送后在对方消息列表显示成功：

对方发送消息给我这端时，didReceiveMessage方法接受到消息后会执行，输出消息内容：

完成~