Java websocket + redis 实现多人单聊天室，多人多聊天室，一对一聊天

最新推荐文章于 2024-11-07 09:36:05 发布

原创

最新推荐文章于 2024-11-07 09:36:05 发布 · 1.1w 阅读

36 ·

CC 4.0 BY-SA版权

文章标签：

#java #聊天 #websocket #redis #并发

本文介绍了使用Java WebSocket和Redis构建聊天室的实现，包括多人单聊、多聊室功能。聊天室具备界面美观、支持文字和图片发送、成员列表、历史消息查看等特性。技术要点涉及CopyOnWriteMap保证线程安全、Redis存储聊天记录和成员信息。未来计划解决高并发问题，并实现聊天室ID隔离的多聊天室功能以及一对一聊天，其中一对一聊天将维护关系并利用Redis存储聊天记录。

多人，单聊天室版

FEATURE

多人聊天，界面简洁美观，使用ueditor支持发送文字，图片信息
群聊成员列表，登入登出公告
存储聊天记录，查看历史消息

技术点

使用CopyOnWriteMap存储websocketServer对象，线程安全
redis存储消息记录
ConcurrentLinkedQueue存储聊天成员

TODO

没有处理高并发，高并发情况对服务器和内存都会产生极大压力解决方案采取实现分布式
当前是所有成员在一个聊天室，计划按照聊天室ID隔离出多聊天室（使用Redis存储）

主逻辑代码

@ServerEndpoint(value="/websocketServer", configurator=SpringConfigurator.class, encoders = { CommonMessageEncoder.class, SystemMessageEncoder.class, HistoryMessageEncoder.class})
public class WebsocketServer {
	//存储每个客户端对应的websocketServer实例与登录名map
    private static CopyOnWriteMap<WebsocketServer, String> webSocketUsernameMap = new CopyOnWriteMap<WebsocketServer, String>();
    
    private MessageDao messageDao = (MessageDao)ContextLoader.getCurrentWebApplicationContext().getBean("messageDao");
    
    //在线成员
    private static ConcurrentLinkedQueue<String> members = new ConcurrentLinkedQueue<String>();

    //每个webscoket客户端与服务器会话
    private Session session;
    
    public WebsocketServer() {
    }
    
    @OnOpen
    public void onOpen(Session session) {
        this.session = session;
    }
    
    @OnClose
    public void onClose() {
        String username = webSocketUsernameMap.get(this);
        removeMember(username);
        webSocketUsernameMap.remove(this);
        
        for (WebsocketServer webSocket : webSocketUsernameMap.keySet()) {
            try {
                sendMsg(webSocket, new SystemMessageResponse(MessageType.SYS_MSG, username, "exit", members));
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
    
    @OnMessage
    public void onMessage(String message, Session session) {
    	JSONObject messageObject = new JSONObject(message);
    	String type = messageObject.getString("messageType");
    	String content = messageObject.getString("message");
    	
    	if (type.equals(MessageType.COM_MSG)) {
            //群发消息
    		Date time = new D