websocket协议介绍
屁话不多说，简单点说，就是一个保持全双工，能够保持长连接的协议，我们用它来进行发送弹幕和发视频，图片进行即时交流的工具。下面进行弹幕的技术实现。
onopen标识客户端与服务器进行通讯连接，在此处可进行业务逻辑的处理，将所需账号提出进行保存。
@Component
@ServerEndpoint(“/chatServer”)
public class ChatServer {
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;
/**
* concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。若要实现服务端与单一客户端通信的话，
* 可以使用Map来存放，其中Key可以为用户标识·1
*/
声明两个类级别的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);
// 释放资源，清空chatServer
chatServer = 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;
}
/**
* 连接关闭调用的方法
*/
@OnClose
public void onClose(Session session) {
try {
removeChatUserFromRoomHashMap(this.roomId, this.chatUserId);
subOnlineCount(); // 在线数减1
} catch (Exception e) {
}
}
关闭调用的方法
/**
* 收到客户端消息后调用的方法。 客户端发送消息的方法是，发送到服务器，消息中指明要传送给那个用户，消息的类型。<br/>
* 此处只管发送，至于发送的模式（紧急通知，普通通知，订单消息，聊天信息），这里并进行解析处理，由客户端自行处理。<br/>
*
*
* @param message
* 客户端发送过来的消息
* @param session
* 可选的参数
*/
@OnMessage
public 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());
}
}
// 处理二进制文件
else if ((StringUtils.equals(chatMessage.getMessageType(), EnumMessageType.BINARY.name()))) {
try {
boolean dealBinaryResult = dealBinary(chatMessage);
if (!dealBinaryResult) {
return;
}
} catch (Exception e) {
// TODO: handle exception
logger.error(“处理文件异常” + e.getMessage());
}
}
chatMessage.setImageBase64(“”);// 清空串
// 不用做NPE判断，因为chatUser如果为空 则推出closeSession 所以不可能为空
chatMessage.setChatName(this.chatUser.getUsername());
sendMessageToEveryoneInRoom(chatMessage);
} catch (Exception e) {// 发生错误即退出
e.printStackTrace();
}
}
进行通讯的模块。具体代码私信我，
## 不足之处为考虑多个终端登录的情况，比如一个用户双开浏览器，这时候这套方案显然是不可行的，具体的解决方案找我咨询
## 弹幕聊天室的实现
