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

  
相关推荐
Bruce_Liuxiaowei17 分钟前
常见高危端口风险分析与防护指南
网络·网络安全·端口·信息搜集
tmacfrank38 分钟前
Android 网络全栈攻略(四)—— TCPIP 协议族与 HTTPS 协议
android·网络·https
liulilittle1 小时前
深度剖析:OPENPPP2 libtcpip 实现原理与架构设计
开发语言·网络·c++·tcp/ip·智能路由器·tcp·通信
cui_win1 小时前
【内存】Linux 内核优化实战 - net.ipv4.tcp_tw_reuse
linux·网络·tcp/ip
2501_916013741 小时前
iOS 多线程导致接口乱序?抓包还原 + 请求调度优化实战
websocket·网络协议·tcp/ip·http·网络安全·https·udp
M1A11 小时前
TCP/IP协议精解:IP协议——互联网世界的邮政编码系统
后端·网络协议·tcp/ip
夏天想2 小时前
优化 WebSocket 实现单例连接用于打印【待测试 】
网络·websocket·网络协议
路长且阻2 小时前
网络协议(TCP/IP、HTTP、HTTPS)
网络协议·tcp/ip·http
杰尼橙子3 小时前
DPDK基础架构解析:EAL环境抽象层的设计与实现
网络协议·性能优化
我是小bā吖3 小时前
阿里云服务网格ASM实践
网络·阿里云·云计算·服务发现