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);
    }
 }

  
相关推荐
游戏开发爱好者823 分钟前
iOS重构期调试实战:架构升级中的性能与数据保障策略
websocket·网络协议·tcp/ip·http·网络安全·https·udp
却道天凉_好个秋3 小时前
音视频学习(三十六):websocket协议总结
websocket·音视频
(:满天星:)4 小时前
第31篇:块设备与字符设备管理深度解析(基于OpenEuler 24.03)
linux·运维·服务器·网络·centos
野蛮人6号5 小时前
虚拟机网络编译器还原默认设置后VMnet8和VMnet1消失了
网络·vmware·虚拟机网络编译器·vmnet8消失
scuter_yu6 小时前
主流零信任安全产品深度介绍
运维·网络·安全
江苏思维驱动智能研究院有限公司6 小时前
Sophos 网络安全:全球领先的自适应安全解决方案提供商
网络·安全·web安全
面朝大海,春不暖,花不开6 小时前
Java网络编程:TCP/UDP套接字通信详解
java·网络·tcp/ip
ChicagoTypewriter6 小时前
计算机网络中的常用表项梳理
网络·计算机网络·智能路由器
DemonAvenger7 小时前
高性能 TCP 服务器的 Go 语言实现技巧:从原理到实践
网络协议·架构·go
Bruce_Liuxiaowei9 小时前
常见高危端口风险分析与防护指南
网络·网络安全·端口·信息搜集