SpringBoot2.0整合集成WebSocket，实现后端数据实时推送！

最新推荐文章于 2024-01-31 19:40:51 发布

原创

最新推荐文章于 2024-01-31 19:40:51 发布 · 1.1k 阅读

本文介绍了如何在SpringBoot 2.0项目中整合WebSocket技术，详细阐述了配置过程和实现后端数据实时推送到前端的方法。

点击上方“Java碎碎念”,关注公众号

优质文章,第一时间送达

之前的一个系统为了实现推送使用的是Ajax轮询，这种方式浏览器需要不断的向服务器发出请求，会浪费很多的带宽等资源，所以研究了下WebSocket，本文将详细介绍下。

一、什么是WebSocket？

WebSocket是HTML5开始提供的一种在单个TCP连接上进行全双工通讯的协议，能更好的节省服务器资源和带宽，并且能够更实时地进行通讯。

WebSocket使客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据，在WebSocketAPI中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输。

二、SpringBoot整合WebSocket

新建一个spring boot项目spring-boot-websocket，按照下面步骤操作。

pom.xml引入jar包

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

新建WebSocket的配置类

这个配置类检测带注解@ServerEndpoint的bean并注册它们，配置类代码如下：

@Configuration
public class WebSocketConfig {
     
     
    /**
     * 给spring容器注入这个ServerEndpointExporter对象
     * 相当于xml：
     * <beans>
     * <bean id="serverEndpointExporter" class="org.springframework.web.socket.server.standard.ServerEndpointExporter"/>
     * </beans>
     * <p>
     * 检测所有带有@serverEndpoint注解的bean并注册他们。
     *
     * @return
     */
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
     
     
        System.out.println("我被注入了");
        return new ServerEndpointExporter();
    }
}

新建WebSocket的处理类

这个处理类需要使用@ServerEndpoint，这个类里监听连接的建立关闭、消息的接收等，具体代码如下：

@ServerEndpoint(value = "/ws/asset")
@Component
public class WebSocketServer {
     
     

    @PostConstruct
    public void init() {
     
     
        System.out.println("websocket 加载");
    }
    private static Logger log = LoggerFactory.getLogger(WebSocketServer.class);
    private static final AtomicInteger OnlineCount = new AtomicInteger(0);
    // concurrent包的线程安全Set，用来存放每个客户端对应的Session对象。
    private static CopyO