webSocket 和java 实现互发信息

淩亂°似流年 2024-04-17 05:50 55阅读 0赞

webSocket.js
    
    class MyWebSocket extends WebSocket {
        constructor(params) {
            const urlPrefix = 'ws://' + window.location.host + '/chat-room/';
            const url = urlPrefix + params.name;
            super(url);
            this.name = params.name;
    
            //批量添加webSocket函数
            const keys = Object.getOwnPropertyNames(params).filter(param => typeof params[param] == "function");
            keys.map(key => {
                this[key] = params[key];
                console.log(params[key])
            })
        }
    
        send1(receive, info) {
            $.get("/chat-room/" + this.name + "/to/" + receive + "?message=" + info, function () {
                console.log("发送成功：" + info)
            })
        }
    }

**调用**

**java部分**  
[参考来源][Link 1]  
POM.XML

<dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-websocket</artifactId>
        </dependency>
    </dependencies>

socket.java

package com.xzff.frame.controller;
    
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.stereotype.Controller;
    import org.springframework.web.bind.annotation.PathVariable;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.ResponseBody;
    
    import javax.websocket.*;
    import javax.websocket.server.PathParam;
    import javax.websocket.server.ServerEndpoint;
    import java.io.IOException;
    
    import static com.xzff.frame.webSocket.WebSocketUtils.LIVING_SESSIONS_CACHE;
    import static com.xzff.frame.webSocket.WebSocketUtils.sendMessage;
    import static com.xzff.frame.webSocket.WebSocketUtils.sendMessageAll;
    
    @Controller
    @ResponseBody
    @ServerEndpoint("/chat-room/{username}")
    public class socket {
        private static final Logger log = LoggerFactory.getLogger(socket.class);
    
        @OnOpen
        public void openSession(@PathParam("username") String username, Session session) {
            LIVING_SESSIONS_CACHE.put(username, session);
            String message = "欢迎用户[" + username + "] 来到聊天室！";
            log.info(message);
            sendMessageAll(message);
        }
    
        @OnMessage
        public void onMessage(@PathParam("username") String username, String message) {
            log.info(message);
            sendMessageAll("用户[" + username + "] : " + message);
        }
    
        @OnClose
        public void onClose(@PathParam("username") String username, Session session) {
            //当前的Session 移除
            LIVING_SESSIONS_CACHE.remove(username);
            //并且通知其他人当前用户已经离开聊天室了
            sendMessageAll("用户[" + username + "] 已经离开聊天室了！");
            try {
                session.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    
        @OnError
        public void onError(Session session, Throwable throwable) {
            try {
                session.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
            throwable.printStackTrace();
        }
    
    
        @RequestMapping("/chat-room/{sender}/to/{receive}")
        public void onMessage(@PathVariable("sender") String sender, @PathVariable("receive") String receive, String message) {
            sendMessage(LIVING_SESSIONS_CACHE.get(receive),message);
        }
    }

WebSocketUtils.java

package com.xzff.frame.webSocket;
    
    import javax.websocket.RemoteEndpoint;
    import javax.websocket.Session;
    import java.io.IOException;
    import java.util.Map;
    import java.util.concurrent.ConcurrentHashMap;
    
    public final class WebSocketUtils {
        /**
         * 模拟存储 websocket session 使用
         */
        public static final Map<String, Session> LIVING_SESSIONS_CACHE = new ConcurrentHashMap<>();
    
        public static void sendMessageAll(String message) {
            LIVING_SESSIONS_CACHE.forEach((sessionId, session) -> sendMessage(session, message));
        }
    
        /**
         * 发送给指定用户消息
         *
         * @param session 用户 session
         * @param message 发送内容
         */
        public static void sendMessage(Session session, String message) {
            if (session == null) {
                return;
            }
            final RemoteEndpoint.Basic basic = session.getBasicRemote();
            if (basic == null) {
                return;
            }
            try {
                basic.sendText(message);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

[Link 1]: https://blog.battcn.com/2018/06/27/springboot/v2-other-websocket/