WebSocket+SpringBoot聊天室（二）

深藏阁楼爱情的钟 2022-05-17 01:16 275阅读 0赞

**目录**

1.用户离开聊天室

2.单聊和多聊

3.项目源码

--------------------

## 1.用户离开聊天室 ##

跟用户进入聊天室一样，用户离开聊天室会触发OnClose，在OnClose广播我将要离开就可以了，思路跟进入聊天室是一样的，只是广播的内容变了。

@OnClose
        public void onClose(){
            webSockets.remove(this);
            map.remove(session.getId());
            System.out.println("有新的断开，总数"+webSockets.size()+"sessionId:"+session.getId());
            String content="恭送"+username+"离开聊天室！";
            Message message=new Message(content,map);
            send(message.toJson());
        }

还要在webSockets集合中把这个通道的WebScoket对象除入，同时在map集合中把session和username除去

## 2.单聊和多聊 ##

思路：先在客户端js中判断选了那几个用户，然后拼成字符串，发送到WebSocket服务器。服务器解析字符串，根据选中的用户来进行广播。

客户端判断：

主要用：var ss=$("\#userList :checked")来获取被选中的CheckBox集合。因为在生成用户列表的时候，我们随便生成了CheckBox，CheckBox的value就是每一个通道的sessionId。所以这时候我们要重新把value的值取出来。 用“-”把每一个ID拼装起来  to += value.getAttribute("value") + "-";在客户端传输数据到服务器时，我们也是用对象转成JSON字符串的方式。这个对象包括：1.要广播的信息、2.区分广播还是私聊的type、3.to就是存储要私聊的id集合。在js中定义这个对象，再转成JSON字符串。

因为客户端是用对象转JSON字符串传过来的，在服务器中我们需要把JSON字符串转对象，但是首先我们得有一个跟客户端类型一样的对象。

package com.example.websocket.vo;
    
    public class ContentVo {
    
        private String to;
    
        private String msg;
    
        private Integer type;
    
        public String getTo() {
            return to;
        }
    
        public void setTo(String to) {
            this.to = to;
        }
    
        public String getMsg() {
            return msg;
        }
    
        public void setMsg(String msg) {
            this.msg = msg;
        }
    
        public Integer getType() {
            return type;
        }
    
        public void setType(Integer type) {
            this.type = type;
        }
    
    }

服务器解析字符串：

@OnMessage
        public void onMessage(String json){
    
            ContentVo contentVo = gson.fromJson(json, ContentVo.class);
    
    
            if (contentVo.getType()==1){
                //广播
                Message message=new Message();
                message.setContent(this.username,contentVo.getMsg());
                message.setNames(map);
                send(message.toJson());
                System.out.println(message.toJson());
            }else {
                //单聊
                Message message=new Message();
                message.setContent(this.username,contentVo.getMsg());
                message.setNames(map);
    
                String to=contentVo.getTo();
                String tos[]=to.substring(0,to.length()-1).split("-");
                List<String> lists=Arrays.asList(tos);
                for (WebSocket webSocket:webSockets){
                    if (lists.contains(webSocket.session.getId())&&webSocket.session.getId()!=this.session.getId()) {
                        try {
                            webSocket.session.getBasicRemote().sendText(message.toJson());
                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                    }
                }
    
            }
        }

获得一个选中的sessionID集合，在循环WebSocket集合中，判断该WebSocket对象的SessionID是否在被选中的集合中，如果是就进行发送信息。

还有一点就是自己发的信息是在右侧，接收的信息是在左侧。在此之前，在广播的时候因为在WebSocket对象集合中，存在要发送给别人的这个WebSocket对象，所以会把信息从服务端发送到自己的客户端来。在改这个功能的时候，当一个用户点击发送的时候，是先把这些信息用js填到对话框中，然后服务端广播的时候就忽略发送信息的这个SessionID。

![70][]

## 3.项目源码 ##

链接：[https://pan.baidu.com/s/1DDAZ7Ig-WLFYWT\_5t-X-Aw][https_pan.baidu.com_s_1DDAZ7Ig-WLFYWT_5t-X-Aw]  密码：9sqk

[70]: /images/20220517/1cafbeadd05c48849369e5d50ba8c928.png
[https_pan.baidu.com_s_1DDAZ7Ig-WLFYWT_5t-X-Aw]: https://pan.baidu.com/s/1DDAZ7Ig-WLFYWT_5t-X-Aw