Websocket实现方式二——注解方式

添加Websocket依赖

xml 复制代码
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

根据@ServerEndpoint注解注册Websocket

java 复制代码
@Configuration
public class AgentWsConfig {

    @Bean
    public ServerEndpointExporter serverEndpointExporter(){
        return new ServerEndpointExporter();
    }

}

创建Websocket注解实现的类@ServerEndpoint注解标记类

java 复制代码
@ServerEndpoint("/websocket/{sessionId}")
@Component
@Slf4j
public class AgentWsServerPoint {
    /**
     * websocket连接集合,根据Redis广播来的sessionId,来管理Session(取用和删除)
     */
    private static final Map<String, Session> SESSION_MAP = new ConcurrentHashMap<>();

    /**
     * 建立连接
     * @param sessionId
     * @param session
     */
    @OnOpen
    public void onOpen(@PathParam("sessionId") String sessionId, Session session) {
        SESSION_MAP.put(sessionId, session);
        log.info("WebSocket已打开,会话ID是:{}", sessionId);
        log.info(sessionId + "建立了连接");
    }

    /**
     * 接收消息并发送消息
     * @param sessionId
     * @param message
     * @param session
     * @return
     */
    @OnMessage
    public String onMessage(@PathParam(value = "sessionId") String sessionId,String message, Session session) throws Exception {
        log.info(sessionId + ":" + message);//客户端推送过来的消息
        
        Session session = SESSION_MAP.get(sessionId);
        try {
               session.getBasicRemote().sendText("Hello World!"); // 给特定用户发送消息
               log.info("消息发送成功");
            } catch (IOException e) {
                e.printStackTrace();
                log.error("消息发送失败");
                throw e;
            }
        return null;
    }

  
    /**
     * 关闭连接
     * @param sessionId
     * @param session
     */
    @OnClose
    public void onClose(@PathParam(value = "sessionId") String sessionId,Session session) throws IOException {
        if(SESSION_MAP.containsKey(sessionId)){
            Session session1 = SESSION_MAP.get(sessionId);
            session1.close();
            SESSION_MAP.remove(sessionId);
        }
        log.info("websocket is close ,session id is {}",sessionId);
    }
 }

  
相关推荐
今天秃头了吗??2 小时前
【JAVA】网络编程
java·网络·网络编程·socket
is08153 小时前
NFS 服务器 使用
服务器·网络
言之。3 小时前
TCP 拥塞控制设计空间课程要点总结
网络·网络协议·tcp/ip
Freed&4 小时前
《没有架构图?用 netstat、ss、tcpdump 还原服务连接与数据流向》
网络·测试工具·tcpdump
千钰v5 小时前
Tcpdump: The Basics Tcpdump 基础
linux·运维·网络·tcpdump·tryhackme
跨境小新5 小时前
ChatGPT大模型训练指南:如何借助动态代理IP提高训练效率
网络协议·tcp/ip·chatgpt
全栈技术负责人5 小时前
前端网络性能优化实践:从 HTTP 请求到 HTTPS 与 HTTP/2 升级
前端·网络·http
独行soc5 小时前
2025年渗透测试面试题总结-67(题目+回答)
网络·python·安全·web安全·网络安全·adb·渗透测试
eybk6 小时前
用python的socket写一个局域网传输文件的程序
服务器·网络·python
树码小子6 小时前
Java网络编程:(socket API编程:UDP协议的 socket API -- 回显程序的服务器端程序的编写)
java·网络·udp