Merge branch 'master' of http://10.153.19.25:10101/r/CanadaMes

wu

2023-12-11 7f1806c6a01318f3b1eff38da7c7c41ae82ff8c0

 springboot-vue3/src/main/java/com/example/springboot/component/WebSocketServer.java

@@ -8,7 +8,6 @@
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
@@ -32,24 +31,26 @@

    static ConfigurableApplicationContext applicationContext;
 
    //解决无法注入mapper问题  //使用方法 homeMapper=WebSocketServer.applicationContext.getBean(HomeMapper.class);
    // 解决无法注入mapper问题 //使用方法

    // homeMapper=WebSocketServer.applicationContext.getBean(HomeMapper.class);

    public static void setApplicationContext(ConfigurableApplicationContext configurableApplicationContext) {
        WebSocketServer.applicationContext = configurableApplicationContext;
    }


   private static final Logger log = LoggerFactory.getLogger(WebSocketServer.class);
    private List<String> messages;
    /**
     * 记录当前在线连接数
     */
    public static final Map<String, WebSocketServer> sessionMap = new ConcurrentHashMap<>();
    public static final Map<String, ArrayList<WebSocketServer>> sessionMap = new ConcurrentHashMap<>();

    
    String username;
    Session session;


    public WebSocketServer() {
        this.messages = new ArrayList<>();
    }


    /**
     * 连接建立成功调用的方法
     */
@@ -57,8 +58,16 @@
    public void onOpen(Session session, @PathParam("username") String username) {
        this.username=username;
        this.session=session;
        sessionMap.put(username, this);
        log.info("有新用户加入，username={}, 当前在线人数为：{}", username, sessionMap.size());
        List<WebSocketServer> webSocketServers = sessionMap.get(username);

        if (webSocketServers == null) {

            ArrayList<WebSocketServer> arrayListwebserver = new ArrayList<WebSocketServer>();

            arrayListwebserver.add(this);

            sessionMap.put(username, arrayListwebserver);

        } else {

            webSocketServers.add(this);

        }

        
        log.info("有新用户加入，username={}, 当前在线人数为：{}", username, sessionMap.get(username).size());

       
        // JSONObject result = new JSONObject();
        // JSONArray array = new JSONArray();
@@ -77,8 +86,21 @@
     */
    @OnClose
    public void onClose(Session session, @PathParam("username") String username) {
        List<WebSocketServer> webSocketServers = sessionMap.get(username);

        ArrayList<WebSocketServer> arrayListwebserver = new ArrayList<WebSocketServer>();

        if (webSocketServers.size()>1) {

            for (WebSocketServer webSocketServer : webSocketServers) {

                if(webSocketServer!=this){

                    arrayListwebserver.add(webSocketServer);

                }

            }

            sessionMap.put(username, arrayListwebserver);

            log.info("移除username={}一名用户session, {}的当前在线人数为：{}", username, username, sessionMap.get(username).size());

        }else{

        sessionMap.remove(username);
        log.info("有一连接关闭，移除username={}的用户session, 当前在线人数为：{}", username, sessionMap.size());
            log.info("移除username={}一名用户session, {}连接关闭, 当前连接数为：{}", username, username, sessionMap.size());

        }

        
    }

    /**
@@ -86,6 +108,7 @@
     * 后台收到客户端发送过来的消息
     * onMessage 是一个消息的中转站
     * 接受 浏览器端 socket.send 发送过来的 json数据
     * 
     * @param message 客户端发送过来的消息
     */
    @OnMessage
@@ -119,19 +142,19 @@
        }
    }

    /**
     * 服务端发送消息给所有客户端
     */
    public void sendAllMessage(String message) {
        try {
            for (WebSocketServer webSocketServer : sessionMap.values()) {
                //log.info("服务端给客户端[{}]发送消息{}", this.session.getId(), message);
                webSocketServer.sendMessage(message);
            }
        } catch (Exception e) {
            log.error("服务端发送消息给客户端失败", e);
        }
    }
    // /**

    //  * 服务端发送消息给所有客户端

    //  */

    // public void sendAllMessage(String message) {

    //     try {

    //         for (WebSocketServer webSocketServer : sessionMap.values()) {

    //             // log.info("服务端给客户端[{}]发送消息{}", this.session.getId(), message);

    //             webSocketServer.sendMessage(message);

    //         }

    //     } catch (Exception e) {

    //         log.error("服务端发送消息给客户端失败", e);

    //     }

    // }


    public List<String> getMessages() {
        return messages;