WebSocket Day04 : 消息推送

前言

随着Web应用程序的不断发展，实时性和交互性成为了用户体验中至关重要的一部分。传统的HTTP协议在处理实时数据传输方面存在一些局限性，而WebSocket作为一种全双工通信协议，为实现实时、高效的消息推送提供了全新的解决方案。

在Web开发领域，消息推送是一项重要的功能，它使得服务器能够将实时更新的信息主动推送给客户端，而不需要客户端频繁地向服务器发送请求。借助于WebSocket技术，开发人员能够轻松构建具有实时交互功能的Web应用程序，例如在线聊天、实时数据监控、多人协作编辑等。

一、前期准备

1、新建项目，结构如下

2、导入依赖

  <!-- websocket 依赖 -->
        <dependency>
            <groupId>javax.websocket</groupId>
            <artifactId>javax.websocket-api</artifactId>
            <version>1.1</version>
            <scope>provided</scope>
        </dependency>

        <!-- 打印日志 -->
        <dependency>
            <groupId>ch.qos.logback</groupId>
            <artifactId>logback-classic</artifactId>
            <version>1.3.8</version>
        </dependency>

        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.24</version>
        </dependency>

        <!-- ch02 -->
        <dependency>
            <groupId>javax.servlet</groupId>
            <artifactId>javax.servlet-api</artifactId>
            <version>4.0.1</version>
            <scope>provided</scope>
        </dependency>

        <dependency>
            <groupId>com.fasterxml.jackson.core</groupId>
            <artifactId>jackson-databind</artifactId>
            <version>2.14.2</version>
        </dependency>

        <!-- ch03 -->
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-webmvc</artifactId>
            <version>5.3.23</version>
        </dependency>
        <!-- 在 spring 4.0 开始支持-->
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-websocket</artifactId>
            <version>5.3.23</version>
        </dependency>

        <!-- ch04 -->
        <!-- 如果想要使用 Spring 的消息代理，则添加以下依赖 -->
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-messaging</artifactId>
            <version>5.3.23</version>
        </dependency>

每一个依赖的作用：

1. javax.websocket-api: 这是Java API for WebSocket的依赖声明，版本为1.1。它提供了在Java应用程序中使用WebSocket所需的接口和类。

2. logback-classic: 这是Logback日志框架的经典模块，版本为1.3.8。Logback是一个灵活而功能强大的日志框架，它可以代替log4j系统。

3. lombok: Lombok是一个Java库，可以通过注解的方式来简化Java代码的开发。它通过消除样板代码（boilerplate code）来提高开发效率。

4. javax.servlet-api: 这是Java Servlet API的依赖声明，版本为4.0.1。它提供了使用Java语言编写的Web应用程序中的Servlet和JSP规范。

5. jackson-databind: Jackson是Java中用于处理JSON数据的流行库，databind模块提供了JSON和Java对象之间的转换功能。

6. spring-webmvc: 这是Spring框架中的Web MVC模块的依赖声明，版本为5.3.23。它提供了创建基于Spring的Web应用程序的支持。

7. spring-websocket: 这是Spring框架中的WebSocket模块的依赖声明，版本为5.3.23。它提供了对WebSocket的支持，使得在Spring应用程序中能够方便地集成和使用WebSocket。

8. spring-messaging: 这是Spring框架中的消息代理模块的依赖声明，版本为5.3.23。它提供了在Spring应用程序中实现消息代理的支持。

以上依赖项将会为您的项目引入WebSocket所需的API、日志、简化代码、Servlet支持以及Spring框架的相关模块，让您可以方便地开发基于WebSocket的Web应用程序，并且与Spring框架无缝集成。

3、配置配置类

1）MvcConfig 类

@Configuration
@EnableWebMvc
public class MvcConfig implements WebMvcConfigurer {

    @Override
    public void configureDefaultServletHandling(DefaultServletHandlerConfigurer configurer) {
        configurer.enable();
    }
}

让我解释一下它的作用：

1. @Configuration: 这个注解表示这是一个配置类，它会被Spring容器识别并用来配置应用程序上下文。

2. @EnableWebMvc: 这个注解启用了Spring MVC，它会在Spring中注册必要的组件，如HandlerMapping、HandlerAdapter等，以支持基于注解的控制器方法。

3. implements WebMvcConfigurer: MvcConfig类实现了WebMvcConfigurer接口，这样可以重写一些默认配置，自定义MVC的行为。

4. configureDefaultServletHandling: 这是WebMvcConfigurer接口中的一个方法，通过重写这个方法，我们可以配置静态资源的处理。在这里的配置中，configurer.enable()启用了默认的Servlet处理，允许DispatcherServlet将对静态资源的请求转发到Servlet容器的default servlet。

总的来说，这段代码配置了Spring MVC，并且使得Spring能够处理静态资源的请求，从而使得静态资源可以被正确地访问和展示。

2）WebSocket 类

@Configuration
// 启用 webSocket 消息代理中间件
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    /**
     * 注册一个连接消息中间件的端点（ 路径url ）
     * @param registry
     */
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("broker");
    }

    /**
     * 配置消息代理，主要是设置相关的主题
     * 消息代理是服务中心的核心，spring-websocket 内置了
     * 一个简单的消息代理，但也只能够满足基本要求，如果
     * 需要强大的消息中心的功能，通常都会集成第三方的消息队列
     * 例如：RabbitMQ 等
     * @param registry
     */
    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        // 启用 Spring 内置简单的消息代理并设置一个主题（topic）的前缀
        // 用于消息的发布和订阅
        registry.enableSimpleBroker("/news","video");

        // 如果需要集成外部其他的消息代理，使用下面等等方法
        // registry.enableStompBrokerRelay();

    }
}

让我来解释一下它的作用：

1. @Configuration: 这个注解表示这是一个配置类，它会被Spring容器识别并用来配置应用程序上下文。

2. @EnableWebSocketMessageBroker: 这个注解启用了基于消息代理的WebSocket支持，它会为您的应用程序提供了实时的、双向通信能力。

3. implements WebSocketMessageBrokerConfigurer: WebSocketConfig类实现了WebSocketMessageBrokerConfigurer接口，这样可以重写一些默认配置，自定义WebSocket的行为。

4. registerStompEndpoints: 在这个方法中，我们注册了一个STOMP端点（也就是WebSocket端点），客户端可以通过该端点进行连接。在这里的配置中，我们注册了一个名为"broker"的端点，客户端将使用这个端点来建立WebSocket连接。

5. configureMessageBroker: 这个方法用来配置消息代理，主要是设置相关的主题。在这里的配置中，我们启用了Spring内置的简单消息代理，并设置了两个主题的前缀 "/news" 和 "/video"，用于消息的发布和订阅。这样客户端就可以订阅这些主题，从而接收相关的消息。

总的来说，这段代码配置了Spring WebSocket的消息代理中间件，使得应用程序能够通过WebSocket进行实时的双向通信，同时设置了相关的主题，以便客户端可以订阅并接收相应的消息。

3）AppConfig 类

@Configuration
@ComponentScan(basePackages = "edu.nf.ch04")
@Import({MvcConfig.class,WebSocketConfig.class})
public class AppConfig {
}

让我来解释一下它的作用：

1. @Configuration: 这个注解表示这是一个配置类，它会被Spring容器识别并用来配置应用程序上下文。

2. @ComponentScan(basePackages = "edu.nf.ch04"): 这个注解指定了Spring容器在初始化时要扫描的包路径，以寻找带有特定注解的组件。在这里的配置中，容器会扫描"edu.nf.ch04"包及其子包，寻找带有@Component、@Service、@Repository等注解的组件，并将它们纳入Spring容器管理。

3. @Import({MvcConfig.class, WebSocketConfig.class}): 这个注解用于导入其他的配置类，将其配置加入到当前的配置类中。在这里的配置中，AppConfig类导入了MvcConfig和WebSocketConfig两个配置类，从而将它们的配置内容整合到当前的配置中。

总的来说，这段代码配置了Spring应用程序的整体配置信息，包括了组件扫描的包路径以及导入了其他的配置类，从而将整个应用程序的配置信息整合在一起。

4）WebConfig 类

public class WebConfig extends AbstractAnnotationConfigDispatcherServletInitializer {
    @Override
    protected Class<?>[] getRootConfigClasses() {
        return new Class[0];
    }

    @Override
    protected Class<?>[] getServletConfigClasses() {
        return new Class[]{AppConfig.class};
    }

    @Override
    protected String[] getServletMappings() {
        return new String[]{"/"};
    }
}

这段代码是一个典型的Spring Web应用程序的配置类，它继承自AbstractAnnotationConfigDispatcherServletInitializer，用于替代传统的web.xml文件，让我来解释一下它的作用：

1. public class WebConfig extends AbstractAnnotationConfigDispatcherServletInitializer: 这个类是Web应用程序的初始化类，它相当于web.xml文件中的配置。

2. @Override protected Class<?>[] getRootConfigClasses() { return new Class[0]; }: 在这个方法中，我们返回一个空数组，表示不需要加载根配置类。通常，根配置类用来配置数据源、持久层框架、安全框架等应用程序级别的组件，但在这里我们没有定义。

3. @Override protected Class<?>[] getServletConfigClasses() { return new Class[]{AppConfig.class}; }: 在这个方法中，我们返回一个包含AppConfig类的数组，表示加载Web应用程序的配置类。在这里，我们将之前提到的AppConfig配置类加载到Servlet上下文中，这样就可以使用其中定义的Bean。

4. @Override protected String[] getServletMappings() { return new String[]{"/"}; }: 这个方法返回一个包含"/"路径的字符串数组，表示DispatcherServlet要映射的路径。在这里的配置中，DispatcherServlet会映射到应用程序的根路径，即所有的请求都会交给DispatcherServlet进行处理。

总的来说，这段代码配置了Spring Web应用程序的初始化信息，包括加载Web应用程序的配置类、指定DispatcherServlet的映射路径等，从而取代了传统的web.xml文件。

4、封装一个信息类

@Data
@AllArgsConstructor
@NoArgsConstructor
public class Message {

    private String content;
    private String sendTime;

}

让我来解释一下使用到的注解和它的作用：

1. @Data: 这是Lombok库提供的一个注解，它会为类自动生成一些通用的方法，如toString()、equals()、hashCode()等，以及针对每个字段的getters和setters方法。使用@Data注解可以减少样板代码，让类看起来更简洁。

2. @AllArgsConstructor: 这是Lombok库提供的另一个注解，它会为类生成一个包含所有字段的构造方法，这样就不需要手动编写构造方法。

3. @NoArgsConstructor: 这也是Lombok库提供的注解，它会生成一个无参的构造方法。

4. private String content;: 这行代码定义了一个私有字段content，用于存储消息的内容。

5. private String sendTime;: 这行代码定义了另一个私有字段sendTime，用于存储消息的发送时间。

总的来说，这段代码使用了Lombok库的注解来自动生成通用方法和构造方法，从而简化了代码编写。同时，定义了一个简单的Java类Message，用于表示消息对象，包括消息内容和发送时间两个属性。

二、编写 controller 类，实现通知

@RestController
@RequiredArgsConstructor
public class PublishController {

    /**
     * 消息处理模板，用于发布消息
     */
    private final SimpMessagingTemplate template;

    /**
     * 后台发布消息
     * @param message
     */
    @PostMapping("/publish/{topic}/{sub}")
    public void publish(@PathVariable("topic") String topic,@PathVariable("sub")String sub, String message){
        // 创建一个时间
        String sendTime = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date());
        Message msg = new Message(message,sendTime);

        // 将消息发布到消息代理指定的主题中
        template.convertAndSend("/"+ topic +"/" + sub,msg);
    }

}

这段代码是一个Spring框架中的REST控制器类PublishController，它使用了一些注解和依赖注入，让我来逐步解释：

1. @RestController: 这个注解表示该类是一个RESTful风格的控制器，它将处理传入的HTTP请求，并返回相应的HTTP响应。相比于@Controller注解，它会自动将返回结果转换为JSON格式。

2. @RequiredArgsConstructor: 这是Lombok库提供的注解，它会为类中标记了final的字段生成一个构造方法，用于进行依赖注入。在这里，它的作用是为template字段生成一个构造方法，实现依赖注入。

3. private final SimpMessagingTemplate template;: 这行代码定义了一个私有的、不可变的SimpMessagingTemplate类型的字段template，用于向WebSocket消息代理发送消息。

4. @PostMapping("/publish/{topic}/{sub}"): 这是一个用于处理HTTP POST请求的方法级注解，表示对应的publish方法将处理POST请求，并且在路径中包含了两个占位符{topic}和{sub}。

5. public void publish(@PathVariable("topic") String topic, @PathVariable("sub") String sub, String message): 这是一个处理POST请求的方法，它接受三个参数，其中@PathVariable注解表示从URL中获取路径参数。方法内部首先创建了消息对象msg，然后使用SimpMessagingTemplate将消息发布到指定的主题中。

总的来说，这个PublishController类是一个RESTful风格的控制器，用于处理POST请求，并通过SimpMessagingTemplate向WebSocket消息代理发布消息。

三、编写页面

1、admin.html --- 发送信息后台

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
    <script src="js/JQuery文件.txt.js"></script>
</head>
<body>
<h1>后台管理</h1>
<form id="f1">
<select id="topic">
    <option value="sport">体育</option>
    <option value="recreation">娱乐</option>
</select>

<input type="text" name="message" id="message"/>
<input type="button" value="发布">
</form>

<script>
    $(function () {
        $(':button').on('click',function () {
            let param = $('#f1').serialize();
            let subTopic = $('#topic').val();
            // alert(subTopic)
            $.ajax({
                url:'../publish/news/' + subTopic,
                type:'post',
                data:param,
                success:function ( data ) {
                    $('#message').val('');
                }
            })
        })
    })
</script>

</body>
</html>

2、index.html --- 体育栏目

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
    <script src="js/JQuery文件.txt.js"></script>
    <script src="js/stomp.min.js"></script>

</head>
<body>
<h1>体育</h1>

<div id="msg">

</div>

<script>
    $(function () {
        // 创建 WebSocket 对象
        var ws = new WebSocket("ws://localhost:8080/broker");

        // 将 WebSocket 包装成 stomp 客户端
        let stompClient = Stomp.over(ws);

        // 连接服务器并订阅消息
        stompClient.connect({},function () {
            // 执行订阅
            stompClient.subscribe('/news/sport',function ( data ) {
                // 接收发布的通知内容
                // alert(data);
                // 取出 stomp 中的 body 部分并解析为 json 对象
                let msg = $.parseJSON(data.body);
                $('#msg').append(msg.sendTime + '<br>');
                $('#msg').append(msg.content + '<br>')
            })
        })
    })
</script>

</body>
</html>

让我来逐行解释上述代码的含义：

1. var ws = new WebSocket("ws://localhost:8080/broker"); - 这一行创建了一个WebSocket对象，并指定了连接的URL为"ws://localhost:8080/broker"。这将尝试连接到本地主机上的WebSocket服务器，并订阅了"/broker"频道。

2. let stompClient = Stomp.over(ws); - 这一行使用了STOMP（Simple Text Oriented Messaging Protocol）库，通过WebSocket创建了一个STOMP客户端对象。

3. stompClient.connect({}, function () {...}); - 这段代码表示当STOMP客户端成功连接到服务器时，执行后续的操作。

4. stompClient.subscribe('/news/sport', function (data) {...}); - 在连接成功后，客户端订阅了"/news/sport"频道，用于接收体育新闻。当有消息到达时，会执行回调函数，并将消息的发送时间和内容显示在页面上。

总体来说，这段代码的作用是在页面加载后，通过WebSocket连接到服务器，订阅体育新闻频道，并将接收到的消息动态显示在页面上。

2、index2.html --- 娱乐栏目

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
    <script src="js/JQuery文件.txt.js"></script>
    <script src="js/stomp.min.js"></script>

</head>
<body>
<h1>娱乐</h1>

<div id="msg">

</div>

<script>
    $(function () {
        // 创建 WebSocket 对象
        var ws = new WebSocket("ws://localhost:8080/broker");

        // 将 WebSocket 包装成 stomp 客户端
        let stompClient = Stomp.over(ws);

        // 连接服务器并订阅消息
        stompClient.connect({},function () {
            // 执行订阅
            stompClient.subscribe('/news/recreation',function ( data ) {
                // 接收发布的通知内容
                // alert(data);
                // 取出 stomp 中的 body 部分并解析为 json 对象
                let msg = $.parseJSON(data.body);
                $('#msg').append(msg.sendTime + '<br>');
                $('#msg').append(msg.content + '<br>')
            })
        })
    })
</script>

</body>
</html>

这部分代码和体育栏目的是一样的。

四、测试

我们每次发送的通知都是实时的通知，在做一些有通知的功能时就能用到这个案例，实时的发布系统通知。

五、gitee 案例

地址：ch04 · qiuqiu/WebSocket-study - 码云 - 开源中国 (gitee.com)