王柳 发表于 2024-5-19 21:44:12

SpringBoot3集成WebSocket

标签:WebSocket,Session,Postman。
一、简介

WebSocket通过一个TCP连接在客户端和服务器之间创建一个全双工、双向的通信通道,使得客户端和服务器之间的数据互换变得更加简单,允许服务端主动向客户端推送数据,在WebSocket的API中,浏览器和服务器只需要完成一次握手,两者之间就直接可以创建长期性的连接,并进行双向数据传输。
【基于Postman工具的WebSocket交互】
Connected to ws://127.0.0.1:8088/web/socket/msg

Handshake Details
Request URL: http://127.0.0.1:8088/web/socket/msg
Request Method: GET
Status Code: 101

Request Headers
Sec-WebSocket-Version: 13
Sec-WebSocket-Key: 5Qrs8JeRLsiY9G/PRJUocQ==
Connection: Upgrade
Upgrade: websocket
Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits
Host: 127.0.0.1:8088

Response Headers
Upgrade: websocket
Connection: upgrade
Sec-WebSocket-Accept: E3aFw2bBzxByPCynmQ7lZ+7BgsU=
Sec-WebSocket-Extensions: permessage-deflate;client_max_window_bits=15二、工程搭建

1、工程结构

https://img2024.cnblogs.com/blog/1691717/202405/1691717-20240512094357532-738635656.png
2、依赖管理

在starter-websocket的依赖中,涉及到spring框架中两个关系较为密切的组件,分别是websocket和messaging组件。
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
    <version>${spring-boot.version}</version>
</dependency>三、WebSocket用法

1、示意图

在下面的案例中,大致模拟这样一个流程,三个客户端分别和服务端创建连接,然后进行客户端之间的会话通信。
https://img2024.cnblogs.com/blog/1691717/202405/1691717-20240512094401796-199742892.png
2、API参考

这里通过4个焦点的方法注解,分别处分析话的差别动作,比如连接的创建和关闭,通信交互和错误处理;在实际的应用场景中,需要在通信方法中设计更多的指令来应对差别的业务场景。
@ServerEndpoint("/web/socket/msg")
public class MsgWebSocket {

    private static finalConcurrentHashMap<String,Session> sessions = new ConcurrentHashMap<>();

    private static final AtomicInteger onlineCount = new AtomicInteger(0);

    /**
   * 建立连接调用的方法
   */
    @OnOpen
    public void onOpen(Session session) {
      String userId = session.getRequestParameterMap().get("userId").get(0);
      // 加入Set中
      sessions.put(userId,session);
      // 在线数增加
      onlineCount.getAndIncrement();
      log.info("session-{},online-count-{}",session.getId(),onlineCount.get());
    }

    /**
   * 客户端消息处理的方法
   */
    @OnMessage
    public void sendMsg(Session sender,String message) throws Exception {
      MsgDTO dto = JSONUtil.toBean(message, MsgDTO.class);
      Session receiver = sessions.get(dto.getUserId());
      if (receiver != null) {
            receiver.getBasicRemote().sendText(dto.getMsg());
      }
    }

    /**
   * 关闭连接调用的方法
   */
    @OnClose
    public void onClose(Session session) {
      String userId = session.getRequestParameterMap().get("userId").get(0);
      // 从Set中删除
      sessions.remove(userId);
      // 在线数减少
      onlineCount.getAndDecrement();
      log.info("session-{},down-line-count-{}",session.getId(),onlineCount.get());
    }

    /**
   * 发生错误调用的方法
   */
    @OnError
    public void onError(Session session, Throwable throwable) throws Exception {
      log.error("Web Stock Error", throwable);
      session.getBasicRemote().sendText(throwable.getMessage());
    }
}测试效果图:留意这里使用的是postman最新版本。
https://img2024.cnblogs.com/blog/1691717/202405/1691717-20240512094405868-958451724.png
四、源码参考

文档仓库:
https://gitee.com/cicadasmile/butte-java-note

源码仓库:
https://gitee.com/cicadasmile/butte-spring-parent
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: SpringBoot3集成WebSocket