WebSocket Day04 : 消息推送

前言

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

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

一、前期准备

1、新建项目,结构如下
2、导入依赖
XML 复制代码
  <!-- 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 类
java 复制代码
@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 类
java 复制代码
@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 类
java 复制代码
@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 类
java 复制代码
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、封装一个信息类
java 复制代码
@Data
@AllArgsConstructor
@NoArgsConstructor
public class Message {

    private String content;
    private String sendTime;

}

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

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

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

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

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

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

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

二、编写 controller 类,实现通知

java 复制代码
@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 --- 发送信息后台
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 --- 体育栏目
java 复制代码
<!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 --- 娱乐栏目
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)

相关推荐
成富1 小时前
文本转SQL(Text-to-SQL),场景介绍与 Spring AI 实现
数据库·人工智能·sql·spring·oracle
鹿屿二向箔2 小时前
基于SSM(Spring + Spring MVC + MyBatis)框架的汽车租赁共享平台系统
spring·mvc·mybatis
半桶水专家2 小时前
用go实现创建WebSocket服务器
服务器·websocket·golang
豪宇刘3 小时前
SpringBoot+Shiro权限管理
java·spring boot·spring
阿尔帕兹3 小时前
构建 HTTP 服务端与 Docker 镜像:从开发到测试
网络协议·http·docker
FeelTouch Labs3 小时前
Netty实现WebSocket Server是否开启压缩深度分析
网络·websocket·网络协议
一只爱打拳的程序猿4 小时前
【Spring】更加简单的将对象存入Spring中并使用
java·后端·spring
千天夜5 小时前
使用UDP协议传输视频流!(分片、缓存)
python·网络协议·udp·视频流
假装我不帅5 小时前
asp.net framework从webform开始创建mvc项目
后端·asp.net·mvc
follycat5 小时前
[极客大挑战 2019]HTTP 1
网络·网络协议·http·网络安全