小型直播系统系列-乐聊TV的开发（四）

这一节，我们讲解一下基于websocket的弹幕实现：首先了解一下websocket协议

websocket

们知道，传统的HTTP协议是无状态的，每次请求（request）都要由客户端（如浏览器）主动发起，服务端进行处理后返回response结果，而服务端很难主动向客户端发送数据；这种客户端是主动方，服务端是被动方的传统Web模式对于信息变化不频繁的Web应用来说造成的麻烦较小，而对于涉及实时信息的Web应用却带来了很大的不便，如带有即时通信、实时数据、订阅推送等功能的应用。在WebSocket规范提出之前，开发人员若要实现这些实时性较强的功能，经常会使用折衷的解决方法：轮询（polling）和Comet技术。其实后者本质上也是一种轮询，只不过有所改进。

　　轮询是最原始的实现实时Web应用的解决方案。轮询技术要求客户端以设定的时间间隔周期性地向服务端发送请求，频繁地查询是否有新的数据改动。明显地，这种方法会导致过多不必要的请求，浪费流量和服务器资源。

　　Comet技术又可以分为长轮询和流技术。长轮询改进了上述的轮询技术，减小了无用的请求。它会为某些数据设定过期时间，当数据过期后才会向服务端发送请求；这种机制适合数据的改动不是特别频繁的情况。流技术通常是指客户端使用一个隐藏的窗口与服务端建立一个HTTP长连接，服务端会不断更新连接状态以保持HTTP长连接存活；这样的话，服务端就可以通过这条长连接主动将数据发送给客户端；流技术在大并发环境下，可能会考验到服务端的性能。

　　这两种技术都是基于请求-应答模式，都不算是真正意义上的实时技术；它们的每一次请求、应答，都浪费了一定流量在相同的头部信息上，并且开发复杂度也较大。

　　伴随着HTML5推出的WebSocket，真正实现了Web的实时通信，使B/S模式具备了C/S模式的实时通信能力。WebSocket的工作流程是这样的：浏览器通过JavaScript向服务端发出建立WebSocket连接的请求，在WebSocket连接建立成功后，客户端和服务端就可以通过 TCP连接传输数据。因为WebSocket连接本质上是TCP连接，不需要每次传输都带上重复的头部数据，所以它的数据传输量比轮询和Comet技术小了很多。本文不详细地介绍WebSocket规范，主要介绍下WebSocket在Java Web中的实现。
　　

websocket弹幕

    @OnOpenpublic void onOpen(Session session) {session.setMaxTextMessageBufferSize((int) MAX_BIG_LONG);addOnlineCount(); // 在线数加1--必须先加1---错误的时候会减1this.session = session;if (!parseQueryString(session)) // 如果未能取得用户id和type，退出return;// 验证账号，防止伪造this.chatUser = loginChatServer(chatUserId);if (this.chatUser == null) {closeSession(session);return;}addChatUserToHashMap(roomId, chatUserId);try {// 发一个应答标记，表示已经成功登陆，没有构造sendMessage("SUCCESS");} catch (IOException e) {// TODO Auto-generated catch blocke.printStackTrace();}Constant.ONLINECOUNT = onlineCount.toString();}

onopen标识客户端与服务器进行通讯连接，在此处可进行业务逻辑的处理，将所需账号提出进行保存

private static Log logger = LogFactory.getLog(ChatServer.class);/** AtomicInteger：线程安全的整数对象 */private static AtomicInteger onlineCount = new AtomicInteger(0);// 线程安全整数对象private static long MAX_BIG_LONG = 1024 * 4 * 1024;/** roomId与一个集合的哈希。集合中存储当前房间的所有用户 */private static ConcurrentHashMap<String, CopyOnWriteArraySet<String>> roomToChatUserHashMap = new ConcurrentHashMap<String, CopyOnWriteArraySet<String>>();/** 用户与chatServer实例的哈希。 */private static ConcurrentHashMap<String, ChatServer> chatUserToChatServer = new ConcurrentHashMap<String, ChatServer>();/** token验证 **/private String token;/** 房间号 **/private String roomId;/** chatUser 主键Id **/private String chatUserId;/** chatUserd对象 **/private ChatUser chatUser;

声明两个类级别的ConcurrentHashMap存储房间号和chatUser之间的对应关系，一个房间对应多个chatUser账号，一个chatUser对应一个chatServer实例，发送消息时只需进行遍历这俩个map找到相应的实例对象，调用它的发消息即可成功的发送

/*** 一个房间对应的一个chatuser列表 发消息时候进行遍历操作* * @param chatUserId* @param chatUserId* @return*/private boolean addChatUserToHashMap(String roomId, String chatUserId) {try {CopyOnWriteArraySet<String> chatUserIdSet = null;if (roomToChatUserHashMap.containsKey(roomId)) {chatUserIdSet = roomToChatUserHashMap.get(roomId);} else {chatUserIdSet = new CopyOnWriteArraySet<String>();}chatUserIdSet.add(chatUserId);roomToChatUserHashMap.put(roomId, chatUserIdSet);chatUserToChatServer.put(chatUserId, this);return true;} catch (Exception e) {e.printStackTrace();return false;}}

这是进入房间时，将自身账号加入到房间聊天群组，用来发广播消息时遍历的

/*** 从哈希中移除已经断开的连接* * @param chatUserId* @param terminalUuid* @return*/private boolean removeChatUserFromRoomHashMap(String roomId, String chatUserId) {try {CopyOnWriteArraySet<String> chatUserIdSet = null;if (roomToChatUserHashMap.containsKey(roomId)) {// 如果存在chatUserIdSet = roomToChatUserHashMap.get(chatUserId);// 取得chatUserId的集合} else {return true;}chatUserIdSet.remove(chatUserId);// 从集合中移除if (chatUserIdSet.size() == 0) {// 如果已经没有连接终端roomToChatUserHashMap.remove(roomId);// 则清除} else {roomToChatUserHashMap.put(roomId, chatUserIdSet);// 更新哈希}ChatServer chatServer = chatUserToChatServer.get(chatUserId);// 释放资源，清空chatServerchatServer = null;chatUserToChatServer.remove(chatUserId);// 将chatServer的实例从哈希中移除。return true;} catch (Exception e) {e.printStackTrace();return false;}}

当断开聊天室时，应该讲其从聊天组移除掉

 * 关闭websocket连接。* * @param session*            要关闭的会话*/private void closeSession(Session session) {try {session.close();} catch (IOException e) {e.printStackTrace();}return;}/*** 连接关闭调用的方法*/@OnClosepublic void onClose(Session session) {try {removeChatUserFromRoomHashMap(this.roomId, this.chatUserId);subOnlineCount(); // 在线数减1} catch (Exception e) {}}

关闭调用的方法

    @OnMessagepublic void onMessage(String message, Session session) {if (StringUtils.isBlank(message)) // 收到的是空串return;if (StringUtils.equals(Constant.SUCCESS_RESPONSE, message)) {return;}Gson gson = null;try {// 解析json串gson = new Gson();ChatMessage chatMessage = gson.fromJson(message, ChatMessage.class);// 解析json出错if (chatMessage == null)return;/*** 目的为了以后对每个模块进行拓展，所以分开写* */// 如果是图片类型的消息体if (StringUtils.equals(chatMessage.getMessageType(), EnumMessageType.IMAGE.name())) {try {boolean dealImageResult = dealBinary(chatMessage);// 处理图片结果if (!dealImageResult) {return;// 如果没生成，则返回}} catch (Exception e) {logger.error("处理图片异常" + e.getMessage());}}// 处理小视频else if ((StringUtils.equals(chatMessage.getMessageType(), EnumMessageType.VIDEO.name()))) {try {boolean dealVideoResult = dealBinary(chatMessage);if (!dealVideoResult) {return;}} catch (Exception e) {logger.error("处理小视频异常" + e.getMessage());}}...................代码不全，需要的找我联系

进行通讯的模块

不足之处

为考虑多个终端登录的情况，比如一个用户双开浏览器，这时候这套方案显然是不可行的，具体的解决方案找我咨询

弹幕聊天室

如果成功了之后就会是下面的场景

开始动手制作你自己的直播间和聊天室吧。