Websocket实现即时通讯

前言

关于我和WebSocket的缘：我从大二在计算机网络课上听老师讲过之后，第一次使用就到了毕业之后的第一份工作。直到最近换了工作，到了一家是含有IM社交聊天功能的app的时候，我觉得我现在可以谈谈我对WebSocket/Socket的一些看法了。要想做IM聊天app，就不得不理解WebSocket和Socket的原理了，听我一一道来。

1.WebSocket使用场景

2.WebSocket诞生由来

3.谈谈WebSocket协议原理

4.WebSocket 和 Socket的区别与联系

5.iOS平台有哪些WebSocket和Socket的开源框架

6.iOS平台如何实现WebSocket协议

一.WebSocket的使用场景

1.社交聊天

最著名的就是微信，QQ，这一类社交聊天的app。这一类聊天app的特点是低延迟，高即时。即时是这里面要求最高的，如果有一个紧急的事情，通过IM软件通知你，假设网络环境良好的情况下，这条message还无法立即送达到你的客户端上，紧急的事情都结束了，你才收到消息，那么这个软件肯定是失败的。

2.弹幕

说到这里，大家一定里面想到了A站和B站了。确实，他们的弹幕一直是一种特色。而且弹幕对于一个视频来说，很可能弹幕才是精华。发弹幕需要实时显示，也需要和聊天一样，需要即时。

3.多玩家游戏

4.协同编辑

现在很多开源项目都是分散在世界各地的开发者一起协同开发，此时就会用到版本控制系统，比如Git，SVN去合并冲突。但是如果有一份文档，支持多人实时在线协同编辑，那么此时就会用到比如WebSocket了，它可以保证各个编辑者都在编辑同一个文档，此时不需要用到Git，SVN这些版本控制，因为在协同编辑界面就会实时看到对方编辑了什么，谁在修改哪些段落和文字。

5.股票基金实时报价

金融界瞬息万变——几乎是每毫秒都在变化。如果采用的网络架构无法满足实时性，那么就会给客户带来巨大的损失。几毫秒钱股票开始大跌，几秒以后才刷新数据，一秒钟的时间内，很可能用户就已经损失巨大财产了。

6.体育实况更新

全世界的球迷，体育爱好者特别多，当然大家在关心自己喜欢的体育活动的时候，比赛实时的赛况是他们最最关心的事情。这类新闻中最好的体验就是利用Websocket达到实时的更新！

7.视频会议/聊天

视频会议并不能代替和真人相见，但是他能让分布在全球天涯海角的人聚在电脑前一起开会。既能节省大家聚在一起路上花费的时间，讨论聚会地点的纠结，还能随时随地，只要有网络就可以开会。

8.基于位置的应用

越来越多的开发者借用移动设备的GPS功能来实现他们基于位置的网络应用。如果你一直记录用户的位置(比如运行应用来记录运动轨迹)，你可以收集到更加细致化的数据。

9.在线教育

在线教育近几年也发展迅速。优点很多，免去了场地的限制，能让名师的资源合理的分配给全国各地想要学习知识的同学手上，Websocket是个不错的选择，可以视频聊天、即时聊天以及其与别人合作一起在网上讨论问题...

10.智能家居

这也是我一毕业加入的一个伟大的物联网智能家居的公司。考虑到家里的智能设备的状态必须需要实时的展现在手机app客户端上，毫无疑问选择了Websocket。

11.总结

从上面我列举的这些场景来看，一个共同点就是，高实时性！

二.WebSocket诞生由来

1.最开始的轮询Polling阶段

这种方式下，是不适合获取实时信息的，客户端和服务器之间会一直进行连接，每隔一段时间就询问一次。客户端会轮询，有没有新消息。这种方式连接数会很多，一个接受，一个发送。而且每次发送请求都会有Http的Header，会很耗流量，也会消耗CPU的利用率。

2.改进版的长轮询Long polling阶段

长轮询是对轮询的改进版，客户端发送HTTP给服务器之后，有没有新消息，如果没有新消息，就一直等待。当有新消息的时候，才会返回给客户端。在某种程度上减小了网络带宽和CPU利用率等问题。但是这种方式还是有一种弊端：例如假设服务器端的数据更新速度很快，服务器在传送一个数据包给客户端后必须等待客户端的下一个Get请求到来，才能传递第二个更新的数据包给客户端，那么这样的话，客户端显示实时数据最快的时间为2×RTT（往返时间），而且如果在网络拥塞的情况下，这个时间用户是不能接受的，比如在股市的的报价上。另外，由于http数据包的头部数据量往往很大（通常有400多个字节），但是真正被服务器需要的数据却很少（有时只有10个字节左右），这样的数据包在网络上周期性的传输，难免对网络带宽是一种浪费。

3.WebSocket诞生

现在急需的需求是能支持客户端和服务器端的双向通信，而且协议的头部又没有HTTP的Header那么大，于是，Websocket就诞生了！

上图就是Websocket和Polling的区别，从图中可以看到Polling里面客户端发送了好多Request，而下图，只有一个Upgrade，非常简洁高效。至于消耗方面的比较就要看下图了

上图中，我们先看蓝色的柱状图，是Polling轮询消耗的流量，

Use case A: 1,000 clients polling every second: Network throughput is (871 x 1,000) = 871,000 bytes = 6,968,000 bits per second (6.6 Mbps)

Use case B: 10,000 clients polling every second: Network throughput is (871 x 10,000) = 8,710,000 bytes = 69,680,000 bits per second (66 Mbps)

Use case C: 100,000 clients polling every 1 second: Network throughput is (871 x 100,000) = 87,100,000 bytes = 696,800,000 bits per second (665 Mbps)

而Websocket的Frame是 just two bytes of overhead instead of 871，仅仅用2个字节就代替了轮询的871字节！

Use case A: 1,000 clients receive 1 message per second: Network throughput is (2 x 1,000) = 2,000 bytes = 16,000 bits per second (0.015 Mbps)

Use case B: 10,000 clients receive 1 message per second: Network throughput is (2 x 10,000) = 20,000 bytes = 160,000 bits per second (0.153 Mbps)

Use case C: 100,000 clients receive 1 message per second: Network throughput is (2 x 100,000) = 200,000 bytes = 1,600,000 bits per second (1.526 Mbps)

相同的每秒客户端轮询的次数，当次数高达10W/s的高频率次数的时候，Polling轮询需要消耗665Mbps，而Websocket仅仅只花费了1.526Mbps，将近435倍！！

三.谈谈WebSocket协议原理

Websocket是应用层第七层上的一个应用层协议，它必须依赖 HTTP 协议进行一次握手，握手成功后，数据就直接从 TCP 通道传输，与 HTTP 无关了。

Websocket的数据传输是frame形式传输的，比如会将一条消息分为几个frame，按照先后顺序传输出去。这样做会有几个好处：

1）大数据的传输可以分片传输，不用考虑到数据大小导致的长度标志位不足够的情况。

2）和http的chunk一样，可以边生成数据边传递消息，即提高传输效率。

QQ截图20160526161900.png

四.WebSocket 和 Socket的区别与联系

首先，Socket 其实并不是一个协议。它工作在 OSI 模型会话层（第5层），是为了方便大家直接使用更底层协议（一般是 TCP 或 UDP ）而存在的一个抽象层。Socket是对TCP/IP协议的封装，Socket本身并不是协议，而是一个调用接口(API)。

Socket通常也称作”套接字”，用于描述IP地址和端口，是一个通信链的句柄。网络上的两个程序通过一个双向的通讯连接实现数据的交换，这个双向链路的一端称为一个Socket，一个Socket由一个IP地址和一个端口号唯一确定。应用程序通常通过”套接字”向网络发出请求或者应答网络请求。

Socket在通讯过程中，服务端监听某个端口是否有连接请求，客户端向服务端发送连接请求，服务端收到连接请求向客户端发出接收消息，这样一个连接就建立起来了。客户端和服务端也都可以相互发送消息与对方进行通讯，直到双方连接断开。

所以基于WebSocket和基于Socket都可以开发出IM社交聊天类的app

五.iOS平台有哪些WebSocket和Socket的开源框架

Socket开源框架有：CocoaAsyncSocket，socketio/socket.io-client-swift

WebSocket开源框架有：facebook/SocketRocket，tidwall/SwiftWebSocket

六.iOS平台如何实现WebSocket协议

Talk is cheap。Show me the code ——Linus Torvalds

我们今天来看看facebook/SocketRocket的实现方法

首先这是SRWebSocket定义的一些成员变量

@property (nonatomic, weak) id  delegate;
/**
 A dispatch queue for scheduling the delegate calls. The queue doesn't need be a serial queue.
 If `nil` and `delegateOperationQueue` is `nil`, the socket uses main queue for performing all delegate method calls.
 */
@property (nonatomic, strong) dispatch_queue_t delegateDispatchQueue;
/**
 An operation queue for scheduling the delegate calls.
 If `nil` and `delegateOperationQueue` is `nil`, the socket uses main queue for performing all delegate method calls.
 */
@property (nonatomic, strong) NSOperationQueue *delegateOperationQueue;
@property (nonatomic, readonly) SRReadyState readyState;
@property (nonatomic, readonly, retain) NSURL *url;
@property (nonatomic, readonly) CFHTTPMessageRef receivedHTTPHeaders;
// Optional array of cookies (NSHTTPCookie objects) to apply to the connections
@property (nonatomic, copy) NSArray *requestCookies;
// This returns the negotiated protocol.
// It will be nil until after the handshake completes.
@property (nonatomic, readonly, copy) NSString *protocol;

下面这些是SRWebSocket的一些方法

// Protocols should be an array of strings that turn into Sec-WebSocket-Protocol.
- (instancetype)initWithURLRequest:(NSURLRequest *)request;
- (instancetype)initWithURLRequest:(NSURLRequest *)request protocols:(NSArray *)protocols;
- (instancetype)initWithURLRequest:(NSURLRequest *)request protocols:(NSArray *)protocols allowsUntrustedSSLCertificates:(BOOL)allowsUntrustedSSLCertificates;
// Some helper constructors.
- (instancetype)initWithURL:(NSURL *)url;
- (instancetype)initWithURL:(NSURL *)url protocols:(NSArray *)protocols;
- (instancetype)initWithURL:(NSURL *)url protocols:(NSArray *)protocols allowsUntrustedSSLCertificates:(BOOL)allowsUntrustedSSLCertificates;
// By default, it will schedule itself on +[NSRunLoop SR_networkRunLoop] using defaultModes.
- (void)scheduleInRunLoop:(NSRunLoop *)aRunLoop forMode:(NSString *)mode;
- (void)unscheduleFromRunLoop:(NSRunLoop *)aRunLoop forMode:(NSString *)mode;
// SRWebSockets are intended for one-time-use only.  Open should be called once and only once.
- (void)open;
- (void)close;
- (void)closeWithCode:(NSInteger)code reason:(NSString *)reason;
///--------------------------------------
#pragma mark Send
///--------------------------------------
//下面是4个发送的方法
/**
 Send a UTF-8 string or binary data to the server.
 @param message UTF-8 String or Data to send.
 @deprecated Please use `sendString:` or `sendData` instead.
 */
- (void)send:(id)message __attribute__((deprecated("Please use `sendString:` or `sendData` instead.")));
- (void)sendString:(NSString *)string;
- (void)sendData:(NSData *)data;
- (void)sendPing:(NSData *)data;
@end

对应5种状态的代理方法

///--------------------------------------
#pragma mark - SRWebSocketDelegate
///--------------------------------------
@protocol SRWebSocketDelegate - (void)webSocket:(SRWebSocket *)webSocket didReceiveMessage:(id)message;
@optional
- (void)webSocketDidOpen:(SRWebSocket *)webSocket;
- (void)webSocket:(SRWebSocket *)webSocket didFailWithError:(NSError *)error;
- (void)webSocket:(SRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean;
- (void)webSocket:(SRWebSocket *)webSocket didReceivePong:(NSData *)pongPayload;
// Return YES to convert messages sent as Text to an NSString. Return NO to skip NSData -> NSString conversion for Text messages. Defaults to YES.
- (BOOL)webSocketShouldConvertTextFrameToString:(SRWebSocket *)webSocket;
@end

didReceiveMessage方法是必须实现的，用来接收消息的。

下面4个did方法分别对应着Open，Fail，Close，ReceivePong不同状态的代理方法

方法就上面这些了，我们实际来看看代码怎么写

先是初始化Websocket连接，注意此处ws://或者wss://连接有且最多只能有一个，这个是Websocket协议规定的

self.ws = [[SRWebSocket alloc] initWithURLRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:[NSString 
stringWithFormat:@"%@://%@:%zd/ws", serverProto, serverIP, serverPort]]]];
    self.ws.delegate = delegate;
    [self.ws open];

发送消息

[self.ws send:message];

接收消息以及其他3个代理方法

//这个就是接受消息的代理方法了，这里接受服务器返回的数据，方法里面就应该写处理数据，存储数据的方法了。
- (void)webSocket:(SRWebSocket *)webSocket didReceiveMessage:(id)message
{
    NSDictionary *data = [NetworkUtils decodeData:message];
    if (!data)
        return;
}
//这里是Websocket刚刚Open之后的代理方法。就想微信刚刚连接中，会显示连接中，当连接上了，就不显示连接中了，取消显示连接的方法就应该写在这里面
- (void)webSocketDidOpen:(SRWebSocket *)webSocket
{
    // Open = silent ping
    [self.ws receivedPing];
}
//这是关闭Websocket的代理方法
- (void)webSocket:(SRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean
{
    [self failedConnection:NSLS(Disconnected)];
}
//这里是连接Websocket失败的方法，这里面一般都会写重连的方法
- (void)webSocket:(SRWebSocket *)webSocket didFailWithError:(NSError *)error
{
    [self failedConnection:NSLS(Disconnected)];
}

最后

以上就是我想分享的一些关于Websocket的心得，文中如果有错误的地方，欢迎大家指点！一般没有微信QQ那么大用户量的app，用Websocket应该都可以完成IM社交聊天的任务。当用户达到亿级别，应该还有很多需要优化，优化性能各种的吧。

最后，微信和QQ的实现方法也许并不是只用Websocket和Socket这么简单，也许是他们自己开发的一套能支持这么大用户，大数据的，各方面也都优化都最优的方法。如果有开发和微信和QQ的大神看到这篇文章，可以留言说说看你们用什么方式实现的，也可以和我们一起分享，我们一起学习！我先谢谢大神们的指点了！

1，Android 客户端使用需要配置网络权限；

2，需要写一个自己的client类来继承WebsocketClient;实现websocket的状态回调和新消息的解析动作；

3，需要监控自己的client的链接状态，维持长链接；

4，发送和接收

下面贴出部分相关代码；

网络权限的不用说了吧！

client类：

常用状态回调方法有4个；

自己可以在对应的函数里面做响应的处理，比如当链接发生错误时要重新去打开该链接，收到消息时即时的保存聊天记录和发送系统通知来提醒用户查看新消息等等；

[html] view plain copy

<pre name="code" class="html"><span style="font-size:18px;">public class TestClient extends WebSocketClient {
public TestClient(URI serverURI) {
super(serverURI);
}
/***
* 链接关闭
*/
@Override
public void onClose(int arg0, String arg1, boolean arg2) {
}
/***
* 链接发生错误
*/
@Override
public void onError(Exception arg0) {
}
/**
* 新消息
*/
@Override
public void onMessage(String arg0) {
}
/***
* 链接打开
*/
@Override
public void onOpen(ServerHandshake arg0) {
// TODO Auto-generated method stub
}
}
</span>

下面是我聊天部分代码，有离线消息/PC同步/多对多的聊天；

仅供参考；

[java] view plain copy

/***
* <h2>WebSocket Android 客户端</h2>
* <ol>
* <li>Socket链接打开回调 {@link WebSocket#onOpen(ServerHandshake)},此处有
* {@link SocketConstant#ON_OPEN} 广播发出;
* <li>Socket链接出现异常错误时回调 {@link WebSocket#onError(Exception)},此处有
* {@link SocketConstant#ON_ERROR}广播发出;
* <li>Socket链接关闭回调 {@link WebSocket #onClose(int, String, boolean)},此处有
* {@link SocketConstant#ON_CLOSES}广播发出;
* <li>Socket链接接收消息回调 {@link WebSocket#onMessage(String)}
* ,此处做收消息的逻辑的处理;包括发送消息服务器返回的发送结果,PC端同步接收的新消息,及外来的新消息;
* <li>检测是否有消息遗漏 {@link WebSocket#checkMsgWebId(String, String)},参数为联系人和webId;
* <li>取得正在桌面运行的activity的名称 {@link WebSocket#getRunningActivityName()}
* <li>接收到的消息的处理 {@link WebSocket#messageHandle(MessageEntity, String)}
* ,参数为消息实体和消息类型 (buyer/server)
* <li>发送新消息的系统通知 {@link WebSocket#sendNotification(String)},参数为联系人;
* <li>保存离线消息 {@link WebSocket#saveOffLineMsg(HashMap)},参数为接收到的离线消息集合;
* <li>保存从服务端获取的联系人的webId {@link WebSocket#saveContactsWebID(HashMap)}
* ,参数为以联系人为key以最大webId为值得map集合;
* </ol>
*
* @author li'mingqi <a> 2014-3-19</a>
*
*/
public class WebSocket extends WebSocketClient {
// 登陆返回的back_type字段
public static final String LOGIN_RETURN_TYPE = "login";
// 发送信息的back_type字段
public static final String SEND_RETURN_TYPE = "send_result";
// 接收信息的back_type字段
public static final String RECEIVER_RETURN_TYPE = "msg";
// 接收客服的信息的back_type字段
public static final String GET_SERVER_RETURN_TYPE = "server_info";
// 接收服务端返回对应联系人的最大顺序ID
public static final String CONTACTS_MAX_WEBID_TYPE = "max_id_return";
// 接收用户的离线消息
public static final String USER_OFFLINE_MSG_TYPE = "offline";
// 上下文对象
private Context mContext;
// socket返回json解析类对象
private WebSocketParser mParser;
// 系统通知管理
public NotificationManager mNotificationManager;
// 系统通知
private Notification mNoti;
// 意图
private PendingIntent mIntent;
// 该系统通知的 id
public static final int NOTIFICATION_ID = 100;
@SuppressWarnings("deprecation")
public SGWebSocket(Context context, URI serverUri, Draft draft) {
super(serverUri, draft);
this.mContext = context;
//新消息的解析类
this.mParser = WebSocketParser.getInstance();
//收到新消息发送的通知
this.mNotificationManager = (NotificationManager) this.mContext
.getSystemService(Context.NOTIFICATION_SERVICE);
this.mNoti = new Notification(R.drawable.system_info, "您有新消息！",
System.currentTimeMillis());
}
/***
* send broadcast <SGSocketConstant>ON_CLOSES filter if this socket closed
* socket 发生关闭时发送的广播，若想提示，可以接受并处理
*
*/
@Override
public void onClose(int arg0, String arg1, boolean arg2) {
// 更改保存的链接状态
UserInfoUtil.saveSocket(mContext, false);
mNotificationManager.cancelAll();
Intent intent = new Intent(SocketConstant.ON_CLOSES);
intent.putExtra(SocketConstant.ON_CLOSES, arg1.toString());
mContext.sendBroadcast(intent);
}
/***
* send broadcast <SGSocketConstant>ON_ERROR filter if this socket has error
* socket 发生错误发送的广播，若想提示，可以接受并处理
*
*/
@Override
public void onError(Exception arg0) {
Intent intent = new Intent(SGSocketConstant.ON_ERROR);
intent.putExtra(SGSocketConstant.ON_ERROR, arg0.toString());
mContext.sendBroadcast(intent);
this.close();
}
// 买家
public static final String MSG_BUYER_TYPE = "1";
// 客服
public static final String MSG_SERVCER_TYPE = "2";
// 游客
// public static final String MSG_RANDOM_TYPE = "3";
/***
* receiver message from server 1,登陆返回 type
* <WebSocket>LOGIN_RETURN_TYPE; 2,发送返回 type
* <WebSocket>SEND_RETURN_TYPE; 3,接收信息返回 type
* <WebSocket>RECEIVER_RETURN_TYPE;
*
* @throws InterruptedException
*/
@Override
public void onMessage(String content) {
// parser
try {
JSONObject object = new JSONObject(content);
Log.i("json", "卖家--" + object.toString());
String back_type = object.getString("back_type");
String activity = getRunningActivityName();
if (SEND_RETURN_TYPE.equals(back_type)) {// 发送具体消息时返回发送结果
// json解析
MessageEntity entity = mParser.sendMessageParser(mContext,
content);
if ("true".equals(entity.getSend_state())) {// 发送成功
// 判断是否是PC端发送的消息,若是PC端发送的消息,则在Android端做同步存储处理
// 1,首先判断数据库中是否包含该条信息
boolean has = MessageDB.getInstance(mContext)
.findMessageByMsgId(entity.get_id());
if (has) {
// Android端发送
MessageDB.getInstance(mContext).update(entity.get_id(),
true, entity.getReceiverTime(),
entity.getWebId());// 更新发送状态为已发送
} else {
// PC端发送,将该消息同步到Android端数据库
entity.setSend_state(SocketConstant.MSG_SEND_SUCCESS_STATE);
MessageDB.getInstance(mContext).insert(entity,
SocketConstant.MSG_TYPE_BUYER);// 卖家发送给买家的
// 通知聊天主页面,更新聊天列表
pcSynAndroid(activity);
}
// 检测是否有消息遗漏
checkMsgWebId(entity.getContacts(), entity.getWebId());
Log.i("miss", "发送返回或者PC同步--" + entity.getContacts() + "--"
+ entity.getWebId());
} else if ("false".equals(entity.getSend_state())) {// 发送失败
MessageDB.getInstance(mContext).update(entity.get_id(),
false, entity.getReceiverTime(), entity.getWebId());
Toast.makeText(mContext, entity.getErrorText(),
Toast.LENGTH_SHORT).show();
}
// 登陆返回记录session
} else if (LOGIN_RETURN_TYPE.equals(back_type)) {
KApplication.session = object.getString("session_id");
String str = object.getString("login_status");
if ("true".equals(str)) {
UserInfoUtil.saveSocket(mContext, true);
// 生成json请求字符串
String maxIdstring = SocketJsonUtil
.getContactsCurrentWebId(UserInfoUtil
.getUser(mContext)[0], "2", MessageDB
.getInstance(mContext)
.findAllContactsAndType());
// 登陆成功,向服务器索取联系人的最大webId
send(maxIdstring);
Log.i("send", maxIdstring);
} else if ("false".equals(str)) {
UserInfoUtil.saveSocket(mContext, false);
}
} else if (RECEIVER_RETURN_TYPE.equals(back_type)) {// 接收到的具体聊天的信息
// json解析
MessageEntity entity = mParser.receiverMessagePrser(mContext,
content);
// 判断数据库中是否有该条消息,有则不处理,无则处理消息;
if (!MessageDB.getInstance(mContext).findMessageByMsgId(
entity.get_id())) {
// 消息处理
if (MSG_BUYER_TYPE.equals(entity.getSenderType())) {
// 买家
messageHandle(entity, SocketConstant.MSG_TYPE_BUYER);
} else if (MSG_SERVCER_TYPE.equals(entity.getSenderType())) {
// 卖家,客服
messageHandle(entity, SocketConstant.MSG_TYPE_SERVER);
}
Log.i("miss", "没有该条消息");
// 检测是否有消息遗漏
checkMsgWebId(entity.getContacts(), entity.getWebId());
}
} else if (GET_SERVER_RETURN_TYPE.equals(back_type)) {// 获取闪聊客服返回的数据
// 客服
ServerEntity entity = mParser.serverInfoParser(content);// 客服对象
Intent intent = new Intent(SocketConstant.GET_SERVER_INFO);
intent.putExtra("server_info", entity);
mContext.sendBroadcast(intent);
} else if (CONTACTS_MAX_WEBID_TYPE.equals(back_type)) {
// 返回的联系人最大的消息id
HashMap<String, String> map = mParser.contactsMaxWebId(content);
// 将联系人和其最大webId存入临时集合
saveContactsWebID(map);
// 开始请求服务器，释放离线消息给客户端;
send(SocketJsonUtil.getOffLine(
UserInfoUtil.getUser(mContext)[0], "2"));
Log.i("send",
SocketJsonUtil.getOffLine(
UserInfoUtil.getUser(mContext)[0], "2"));
} else if (USER_OFFLINE_MSG_TYPE.equals(back_type)) {
// 用户的离线消息
HashMap<String, ArrayList<MessageEntity>> map = mParser
.offLineMsg(mContext, content);
// 将离线消息入库
saveOffLineMsg(map);
}
} catch (JSONException e) {
this.close();
}
}
/***
* send broadcast <SocketConstant>ON_OPEN filter if this socket opened
* socket 打开时发送的广播，若想提示，可以接受并处理
*
*/
@Override
public void onOpen(ServerHandshake arg0) {
Intent intent = new Intent(SGSocketConstant.ON_OPEN);
mContext.sendBroadcast(intent);
}
/***
* 检测正在运行tasktop的activity
* @return current running activity name
*
*/
private String getRunningActivityName() {
ActivityManager activityManager = (ActivityManager) mContext
.getSystemService(Context.ACTIVITY_SERVICE);
String runningActivity = activityManager.getRunningTasks(1).get(0).topActivity
.getClassName();
return runningActivity;
}
/***
* send notification for this contacts
* 发送通知
* @param contacts
*
*/
@SuppressWarnings("deprecation")
private void sendNotification(String contacts) {
Intent intent = new Intent(mContext, MainActivity.class);
mIntent = PendingIntent.getActivity(mContext, 100, intent, 0);
mNoti.flags = Notification.FLAG_AUTO_CANCEL;
mNoti.defaults = Notification.DEFAULT_VIBRATE;
mNoti.setLatestEventInfo(mContext, "标题", "您有新消息！", mIntent);
mNoti.contentView = new RemoteViews(mContext.getApplicationContext()
.getPackageName(), R.layout.notification_item);
mNoti.contentView.setTextViewText(R.id.noti_message, "收到来自" + contacts
+ "的新消息");
mNotificationManager.notify(NOTIFICATION_ID, mNoti);
}
/***
* 具体聊天收到的外来消息处理
*
* @param entity
* 消息实体
* @param messageType
* 消息类型(买家/客服)
*/
private void messageHandle(MessageEntity entity, String messageType) {
String activity = getRunningActivityName();
// 处于聊天的页面
if ("com.ui.activity.ManageChartActivity".equals(activity)) {
// 处于正在聊天对象的页面,将数据写入数据库，并发送广播更新页面数据
if (KApplication.crurentContacts.equals(entity.getContacts())) {
/**
* 接收到的消息,消息实体entity的send_state字段状态设置为
* MSG_SEND_SUCCESS_STATE(即201)
**/
entity.setSend_state(SocketConstant.MSG_SEND_SUCCESS_STATE);// 收到的信息,设置信息的状态
entity.setRead(SocketConstant.READ_STATE);
MessageDB.getInstance(mContext).insert(entity, messageType);// 将数据写入数据库,
Intent intent = new Intent(SocketConstant.NEW_MESSAGE);
intent.putExtra("newmsg", entity);
mContext.sendBroadcast(intent);
// 没有处于闪聊对象的页面，将数据写入数据库，发送系统通知
} else {
entity.setSend_state(SocketConstant.MSG_SEND_SUCCESS_STATE);// 收到的信息,设置信息的状态
entity.setRead(SocketConstant.DEFAULT_READ_STATE);
MessageDB.getInstance(mContext).insert(entity, messageType);
if (KApplication.sp.getBoolean(RefreshUtils.noteFlag, false)) {
sendNotification(entity.getContacts());
}
Intent intent = new Intent(
SocketConstant.RECEIVER_NEW_MESSAGE);
mContext.sendBroadcast(intent);
}
// 将数据写入数据库，发送系统通知
} else {
entity.setSend_state(SocketConstant.MSG_SEND_SUCCESS_STATE);// 收到的信息,设置信息的状态
entity.setRead(SocketConstant.DEFAULT_READ_STATE);
MessageDB.getInstance(mContext).insert(entity, messageType);
Intent intent = new Intent();
if ("com.ui.activity.ManageConversationActivity"
.equals(activity)
|| "com.ui.activity.MainActivity"
.equals(activity)) {
intent.setAction(SocketConstant.RECEIVER_NEW_MESSAGE);// 聊天页面
} else {
intent.setAction(SocketConstant.RECEIVER_NEW_MESSAGE_OTHER);// 其他页面
}
mContext.sendBroadcast(intent);
if (KApplication.sp.getBoolean(RefreshUtils.noteFlag, false)) {
sendNotification(entity.getContacts());
}
}
}
/***
* 电脑与手机同步信息
*
* @param currentActivity
*/
public void pcSynAndroid(String currentActivity) {
if ("com.iflashseller.ui.activity.ManageChartActivity"
.equals(currentActivity)) {
// 正好与该联系人对话的页面
Intent intent = new Intent(SocketConstant.CHART_ACTIVITY);
mContext.sendBroadcast(intent);
} else {
// 其他页面
Intent intent = new Intent(SocketConstant.GROUPS_ACTIVITY);
mContext.sendBroadcast(intent);
}
}
/***
* 检测是否有消息遗漏
*
* @param contacts
* 联系人
* @param webId
* 服务端给出的消息Id
*/
public void checkMsgWebId(String contacts, int webId) {
// 集合中含有该联系人
if (KApplication.webIds.containsKey(contacts)) {
Log.i("miss", "保存的--" + KApplication.webIds.get(contacts));
// 临时集合中保存的webId
int c = KApplication.webIds.get(contacts);
/***
* 如果新收到的消息的webId大于临时集合中保存的改联系人的webId,且他们之间的差值大于1,
* 则请求服务器推送疑似丢失的webId对应的消息
*/
if (webId > c && (webId - 1) != c) {
// id不连续
for (int i = c + 1; i < webId; i++) {
// 向服务器发送请求,获取遗漏的消息
String miss = SocketJsonUtil.getMissMsg(
UserInfoUtil.getUser(mContext)[0], "2", contacts,
"1", i + "");
this.send(miss);
Log.i("miss", miss);
}
/***
* 如果他们之间的差值正好为1,则修改临时集合的改联系人的webId,
*/
} else if (webId > c && (webId - 1) == c) {
KApplication.webIds.put(contacts, webId);
Log.i("miss", "修改的--" + contacts + "--" + webId);
}
/****
* 临时集合中没有改联系人的信息,则将该联系人的webId存入临时集合.
*/
} else {
KApplication.webIds.put(contacts, webId);
Log.i("miss", "新增--" + contacts + "--" + webId);
}
}
/***
* 将从服务端获取的联系人的webId存入临时集合
*
* @param map
*/
public void saveContactsWebID(HashMap<String, String> map) {
Iterator<Entry<String, String>> iter = map.entrySet().iterator();
while (iter.hasNext()) {
Entry<String, String> es = iter.next();
String contacts = es.getKey();
String maxWebID = es.getValue();
KApplication.webIds.put(contacts, Integer.parseInt(maxWebID));
}
}
/***
* 将离线消息入库
*
* @param map
*/
public void saveOffLineMsg(HashMap<String, ArrayList<MessageEntity>> map) {
Iterator<Entry<String, ArrayList<MessageEntity>>> iter = map.entrySet()
.iterator();
while (iter.hasNext()) {
ArrayList<MessageEntity> msgs = iter.next().getValue();
for (int i = 0; i < msgs.size(); i++) {
threadSleep(100);
MessageDB.getInstance(mContext).insert(msgs.get(i),
SocketConstant.MSG_TYPE_BUYER);
Log.i("write", "离线数据入库---" + msgs.get(i).toString());
}
/***
* 如果服务端一次释放的离线消息大于等于10条，则继续请求释放离线消息.
*/
if (msgs.size() >= 10) {
send(SocketJsonUtil.getOffLine(
UserInfoUtil.getUser(mContext)[0], "2"));
Log.i("send",
SocketJsonUtil.getOffLine(
UserInfoUtil.getUser(mContext)[0], "2"));
}
}
// 一轮消息入库结束,发送通知,更新UI;
mContext.sendBroadcast(new Intent(
SocketConstant.OFFLINE_MSG_RECEIVER_SUCCESS));
}
private void threadSleep(long time) {
try {
Thread.currentThread();
Thread.sleep(time);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}

至于数据库的一部分代码就不贴出了，无非是增删改查。

下面贴出部分监控链接状态的代码，以保证能即时的收到消息；

[java] view plain copy

public class LApplication extends Application {
public static String TAG = LApplication.class.getSimpleName();
/** 接收消息广播 **/
private LPullReceiver mPullReceiver;
/** 是否是正在登录 **/
public static boolean isLoging = false;
/** socket管理类 **/
private LPushManager mWebSocket;
@Override
public void onCreate() {
super.onCreate();
/***
* 注册接收消息的广播
*/
mPullReceiver = new LPullReceiver();
// 广播过滤
IntentFilter filter = new IntentFilter();
// 时钟信息发生变化
filter.addAction(Intent.ACTION_TIME_TICK);
// 开机广播
filter.addAction(Intent.ACTION_BOOT_COMPLETED);
// 网络状态发生变化
filter.addAction(ConnectivityManager.CONNECTIVITY_ACTION);
// 屏幕打开
filter.addAction(Intent.ACTION_SCREEN_ON);
// 注册广播
registerReceiver(mPullReceiver, filter);
// 实例化socket管理类
mWebSocket = new LPushManager(getApplicationContext());
// 应用重启一次，默认socket为关闭状态
LPushUser.saveSocket(getApplicationContext(), false);
// 默认链接没有被拒绝
LPushUser.saveConnect(getApplicationContext(), false);
// 1,获取当前时间
long currentTime = System.currentTimeMillis();
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
Date date = new Date(currentTime);
String time = sdf.format(date);
// 修改标记的时间,保证5分钟内链接一次
LPushUser.saveOpenTime(getApplicationContext(), time);
}
/**
* 广播接口类
* <ol>
* <li>接收时钟发生变化的广播
* <li>接收网络发生变化的广播
* <li>接收开机发送广播
* <li>接收用户登录后发送的广播
* </ol>
*
* @author li'mingqi
*
*/
class LPullReceiver extends BroadcastReceiver {
@SuppressLint("SimpleDateFormat")
@Override
public void onReceive(Context context, Intent intent) {
String action = intent.getAction();
if (Intent.ACTION_TIME_TICK.equals(action)
|| ConnectivityManager.CONNECTIVITY_ACTION.equals(action)
|| Intent.ACTION_BOOT_COMPLETED.equals(action)
|| Intent.ACTION_SCREEN_ON.equals(action)) {
if (LPushUser.GETLOG(getApplicationContext()))
Log.i("lpush",
"socket的链接状态----"
+ LPushUser
.getSocket(getApplicationContext())
+ "是否正在链接--" + isLoging + "----" + action);
// 当时钟或者网络发生变化或者socket关闭或者发生异常时或者开机时,判断socket连接是否出现异常
if (!LPushUser.getSocket(getApplicationContext()) && !isLoging
&& LPushUser.getSocketEnable(getApplicationContext())
&& !"".equals(IP) && !"".equals(PORT)
&& !LPushUser.getConnectState(getApplicationContext())) {
// 掉线,执行登录动作
if (LNetworkUtil.netIsEnable(getApplicationContext())) {
mWebSocket.secondMethod(IP, PORT);
// 开始登录了，标记打开时间
// 1,获取当前时间
long currentTime = System.currentTimeMillis();
SimpleDateFormat sdf = new SimpleDateFormat(
"yyyy-MM-dd HH:mm:ss");
Date date = new Date(currentTime);
String time = sdf.format(date);
// 修改标记的时间,保证5分钟嗅探链接一次
LPushUser.saveOpenTime(getApplicationContext(), time);
}
} else {
// APP端已经处于链接正常状态 -----5分钟嗅探链接一次
// 1,获取当前时间
long currentTime = System.currentTimeMillis();
SimpleDateFormat sdf = new SimpleDateFormat(
"yyyy-MM-dd HH:mm:ss");
Date date = new Date(currentTime);
String time = sdf.format(date);
// 2,比对链接打开时间
long minTime = LStringManager.dateDifference(
LPushUser.getOpenTime(getApplicationContext()),
time);
if (LPushUser.GETLOG(getApplicationContext())) {
Log.i("lpush",
"链接时长----现在时间:"
+ time
+ ";保存时间:"
+ LPushUser
.getOpenTime(getApplicationContext())
+ ";时差" + minTime + "分钟");
}
if (minTime >= 5) {
// 大于等于5分钟,则重新链接
// 5分钟之后重新链接
// 修改被拒绝状态
if (LPushUser.getConnectState(getApplicationContext())) {
LPushUser.saveConnect(getApplicationContext(),
false);
}
if (LNetworkUtil.netIsEnable(getApplicationContext())
&& LPushUser
.getSocketEnable(getApplicationContext())) {
mWebSocket.secondMethod(IP, PORT);
// 修改标记的时间,保证5分钟嗅探链接一次
LPushUser.saveOpenTime(getApplicationContext(),
time);
}
}
}
}
}
}
/***
* 设置推送功能的使用与否,默认使用推送功能,若是关闭推送功能请设置false;
*
* @param enable
* 是否使用
*
* li'mingqi
*/
protected void setLPushEnable(boolean enable) {
LPushUser.saveSocketEnable(getApplicationContext(), enable);
}
/***
*
*
* @param ip
* ip信息
* @param port
* 端口信息
*
* li'mingqi
*/
protected void setSocketIPInfo(String ip, String port) {
this.IP = ip;
this.PORT = port;
}
/**
* 设置用户的Uid
*
* @param uid
* li'mingqi
*/
public void setUserInfo(int uid, String code) {
/***
* 数据验证
*/
// if (0 == uid || null == code || "".equals(code)) {
// Log.e(TAG, "您输入的用户ID或者CODE值为空");
// new NullPointerException("您输入的用户ID或者CODE值为空").printStackTrace();
// return;
// }
// 保存用户ID
LPushUser.saveUserID(getApplicationContext(), uid);
// 保存用户CODE
LPushUser.saveUserCode(getApplicationContext(), code);
// 重启链接
mWebSocket.close();
}
/***
* 设置是否查看日志
*
* @param flag
* 是否查看日志
* @version 1.2 li'mingqi
*/
public void openLogInfo(boolean flag) {
LPushUser.SAVELOG(getApplicationContext(), flag);
}
/***
* socket链接重置,服务器一直处于拒绝链接状态,客户端链接一次遭拒后标记了遭拒的状态,重置之后可进行再次开启链接;
*
* @version 1.3 li'mingqi
*/
private void reset() {
LPushUser.saveConnect(getApplicationContext(), false);
}
}

UI界面显示部分就不贴出了，后面贴出Application类的目的就是监控各种手机系统的广播来嗅探自己的websocket链接的状态；用来维持它以保证能即时的收到消息；

posted @ 2017-08-22 17:52 BarryW 阅读(11188) 评论(1) 编辑收藏举报

刷新页面返回顶部

BarryWang

Websocket实现即时通讯

公告