SpringBoot 实现 Websocket 通信详解

SpringBoot 实现 Websocket 通信详解

2020-06-30 23:49:00 springboot java websocket stomp

文章目录

  !版权声明：本博客内容均为原创,每篇博文作为知识积累,写博不易,转载请注明出处。

参考地址：

示例地址：

环境信息：

  • SpringBoot 版本：2.3.1.RELEASE
  • Spring WebSocket 版本：5.2.7.RELEASE
  • 前端使用的框架：bootstrap 3.4.1、Jquery 3.5.1、sockjs-client 1.4.0、stomp.js 2.3.3

如果对你有帮助，别忘了 Github 点颗星嘞~

一、WebSocket 简介

       WebSocket 是一种基于 TCP 的网络协议。在 2009 年诞生，于 2011 年被 IETF 定为标准 RFC 6455 通信标准，并由 RFC7936 补充规范。WebSocket API 也被 W3C 定为标准。

WebSocket 也是一种全双工通信的协议，既允许客户端向服务器主动发送消息，也允许服务器主动向客户端发送消息。在 WebSocket 中，浏览器和服务器只需要完成一次握手，两者之间就可以建立持久性的连接，进行双向数据传输。

二、WebSocket 特点

  • 连接握手阶段使用 HTTP 协议；
  • 协议标识符是 ws，如果采用加密则是 wss；
  • 数据格式比较轻量，性能开销小，通信高效；
  • 没有同源限制，客户端可以与任意服务器通信；
  • 建立在 TCP 协议之上，服务器端的实现比较容易；
  • 通过 WebSocket 可以发送文本，也可以发送二进制数据；
  • 与 HTTP 协议有着良好的兼容性。默认端口也是 80 和 443，并且握手阶段采用 HTTP 协议，因此握手时不容易屏蔽，能通过各种 HTTP 代理服务器；

三、为什么需要 WebSocket

       谈起为什么需要 WebSocket 前，那得先了解在没有 WebSocket 那段时间说起，那时候基于 Web 的消息基本上是靠 Http 协议进行通信，而经常有"聊天室"、"消息推送"、"股票信 息实时动态"等这样需求，而实现这样的需求常用的有以下几种解决方案：

(1)、短轮询(Traditional Polling)

短轮询是指客户端每隔一段时间就询问一次服务器是否有新的消息，如果有就接收消息。这样方式会增加很多次无意义的发送请求信息，每次都会耗费流量及处理器资源。

  • 优点：短连接，服务器处理简单，支持跨域、浏览器兼容性较好。
  • 缺点：有一定延迟、服务器压力较大，浪费带宽流量、大部分是无效请求。

 ------------------

(2)、长轮询(Long Polling)

长轮询是段轮询的改进，客户端执行 HTTP 请求发送消息到服务器后，等待服务器回应，如果没有新的消息就一直等待，知道服务器有新消息传回或者超时。这也是个反复的过程，这种做法只是减小了网络带宽和处理器的消耗，但是带来的问题是导致消息实时性低，延迟严重。而且也是基于循环，最根本的带宽及处理器资源占用并没有得到有效的解决。

  • 优点：减少轮询次数，低延迟，浏览器兼容性较好。
  • 缺点：服务器需要保持大量连接。

 ------------------

(3)、服务器发送事件(Server-Sent Event)

目前除了 IE/Edge，其他浏览器都支持。

服务器发送事件是一种服务器向浏览器客户端发起数据传输的技术。一旦创建了初始连接，事件流将保持打开状态，直到客户端关闭。该技术通过传统的 HTTP 发送，并具有 WebSockets 缺乏的各种功能，例如"自动重新连接"、"事件ID" 及 "发送任意事件"的能力。

服务器发送事件是单向通道，只能服务器向浏览器发送，因为流信息本质上就是下载。

  • 优点：适用于更新频繁、低延迟并且数据都是从服务端发到客户端。
  • 缺点：浏览器兼容难度高。

 ------------------

显然，上面这几种方式都有各自的优缺点，虽然靠轮询方式能够实现这些一些功能，但是其对性能的开销和低效率是非常致命的，尤其是在移动端流行的现在。现在客户端与服务端双向通信的需求越来越多，且现在的浏览器大部分都支持 WebSocket。所以对实时性和双向通信及其效率有要求的话，比较推荐使用 WebSocket。

四、WebSocket 连接流程

(1)、客户端先用带有 Upgrade:Websocket 请求头的 HTTP 请求，向服务器端发起连接请求，实现握手(HandShake)。

客户端 HTTP 请求的 Header 头信息如下：

1Connection: Upgrade
2Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits
3Sec-WebSocket-Key: IRQYhWINfX5Fh1zdocDl6Q==
4Sec-WebSocket-Version: 13
5Upgrade: websocket
  • Connection： Upgrade 表示要升级协议。
  • Upgrade： Websocket 要升级协议到 websocket 协议。
  • Sec-WebSocket-Extensions： 表示客户端所希望执行的扩展(如消息压缩插件)。
  • Sec-WebSocket-Key： 主要用于WebSocket协议的校验，对应服务端响应头的 Sec-WebSocket-Accept。
  • Sec-WebSocket-Version： 表示 websocket 的版本。如果服务端不支持该版本，需要返回一个Sec-WebSocket-Versionheader，里面包含服务端支持的版本号。

(2)、握手成功后，由 HTTP 协议升级成 Websocket 协议，进行长连接通信，两端相互传递信息。

服务端响应的 HTTP Header 头信息如下：

1Connection: upgrade
2Sec-Websocket-Accept: TSF8/KitM+yYRbXmjclgl7DwbHk=
3Upgrade: websocket
  • Connection： Upgrade 表示要升级协议。
  • Upgrade： Websocket 要升级协议到 websocket 协议。
  • Sec-Websocket-Accept： 对应 Sec-WebSocket-Key 生成的值，主要是返回给客户端，让客户端对此值进行校验，证明服务端支持 WebSocket。

五、WebSocket 使用场景

  • 数据流状态： 比如说上传下载文件，文件进度，文件是否上传成功。
  • 协同编辑文档： 同一份文档，编辑状态得同步到所有参与的用户界面上。
  • 多玩家游戏： 很多游戏都是协同作战的，玩家的操作和状态肯定需要及时同步到所有玩家。
  • 多人聊天： 很多场景下都需要多人参与讨论聊天，用户发送的消息得第一时间同步到所有用户。
  • 社交订阅： 有时候我们需要及时收到订阅消息，比如说开奖通知，比如说在线邀请，支付结果等。
  • 股票虚拟货币价格： 股票和虚拟货币的价格都是实时波动的，价格跟用户的操作息息相关，及时推送对用户跟盘有很大的帮助。

六、WebSocket 中子协议支持

       WebSocket 确实指定了一种消息传递体系结构，但并不强制使用任何特定的消息传递协议。而且它是 TCP 上的一个非常薄的层，它将字节流转换为消息流（文本或二进制）仅此而已。由应用程序来解释消息的含义。

与 HTTP（它是应用程序级协议）不同，在 WebSocket 协议中，传入消息中根本没有足够的信息供框架或容器知道如何路由或处理它。因此，对于非常琐碎的应用程序而言 WebSocket 协议的级别可以说太低了。可以做到的是引导在其上面再创建一层框架。这就相当于当今大多数 Web 应用程序使用的是 Web 框架，而不直接仅使用 Servlet API 进行编码一样。

WebSocket RFC 定义了子协议的使用。在握手过程中，客户机和服务器可以使用头 Sec-WebSocket 协议商定子协议，即使不需要使用子协议，而是用更高的应用程序级协议，但应用程序仍需要选择客户端和服务器都可以理解的消息格式。且该格式可以是自定义的、特定于框架的或标准的消息传递协议。

Spring 框架支持使用 STOMP，这是一个简单的消息传递协议，最初创建用于脚本语言，框架灵感来自 HTTP。STOMP 被广泛支持，非常适合在 WebSocket 和 web 上使用。

七、什么是 STOMP 协议

关于详细的 STOMP 请查看 STOMP 协议规范 了解。

(1)、STOMP 协议概述

       STOMP（Simple Text-Orientated Messaging Protocol）是一种简单的面向文本的消息传递协议。它提供了一个可互操作的连接格式，允许 STOMP 客户端与任意 STOMP 消息代理（Broker）进行交互。STOMP 协议由于设计简单，易于开发客户端，因此在多种语言和多种平台上得到广泛地应用。

(2)、简单介绍可以分为以下几点：

  • STOMP 是基于帧的协议，其帧以 HTTP 为模型。
  • STOMP 框架由命令，一组可选的标头和可选的主体组成。
  • STOMP 基于文本，但也允许传输二进制消息。
  • STOMP 的默认编码为 UTF-8，但它支持消息正文的替代编码的规范。

(3)、STOMP 客户端是一种用户代理，可以分为两种模式运行：

  • 作为生产者，通过 SEND 帧将消息发送到目标服务器上。
  • 作为消费者，对目标地址发送 SUBSCRIBE 帧，并作为 MESSAGE 帧从服务器接收消息。

(4)、STOMP 帧

STOMP 是基于帧的协议，其帧以 HTTP 为模型。STOMP 结构为：

1COMMAND
2header1:value1
3header2:value2
4
5Body^@

客户端可以使用 SEND 或 SUBSCRIBE 命令发送或订阅消息，还可以使用 "destination" 头来描述消息的内容和接收者。这支持一种简单的发布-订阅机制，可用于通过代理将消息发送到其他连接的客户端，或将消息发送到服务器以请求执行某些工作。

(5)、Stomp 常用帧

STOMP 的客户端和服务器之间的通信是通过"帧"（Frame）实现的，每个帧由多"行"（Line）组成，其包含的帧如下：

(6)、Stomp 与 WebSocket 的关系

       直接使用 WebSocket 就很类似于使用 TCP 套接字来编写 Web 应用，因为没有高层级的应用协议（wire protocol），因而就需要我们定义应用之间所发送消息的语义，还需要确保连接的两端都能遵循这些语义。同 HTTP 在 TCP 套接字上添加请求-响应模型层一样，STOMP 在 WebSocket 之上提供了一个基于帧的线路格式层，用来定义消息语义。

(7)、使用 STOMP 作为 WebSocket 子协议的好处

  • 无需发明自定义消息格式
  • 在浏览器中 使用现有的stomp.js客户端
  • 能够根据目的地将消息路由到
  • 可以使用成熟的消息代理（例如RabbitMQ，ActiveMQ等）进行广播的选项
  • 使用STOMP（相对于普通 WebSocket）使 Spring Framework 能够为应用程序级使用提供编程模型，就像 Spring MVC 提供基于 HTTP 的编程模型一样。

八、Spring 封装的 STOMP

       使用 Spring 的 STOMP 支持时，Spring WebSocket 应用程序充当客户端的 STOMP 代理。消息被路由到 @Controller 消息处理方法或简单的内存中代理，该代理跟踪订阅并向订阅的用户广播消息。还可以将 Spring 配置为与专用的 STOMP 代理（例如 RabbitMQ，ActiveM Q等）一起使用，以实际广播消息。在那种情况下，Spring 维护与代理的 TCP 连接，将消息中继到该代理，并将消息从该代理向下传递到已连接的 WebSocket 客户端。因此 Spring Web 应用程序可以依赖基于统一 HTTP 的安全性，通用验证以及熟悉的编程模型消息处理工作。

Spring 官方提供的处理流图:

上面中的一些概念关键词：

  • Message： 消息，里面带有 header 和 payload。
  • MessageHandler： 处理 client 消息的实体。
  • MessageChannel： 解耦消息发送者与消息接收者的实体
    • clientInboundChannel：用于从 WebSocket 客户端接收消息。
    • clientOutboundChannel：用于将服务器消息发送给 WebSocket 客户端。
    • brokerChannel：用于从服务器端、应用程序中向消息代理发送消息
  • Broker： 存放消息的中间件，client 可以订阅 broker 中的消息。

上面的设置包括3个消息通道：

  • clientInboundChannel： 用于来自WebSocket客户端的消息。
  • clientOutboundChannel： 用于向WebSocket客户端发送消息。
  • brokerChannel： 从应用程序内部发送给代理的消息。

九、SpringBoot 实现 WebSocket 示例一：实现简单的广播模式

WebSocket 常分为广播与队列模式，广播模式是向订阅广播的用户发送信息，只要订阅相关广播就能收到对应信息。而队列模式常用于点对点模式，为单个用户向另一个用户发送信息，这里先介绍下广播模式的实现示例。

1、Maven 引入相关依赖

这里使用 Maven 工具管理依赖包，分别引入下面依赖：

  • lombok： Lombok 工具依赖，便于生成实体对象的 Get 与 Set 方法。
  • spring-boot-starter-websocket：SpringBoot 实现 WebSocket 的依赖，里面对 WebSocket 进行了一些列封装，并且也包含了 SpringBoot Web 依赖。
 1<?xml version="1.0" encoding="UTF-8"?>
 2<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 3         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
 4    <modelVersion>4.0.0</modelVersion>
 5
 6    <parent>
 7        <groupId>org.springframework.boot</groupId>
 8        <artifactId>spring-boot-starter-parent</artifactId>
 9        <version>2.3.1.RELEASE</version>
10    </parent>
11
12    <groupId>mydlq.club</groupId>
13    <artifactId>springboot-websocket-topic-example</artifactId>
14    <version>0.0.1</version>
15    <name>springboot-websocket-topic-example</name>
16    <description>Spring Boot WebSocket Example</description>
17
18    <properties>
19        <java.version>1.8</java.version>
20    </properties>
21
22    <dependencies>
23        <!-- SpringBoot WebSocket -->
24        <dependency>
25            <groupId>org.springframework.boot</groupId>
26            <artifactId>spring-boot-starter-websocket</artifactId>
27        </dependency>
28        <!-- Lombok -->
29        <dependency>
30            <groupId>org.projectlombok</groupId>
31            <artifactId>lombok</artifactId>
32            <scope>provided</scope>
33        </dependency>
34    </dependencies>
35
36    <build>
37        <plugins>
38            <plugin>
39                <groupId>org.springframework.boot</groupId>
40                <artifactId>spring-boot-maven-plugin</artifactId>
41            </plugin>
42        </plugins>
43    </build>
44
45</project>

2、创建测试实体类

创建便于传输消息的实体类，里面字段内容如下：

1import lombok.Data;
2
3@Data
4public class MessageBody {
5    /** 消息内容 */
6    private String content;
7    /** 广播转发的目标地址（告知 STOMP 代理转发到哪个地方） */
8    private String destination;
9}

3、创建 WebSocket 配置类

创建 WebSocket 配置类，配置进行连接注册的端点 /mydlq 和消息代理前缀 /topic 及接收客户端发送消息的前缀 /app。

 1import org.springframework.context.annotation.Configuration;
 2import org.springframework.messaging.simp.config.MessageBrokerRegistry;
 3import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
 4import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
 5import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
 6
 7@Configuration
 8@EnableWebSocketMessageBroker
 9public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
10
11    /**
12     * 配置 WebSocket 进入点，及开启使用 SockJS，这些配置主要用配置连接端点，用于 WebSocket 连接
13     *
14     * @param registry STOMP 端点
15     */
16    @Override
17    public void registerStompEndpoints(StompEndpointRegistry registry) {
18        registry.addEndpoint("/mydlq").withSockJS();
19    }
20
21    /**
22     * 配置消息代理选项
23     *
24     * @param registry 消息代理注册配置
25     */
26    @Override
27    public void configureMessageBroker(MessageBrokerRegistry registry) {
28        // 设置一个或者多个代理前缀，在 Controller 类中的方法里面发生的消息，会首先转发到代理从而发送到对应广播或者队列中。
29        registry.enableSimpleBroker("/topic");
30        // 配置客户端发送请求消息的一个或多个前缀，该前缀会筛选消息目标转发到 Controller 类中注解对应的方法里
31        registry.setApplicationDestinationPrefixes("/app");
32    }
33
34}

4、创建测试 Controller 类

创建 Controller 类，该类也类似于正常 Web 项目中 Controller 写法一样，在方法上面添加 @MessageMapping 注解，当客户端发送消息请求的前缀匹配上 WebSocket 配置类中的 /app 前缀后，会进入到 Controller 类中进行匹配，如果匹配成功则执行注解所在的方法内容。

 1import mydlq.club.model.MessageBody;
 2import org.springframework.beans.factory.annotation.Autowired;
 3import org.springframework.messaging.handler.annotation.MessageMapping;
 4import org.springframework.messaging.simp.SimpMessageSendingOperations;
 5import org.springframework.stereotype.Controller;
 6
 7@Controller
 8public class MessageController {
 9
10    /** 消息发送工具对象 */
11    @Autowired
12    private SimpMessageSendingOperations simpMessageSendingOperations;
13
14    /** 广播发送消息，将消息发送到指定的目标地址 */
15    @MessageMapping("/test")
16    public void sendTopicMessage(MessageBody messageBody) {
17        // 将消息发送到 WebSocket 配置类中配置的代理中（/topic）进行消息转发
18        simpMessageSendingOperations.convertAndSend(messageBody.getDestination(), messageBody);
19    }
20
21}

5、SpringBoot 启动类

 1import org.springframework.boot.SpringApplication;
 2import org.springframework.boot.autoconfigure.SpringBootApplication;
 3
 4@SpringBootApplication
 5public class Application {
 6
 7    public static void main(String[] args) {
 8        SpringApplication.run(Application.class, args);
 9    }
10
11}

6、创建测试的前端 html 和 js

(1)、创建 WebSocket JS

创建用于操作 WebSocket 的 JS 文件 app-websocket.js，内容如下：

 1// 设置 STOMP 客户端
 2var stompClient = null;
 3// 设置 WebSocket 进入端点
 4var SOCKET_ENDPOINT = "/mydlq";
 5// 设置订阅消息的请求前缀
 6var SUBSCRIBE_PREFIX = "/topic"
 7// 设置订阅消息的请求地址
 8var SUBSCRIBE = "";
 9// 设置服务器端点，访问服务器中哪个接口
10var SEND_ENDPOINT = "/app/test";
11
12/* 进行连接 */
13function connect() {
14    // 设置 SOCKET
15    var socket = new SockJS(SOCKET_ENDPOINT);
16    // 配置 STOMP 客户端
17    stompClient = Stomp.over(socket);
18    // STOMP 客户端连接
19    stompClient.connect({}, function (frame) {
20        alert("连接成功");
21    });
22}
23
24/* 订阅信息 */
25function subscribeSocket(){
26    // 设置订阅地址
27    SUBSCRIBE = SUBSCRIBE_PREFIX + $("#subscribe").val();
28    // 输出订阅地址
29    alert("设置订阅地址为：" + SUBSCRIBE);
30    // 执行订阅消息
31    stompClient.subscribe(SUBSCRIBE, function (responseBody) {
32        var receiveMessage = JSON.parse(responseBody.body);
33        $("#information").append("<tr><td>" + receiveMessage.content + "</td></tr>");
34    });
35}
36
37/* 断开连接 */
38function disconnect() {
39    stompClient.disconnect(function() {
40        alert("断开连接");
41    });
42}
43
44/* 发送消息并指定目标地址（这里设置的目标地址为自身订阅消息的地址，当然也可以设置为其它地址） */
45function sendMessageNoParameter() {
46    // 设置发送的内容
47    var sendContent = $("#content").val();
48    // 设置待发送的消息内容
49    var message = '{"destination": "' + SUBSCRIBE + '", "content": "' + sendContent + '"}';
50    // 发送消息
51    stompClient.send(SEND_ENDPOINT, {}, message);
52}

(2)、创建 WebSocket HTML

创建用于展示 WebSocket 相关功能的 WEB HTML 页面 index.html，内容如下：

 1<!DOCTYPE html>
 2<html>
 3<head>
 4    <title>Hello WebSocket</title>
 5    <link href="https://cdn.bootcdn.net/ajax/libs/twitter-bootstrap/3.4.1/css/bootstrap.min.css" rel="stylesheet">
 6    <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
 7    <script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.5.1/jquery.js"></script>
 8    <script src="https://cdn.bootcdn.net/ajax/libs/sockjs-client/1.4.0/sockjs.min.js"></script>
 9    <script src="https://cdn.bootcdn.net/ajax/libs/stomp.js/2.3.3/stomp.min.js"></script>
10    <script src="app-websocket.js"></script>
11</head>
12<body>
13    <div id="main-content" class="container" style="margin-top: 10px;">
14        <div class="row">
15            <form class="navbar-form" style="margin-left:0px">
16                <div class="col-md-12">
17                    <div class="form-group">
18                        <label>WebSocket 连接：</label>
19                        <button class="btn btn-primary" type="button" onclick="connect();">进行连接</button>
20                        <button class="btn btn-danger" type="button" onclick="disconnect();">断开连接</button>
21                    </div>
22                    <label>订阅地址：</label>
23                    <div class="form-group">
24                        <input type="text" id="subscribe" class="form-control" placeholder="订阅地址">
25                    </div>
26                    <button class="btn btn-warning" onclick="subscribeSocket();" type="button">订阅</button>
27                </div>
28            </form>
29        </div>
30        </br>
31        <div class="row">
32            <div class="form-group">
33                <label for="content">发送的消息内容：</label>
34                <input type="text" id="content" class="form-control" placeholder="消息内容">
35            </div>
36            <button class="btn btn-info" onclick="sendMessageNoParameter();" type="button">发送</button>
37        </div>
38        </br>
39        <div class="row">
40            <div class="col-md-12">
41                <h5 class="page-header" style="font-weight:bold">接收到的消息：</h5>
42                <table class="table table-striped">
43                    <tbody id="information"></tbody>
44                </table>
45            </div>
46        </div>
47    </div>
48</body>
49</html>

7、启动并进行测试

输入地址 http://localhost:8080/index.html 访问测试的前端页面，然后执行下面步骤进行测试：

  • (1)、点击"进行连接"按钮，连接 WebSocket 服务端；
  • (2)、在订阅地址栏输入订阅地址（因为本人设置的订阅地址和接收消息的地址是一个，所以随意输入）；
  • (3)、点击订阅按钮订阅对应地址的消息；
  • (4)、在发送消息内容的输入框中输入"hello world!"，然后点击"发送"按钮发送消息；

执行完上面步骤成后，可以观察到成功接收到订阅地址的消息，如下：

十、SpringBoot 实现 WebSocket 示例二：实现点对点模式(引入 Spring Security 实现鉴权)

1、Maven 引入相关依赖

这里使用 Maven 工具管理依赖包，分别引入下面依赖：

  • lombok： Lombok 工具依赖，便于生成实体对象的 Get 与 Set 方法。
  • spring-boot-starter-websocket：SpringBoot 实现 WebSocket 的依赖，里面对 WebSocket 进行了一些列封装，并且也包含了 SpringBoot Web 依赖。
  • spring-boot-starter-security：Spring Security，这是一种基于 Spring AOP 和 Servlet 过滤器的安全框架。它提供全面的安全性解决方案，同时在 Web 请求级和方法调用级处理身份确认和授权。
 1<?xml version="1.0" encoding="UTF-8"?>
 2<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 3         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
 4    <modelVersion>4.0.0</modelVersion>
 5
 6    <parent>
 7        <groupId>org.springframework.boot</groupId>
 8        <artifactId>spring-boot-starter-parent</artifactId>
 9        <version>2.3.1.RELEASE</version>
10    </parent>
11
12    <groupId>mydlq.club</groupId>
13    <artifactId>springboot-websocket-queue-security-example</artifactId>
14    <version>0.0.1</version>
15    <name>springboot-websocket-queue-security-example</name>
16    <description>Spring Boot WebSocket Example</description>
17
18    <properties>
19        <java.version>1.8</java.version>
20    </properties>
21
22    <dependencies>
23        <!-- SpringBoot WebSocket -->
24        <dependency>
25            <groupId>org.springframework.boot</groupId>
26            <artifactId>spring-boot-starter-websocket</artifactId>
27        </dependency>
28        <!-- Security -->
29        <dependency>
30            <groupId>org.springframework.boot</groupId>
31            <artifactId>spring-boot-starter-security</artifactId>
32        </dependency>
33        <!-- Lombok -->
34        <dependency>
35            <groupId>org.projectlombok</groupId>
36            <artifactId>lombok</artifactId>
37            <scope>provided</scope>
38        </dependency>
39    </dependencies>
40
41    <build>
42        <plugins>
43            <plugin>
44                <groupId>org.springframework.boot</groupId>
45                <artifactId>spring-boot-maven-plugin</artifactId>
46            </plugin>
47        </plugins>
48    </build>
49
50</project>

2、创建测试实体类

创建便于传输消息的实体类，里面字段内容如下：

 1import lombok.Data;
 2
 3@Data
 4public class MessageBody {
 5    /** 发送消息的用户 */
 6    private String from;
 7    /** 消息内容 */
 8    private String content;
 9    /** 目标用户（告知 STOMP 代理转发到哪个用户） */
10    private String targetUser;
11    /** 广播转发的目标地址（告知 STOMP 代理转发到哪个地方） */
12    private String destination;
13}

3、创建 WebSocket 配置类

创建 WebSocket 配置类，配置进行连接注册的端点 /mydlq 和消息代理前缀 /queue 及接收客户端发送消息的前缀 /app。

 1import org.springframework.context.annotation.Configuration;
 2import org.springframework.messaging.simp.config.MessageBrokerRegistry;
 3import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
 4import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
 5import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
 6
 7@Configuration
 8@EnableWebSocketMessageBroker
 9public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
10
11    /**
12     * 配置 WebSocket 进入点，及开启使用 SockJS，这些配置主要用配置连接端点，用于 WebSocket 连接
13     *
14     * @param registry STOMP 端点
15     */
16    @Override
17    public void registerStompEndpoints(StompEndpointRegistry registry) {
18        registry.addEndpoint("/mydlq").withSockJS();
19    }
20
21    /**
22     * 配置消息代理选项
23     *
24     * @param registry 消息代理注册配置
25     */
26    @Override
27    public void configureMessageBroker(MessageBrokerRegistry registry) {
28        // 设置一个或者多个代理前缀，在 Controller 类中的方法里面发生的消息，会首先转发到代理从而发送到对应广播或者队列中。
29        registry.enableSimpleBroker("/queue");
30        // 配置客户端发送请求消息的一个或多个前缀，该前缀会筛选消息目标转发到 Controller 类中注解对应的方法里
31        registry.setApplicationDestinationPrefixes("/app");
32        // 服务端通知特定用户客户端的前缀，可以不设置，默认为user
33        registry.setUserDestinationPrefix("/user");
34    }
35
36}

4、创建 Security 配置

Spring Security 的配置类，可以在该类中配置权限认证及测试的两个用户相关信息：

  • 测试用户名/密码1：mydlq1/123456
  • 测试用户名/密码2：mydlq2/123456
 1import org.springframework.context.annotation.Bean;
 2import org.springframework.context.annotation.Configuration;
 3import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder;
 4import org.springframework.security.config.annotation.web.builders.HttpSecurity;
 5import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter;
 6import org.springframework.security.crypto.password.NoOpPasswordEncoder;
 7import org.springframework.security.crypto.password.PasswordEncoder;
 8
 9@Configuration
10public class SecurityConfig extends WebSecurityConfigurerAdapter {
11
12    /**
13     * 设置密码编码的配置参数，这里设置为 NoOpPasswordEncoder，不配置密码加密，方便测试。
14     *
15     * @return 密码编码实例
16     */
17    @Bean
18    PasswordEncoder passwordEncoder() {
19        return NoOpPasswordEncoder.getInstance();
20    }
21
22    /**
23     * 设置权限认证参数，这里用于创建两个用于测试的用户信息。
24     *
25     * @param auth SecurityBuilder 用于创建 AuthenticationManager。
26     * @throws Exception 抛出的异常
27     */
28    @Override
29    protected void configure(AuthenticationManagerBuilder auth) throws Exception {
30        auth.inMemoryAuthentication()
31                .withUser("mydlq1")
32                .password("123456")
33                .roles("admin")
34                .and()
35                .withUser("mydlq2")
36                .password("123456")
37                .roles("admin");
38    }
39
40    /**
41     * 设置 HTTP 安全相关配置参数
42     *
43     * @param http HTTP Security 对象
44     * @throws Exception 抛出的异常信息
45     */
46    @Override
47    protected void configure(HttpSecurity http) throws Exception {
48        http.authorizeRequests()
49                .anyRequest().authenticated()
50                .and()
51                .formLogin()
52                .permitAll();
53    }
54
55}

5、创建测试 Controller 类

跟上面介绍广播模式一样，作用也是根据 WebSocket 配置类中 /app 前缀匹配后进入 Controller 类进行逻辑处理操作。

 1import mydlq.club.example.model.MessageBody;
 2import org.springframework.beans.factory.annotation.Autowired;
 3import org.springframework.messaging.handler.annotation.MessageMapping;
 4import org.springframework.messaging.simp.SimpMessageSendingOperations;
 5import org.springframework.stereotype.Controller;
 6import java.security.Principal;
 7
 8@Controller
 9public class MessageController {
10
11    @Autowired
12    private SimpMessageSendingOperations simpMessageSendingOperations;
13
14    /**
15     * 点对点发送消息，将消息发送到指定用户
16     */
17    @MessageMapping("/test")
18    public void sendUserMessage(Principal principal, MessageBody messageBody) {
19        // 设置发送消息的用户
20        messageBody.setFrom(principal.getName());
21        // 调用 STOMP 代理进行消息转发
22        simpMessageSendingOperations.convertAndSendToUser(messageBody.getTargetUser(), messageBody.getDestination(), messageBody);
23    }
24
25}

6、SpringBoot 启动类

 1import org.springframework.boot.SpringApplication;
 2import org.springframework.boot.autoconfigure.SpringBootApplication;
 3
 4@SpringBootApplication
 5public class Application {
 6
 7    public static void main(String[] args) {
 8        SpringApplication.run(Application.class, args);
 9    }
10
11}

7、创建测试的前端 html 和 js

(1)、创建 WebSocket JS

创建用于操作 WebSocket 的 JS 文件 app-websocket.js，内容如下：

 1// 设置 STOMP 客户端
 2var stompClient = null;
 3
 4// 设置 WebSocket 进入端点
 5var SOCKET_ENDPOINT = "/mydlq";
 6// 设置订阅消息的请求地址前缀
 7var SUBSCRIBE_PREFIX  = "/queue";
 8// 设置订阅地址
 9var SUBSCRIBE = "";
10// 设置服务器端点，访问服务器中哪个接口
11var SEND_ENDPOINT = "/app/test";
12
13/* 进行连接 */
14function connect() {
15    // 设置 SOCKET
16    var socket = new SockJS(SOCKET_ENDPOINT);
17    // 配置 STOMP 客户端
18    stompClient = Stomp.over(socket);
19    // STOMP 客户端连接
20    stompClient.connect({}, function (frame) {
21        alert("连接成功");
22    });
23}
24
25/* 订阅信息 */
26function subscribeSocket(){
27    // 设置订阅地址
28    SUBSCRIBE = SUBSCRIBE_PREFIX + $("#subscribe").val();
29    // 输出订阅地址
30    alert("设置订阅地址为：" + SUBSCRIBE);
31    // 执行订阅消息
32    stompClient.subscribe("/user" + SUBSCRIBE, function (responseBody) {
33        var receiveMessage = JSON.parse(responseBody.body);
34        console.log(receiveMessage);
35        $("#information").append("<tr><td>" + receiveMessage.content + "</td></tr>");
36    });
37}
38
39/* 断开连接 */
40function disconnect() {
41    stompClient.disconnect(function() {
42        alert("断开连接");
43    });
44}
45
46/* 发送消息并指定目标地址 */
47function sendMessageNoParameter() {
48    // 设置发送的内容
49    var sendContent = $("#content").val();
50    // 设置发送的用户
51    var sendUser = $("#targetUser").val();
52    // 设置待发送的消息内容
53    var message = '{"targetUser":"' + sendUser + '", "destination": "' + SUBSCRIBE + '", "content": "' + sendContent + '"}';
54    // 发送消息
55    stompClient.send(SEND_ENDPOINT, {}, message);
56}

(2)、创建 WebSocket HTML

创建用于展示 WebSocket 相关功能的 WEB HTML 页面 index.html，内容如下：

 1<!DOCTYPE html>
 2<html>
 3<head>
 4    <title>Hello WebSocket</title>
 5    <link href="https://cdn.bootcdn.net/ajax/libs/twitter-bootstrap/3.4.1/css/bootstrap.min.css" rel="stylesheet">
 6    <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
 7    <script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.5.1/jquery.js"></script>
 8    <script src="https://cdn.bootcdn.net/ajax/libs/sockjs-client/1.4.0/sockjs.min.js"></script>
 9    <script src="https://cdn.bootcdn.net/ajax/libs/stomp.js/2.3.3/stomp.min.js"></script>
10    <script src="app-websocket.js"></script>
11</head>
12<body>
13<div id="main-content" class="container" style="margin-top: 10px;">
14    <div class="row">
15        <form class="navbar-form" style="margin-left:0px">
16            <div class="col-md-12">
17                <div class="form-group">
18                    <label>WebSocket 连接：</label>
19                    <button class="btn btn-primary" type="button" onclick="connect();">进行连接</button>
20                    <button class="btn btn-danger" type="button" onclick="disconnect();">断开连接</button>
21                </div>
22                <label>订阅地址：</label>
23                <div class="form-group">
24                    <input type="text" id="subscribe" class="form-control" placeholder="订阅地址">
25                </div>
26                <button class="btn btn-warning" onclick="subscribeSocket();" type="button">订阅</button>
27            </div>
28        </form>
29    </div>
30    </br>
31    <div class="row">
32        <div class="form-group">
33            <label>发送的目标用户：</label>
34            <input type="text" id="targetUser" class="form-control" placeholder="发送的用户">
35            <label for="content">发送的消息内容：</label>
36            <input type="text" id="content" class="form-control" placeholder="消息的内容">
37        </div>
38        <button class="btn btn-info" onclick="sendMessageNoParameter();" type="button">发送</button>
39    </div>
40    </br>
41    <div class="row">
42        <div class="col-md-12">
43            <h5 class="page-header" style="font-weight:bold">接收到的消息：</h5>
44            <table class="table table-striped">
45                <tbody id="information"></tbody>
46            </table>
47        </div>
48    </div>
49</div>
50</body>
51</html>

8、启动并进行测试

为了方便测试，需要打开两个不同类型浏览器（因为用户登录后会存 Session，如果一个浏览器不同用户登录会使之前 Session 失效）来进行测试，两个浏览器同时输入地址 http://localhost:8080/index.html 访问测试的前端页面，然后可以看到并没有进入 /index.html 页面，而是跳转到 Spring Security 提供的登录的 /login 页面，如下：

  • 浏览器1登录的用户：mydlq1，密码：123456
  • 浏览器2登录的用户：mydlq2，密码：123456

两个浏览器中都输入用户名/密码 mydlq1/123456 与 mydlq2/123456 进行登录，然后会回到 /index.html 页面，然后执行下面步骤进行测试：

  • (1)、"浏览器1"和"浏览器2"点击"进行连接"按钮，连接 WebSocket 服务端；
  • (2)、"浏览器1"和"浏览器2"中同时设置订阅地址为"/abc"，然后点击订阅按钮进行消息订阅；
  • (3)、"浏览器1"(用户 mydlq1)设置发送目标用户为"/mydlq2"，"浏览器2"(用户 mydlq2)设置发送目标用户为"/mydlq1"；
  • (4)、"浏览器1"(用户 mydlq1)设置发送消息为"Hi, I'm mydlq1"，"浏览器2"(用户 mydlq2)设置发送消息为"Hi, I'm mydlq2"；
  • (5)、点击"发送"按钮发送消息；

执行完上面步骤成后，可以在两个不同浏览器中观察到如下内容：

十一、SpringBoot 实现 WebSocket 示例三：实现点对点模式(根据请求头 Header 实现鉴权)

1、Maven 引入相关依赖

这里使用 Maven 工具管理依赖包，分别引入下面依赖：

  • lombok： Lombok 工具依赖，便于生成实体对象的 Get 与 Set 方法。
  • spring-boot-starter-security：Spring Security，这是一种基于 Spring AOP 和 Servlet 过滤器的安全框架。它提供全面的安全性解决方案，同时在 Web 请求级和方法调用级处理身份确认和授权。
 1<?xml version="1.0" encoding="UTF-8"?>
 2<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 3         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
 4    <modelVersion>4.0.0</modelVersion>
 5
 6    <parent>
 7        <groupId>org.springframework.boot</groupId>
 8        <artifactId>spring-boot-starter-parent</artifactId>
 9        <version>2.3.1.RELEASE</version>
10    </parent>
11
12    <groupId>mydlq.club</groupId>
13    <artifactId>springboot-websocket-queue-security-custom-example</artifactId>
14    <version>0.0.1</version>
15    <name>springboot-websocket-queue-security-custom-example</name>
16    <description>Spring Boot WebSocket Example</description>
17
18    <properties>
19        <java.version>1.8</java.version>
20    </properties>
21
22    <dependencies>
23        <!-- SpringBoot WebSocket -->
24        <dependency>
25            <groupId>org.springframework.boot</groupId>
26            <artifactId>spring-boot-starter-websocket</artifactId>
27        </dependency>
28        <!-- Lombok -->
29        <dependency>
30            <groupId>org.projectlombok</groupId>
31            <artifactId>lombok</artifactId>
32            <scope>provided</scope>
33        </dependency>
34    </dependencies>
35
36    <build>
37        <plugins>
38            <plugin>
39                <groupId>org.springframework.boot</groupId>
40                <artifactId>spring-boot-maven-plugin</artifactId>
41            </plugin>
42        </plugins>
43    </build>
44
45</project>

2、创建测试实体类

MessageBody

 1import lombok.Data;
 2
 3@Data
 4public class MessageBody {
 5    /** 发送消息的用户 */
 6    private String from;
 7    /** 消息内容 */
 8    private String content;
 9    /** 目标用户（告知 STOMP 代理转发到哪个用户） */
10    private String targetUser;
11    /** 广播转发的目标地址（告知 STOMP 代理转发到哪个地方） */
12    private String destination;
13}

User

1import lombok.AllArgsConstructor;
2import lombok.Data;
3
4@Data
5@AllArgsConstructor
6public class User {
7    private String username;
8    private String token;
9}

3、配置 WebSocket 通道拦截器

配置 WebSocket 通道拦截器，里面添加两个模拟用户：

  • 用户 mydlq1，Token：123456-1
  • 用户 mydlq2，Token：123456-2
 1import mydlq.club.model.User;
 2import org.springframework.messaging.Message;
 3import org.springframework.messaging.MessageChannel;
 4import org.springframework.messaging.simp.stomp.StompCommand;
 5import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
 6import org.springframework.messaging.support.ChannelInterceptor;
 7import org.springframework.messaging.support.MessageHeaderAccessor;
 8import java.security.Principal;
 9import java.util.List;
10import java.util.Map;
11
12/**
13 * WebSocket 通道拦截器（这里模拟两个测试 Token 方便测试，不做具体 Token 鉴权实现）
14 *
15 * @author mydlq
16 */
17public class MyChannelInterceptor implements ChannelInterceptor {
18
19    /** 测试用户与 token 1 */
20    private User mydlq1 = new User("","123456-1");
21    /** 测试用户与 token 2 */
22    private User mydlq2 = new User("","123456-2");
23
24    /**
25     * 从 Header 中获取 Token 进行验证，根据不同的 Token 区别用户
26     *
27     * @param message 消息对象
28     * @param channel 通道对象
29     * @return 验证后的用户信息
30     */
31    @Override
32    public Message<?> preSend(Message<?> message, MessageChannel channel) {
33        StompHeaderAccessor accessor = MessageHeaderAccessor.getAccessor(message, StompHeaderAccessor.class);
34        String token = getToken(message);
35        if (token!=null && accessor != null && StompCommand.CONNECT.equals(accessor.getCommand())) {
36            Principal user = null;
37            // 提前创建好两个测试 token 进行匹配，方便测试
38            if (mydlq1.getToken().equals(token)){
39                user = () -> mydlq1.getUsername();
40            } else if (mydlq2.getToken().equals(token)){
41                user = () -> mydlq2.getUsername();
42            }
43            accessor.setUser(user);
44        }
45        return message;
46    }
47
48    /**
49     * 从 Header 中获取 TOKEN
50     *
51     * @param message 消息对象
52     * @return TOKEN
53     */
54    private String getToken(Message<?> message){
55        Map<String,Object> headers = (Map<String, Object>) message.getHeaders().get("nativeHeaders");
56        if (headers !=null && headers.containsKey("token")){
57            List<String> token = (List<String>)headers.get("token");
58            return String.valueOf(token.get(0));
59        }
60        return null;
61    }
62
63}

4、创建 WebSocket 配置类

创建 WebSocket 配置类，配置进行连接注册的端点 /mydlq 和消息代理前缀 /queue 及接收客户端发送消息的前缀 /app。

 1import org.springframework.context.annotation.Configuration;
 2import org.springframework.messaging.simp.config.ChannelRegistration;
 3import org.springframework.messaging.simp.config.MessageBrokerRegistry;
 4import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
 5import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
 6import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
 7
 8@Configuration
 9@EnableWebSocketMessageBroker
10public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
11
12    /**
13     * 配置 WebSocket 进入点，及开启使用 SockJS，这些配置主要用配置连接端点，用于 WebSocket 连接
14     *
15     * @param registry STOMP 端点
16     */
17    @Override
18    public void registerStompEndpoints(StompEndpointRegistry registry) {
19        registry.addEndpoint("/mydlq").withSockJS();
20    }
21
22    /**
23     * 配置消息代理选项
24     *
25     * @param registry 消息代理注册配置
26     */
27    @Override
28    public void configureMessageBroker(MessageBrokerRegistry registry) {
29        // 设置一个或者多个代理前缀，在 Controller 类中的方法里面发生的消息，会首先转发到代理从而发送到对应广播或者队列中。
30        registry.enableSimpleBroker("/queue");
31        // 配置客户端发送请求消息的一个或多个前缀，该前缀会筛选消息目标转发到 Controller 类中注解对应的方法里
32        registry.setApplicationDestinationPrefixes("/app");
33        // 服务端通知特定用户客户端的前缀，可以不设置，默认为user
34        registry.setUserDestinationPrefix("/user");
35    }
36
37    /**
38     * 配置通道拦截器，用于获取 Header 的 Token 进行鉴权
39     *
40     * @param registration 注册通道配置类
41     */
42    @Override
43    public void configureClientInboundChannel(ChannelRegistration registration) {
44        registration.interceptors(new MyChannelInterceptor());
45    }
46
47}

5、创建测试 Controller 类

 1import mydlq.club.example.model.MessageBody;
 2import org.springframework.beans.factory.annotation.Autowired;
 3import org.springframework.messaging.handler.annotation.MessageMapping;
 4import org.springframework.messaging.simp.SimpMessageSendingOperations;
 5import org.springframework.stereotype.Controller;
 6import java.security.Principal;
 7
 8@Controller
 9public class MessageController {
10
11    @Autowired
12    private SimpMessageSendingOperations simpMessageSendingOperations;
13
14    /**
15     * 点对点发送消息，将消息发送到指定用户
16     */
17    @MessageMapping("/test")
18    public void sendUserMessage(Principal principal, MessageBody messageBody) {
19        // 设置发送消息的用户
20        messageBody.setFrom(principal.getName());
21        // 调用 STOMP 代理进行消息转发
22        simpMessageSendingOperations.convertAndSendToUser(messageBody.getTargetUser(), messageBody.getDestination(), messageBody);
23    }
24
25}

6、SpringBoot 启动类

 1import org.springframework.boot.SpringApplication;
 2import org.springframework.boot.autoconfigure.SpringBootApplication;
 3
 4@SpringBootApplication
 5public class Application {
 6
 7    public static void main(String[] args) {
 8        SpringApplication.run(Application.class, args);
 9    }
10
11}

7、创建测试的前端 html 和 js

(1)、创建 WebSocket JS

创建用于操作 WebSocket 的 JS 文件 app-websocket.js，内容如下：

 1// 设置 STOMP 客户端
 2var stompClient = null;
 3
 4// 设置 WebSocket 进入端点
 5var SOCKET_ENDPOINT = "/mydlq";
 6// 设置订阅消息的请求地址前缀
 7var SUBSCRIBE_PREFIX  = "/queue";
 8// 设置订阅地址
 9var SUBSCRIBE = "";
10// 设置服务器端点，访问服务器中哪个接口
11var SEND_ENDPOINT = "/app/test";
12
13/* 进行连接 */
14function connect() {
15    // 设置 SOCKET
16    var socket = new SockJS(SOCKET_ENDPOINT);
17    // 配置 STOMP 客户端
18    stompClient = Stomp.over(socket);
19    // 获取 TOKEN
20    var myToken = $("#myToken").val();
21    // STOMP 客户端连接
22    stompClient.connect({token: myToken}, function (frame) {
23        alert("连接成功");
24    });
25}
26
27/* 订阅信息 */
28function subscribeSocket(){
29    // 设置订阅地址
30    SUBSCRIBE = SUBSCRIBE_PREFIX + $("#subscribe").val();
31    // 输出订阅地址
32    alert("设置订阅地址为：" + SUBSCRIBE);
33    // 执行订阅消息
34    stompClient.subscribe("/user" + SUBSCRIBE, function (responseBody) {
35        var receiveMessage = JSON.parse(responseBody.body);
36        console.log(receiveMessage);
37        $("#information").append("<tr><td>" + receiveMessage.content + "</td></tr>");
38    });
39}
40
41/* 断开连接 */
42function disconnect() {
43    stompClient.disconnect(function() {
44        alert("断开连接");
45    });
46}
47
48/* 发送消息并指定目标地址 */
49function sendMessageNoParameter() {
50    // 设置发送的内容
51    var sendContent = $("#content").val();
52    // 设置发送的用户
53    var sendUser = $("#targetUser").val();
54    // 设置待发送的消息内容
55    var message = '{"targetUser":"' + sendUser + '", "destination": "' + SUBSCRIBE + '", "content": "' + sendContent + '"}';
56    // 发送消息
57    stompClient.send(SEND_ENDPOINT, {}, message);
58}

(2)、创建 WebSocket HTML

创建用于展示 WebSocket 相关功能的 WEB HTML 页面 index.html，内容如下：

 1<!DOCTYPE html>
 2<html>
 3<head>
 4    <title>Hello WebSocket</title>
 5    <link href="https://cdn.bootcdn.net/ajax/libs/twitter-bootstrap/3.4.1/css/bootstrap.min.css" rel="stylesheet">
 6    <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
 7    <script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.5.1/jquery.js"></script>
 8    <script src="https://cdn.bootcdn.net/ajax/libs/sockjs-client/1.4.0/sockjs.min.js"></script>
 9    <script src="https://cdn.bootcdn.net/ajax/libs/stomp.js/2.3.3/stomp.min.js"></script>
10    <script src="app-websocket.js"></script>
11</head>
12<body>
13<div id="main-content" class="container" style="margin-top: 10px;">
14    <div class="row">
15        <form class="navbar-form" style="margin-left:0px">
16            <div class="col-md-12">
17                <div class="form-group">
18                    <label>WebSocket 连接：</label>
19                    <button class="btn btn-primary" type="button" onclick="connect();">进行连接</button>
20                    <button class="btn btn-danger" type="button" onclick="disconnect();">断开连接</button>
21                </div>
22                <label>订阅地址：</label>
23                <div class="form-group">
24                    <input type="text" id="subscribe" class="form-control" placeholder="订阅地址">
25                </div>
26                <button class="btn btn-warning" onclick="subscribeSocket();" type="button">订阅</button>
27            </div>
28        </form>
29    </div>
30    </br>
31    <div class="row">
32        <div class="form-group">
33            <label>TOKEN 信息：</label>
34            <input type="text" id="myToken" class="form-control" placeholder="TOKEN 信息">
35            <label>发送的目标用户：</label>
36            <input type="text" id="targetUser" class="form-control" placeholder="发送的用户">
37            <label for="content">发送的消息内容：</label>
38            <input type="text" id="content" class="form-control" placeholder="消息的内容">
39        </div>
40        <button class="btn btn-info" onclick="sendMessageNoParameter();" type="button">发送</button>
41    </div>
42    </br>
43    <div class="row">
44        <div class="col-md-12">
45            <h5 class="page-header" style="font-weight:bold">接收到的消息：</h5>
46            <table class="table table-striped">
47                <tbody id="information"></tbody>
48            </table>
49        </div>
50    </div>
51</div>
52</body>
53</html>

8、启动并进行测试

为了方便测试，需要打开两个不同类型浏览器（这里模拟通过 Header 传 Token 的方式进行用户验证，具体登录逻辑不实现，而是直接使用事先配置好的两个用户 Token 进行模拟）来进行测试，两个浏览器同时输入地址 http://localhost:8080/index.html 访问测试的前端页面 /index.html 如下：

  • 浏览器1：
    • 用户：mydlq1
    • Token：123456789-1
  • 浏览器2：
    • 登录的用户：mydlq2
    • Token：123456789-2

两个浏览器中都执行下面步骤进行测试：

  • (1)、"浏览器1"和"浏览器2"点击"进行连接"按钮，连接 WebSocket 服务端；
  • (2)、"浏览器1"和"浏览器2"中同时设置订阅地址为"/abc"，然后点击订阅按钮进行消息订阅；
  • (3)、"浏览器1"(用户 mydlq1)在 TOken 信息一栏中填写模拟用户 mydlq1 的 Token 串，"浏览器2"(用户 mydlq2)填写模拟用户 mydlq2 的 Token 串；
  • (4)、"浏览器1"(用户 mydlq1)设置发送目标用户为"/mydlq2"，"浏览器2"(用户 mydlq2)设置发送目标用户为"/mydlq1"；
  • (5)、"浏览器1"(用户 mydlq1)设置发送消息为"Hi, I'm mydlq1"，"浏览器2"(用户 mydlq2)设置发送消息为"Hi, I'm mydlq2"；
  • (6)、点击"发送"按钮发送消息；

执行完上面步骤成后，可以在两个不同浏览器中观察到如下内容：

十二、SpringBoot 实现 WebSocket 示例四：实现点对点模式(根据 HTTP Session 实现鉴权)

1、Maven 引入相关依赖

这里使用 Maven 工具管理依赖包，分别引入下面依赖：

  • lombok： Lombok 工具依赖，便于生成实体对象的 Get 与 Set 方法。
  • spring-boot-starter-websocket：SpringBoot 实现 WebSocket 的依赖，里面对 WebSocket 进行了一些列封装，并且也包含了 SpringBoot Web 依赖。
 1<?xml version="1.0" encoding="UTF-8"?>
 2<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 3         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
 4    <modelVersion>4.0.0</modelVersion>
 5
 6    <parent>
 7        <groupId>org.springframework.boot</groupId>
 8        <artifactId>spring-boot-starter-parent</artifactId>
 9        <version>2.3.1.RELEASE</version>
10    </parent>
11
12    <groupId>mydlq.club</groupId>
13    <artifactId>springboot-websocket-queue-security-example</artifactId>
14    <version>0.0.1</version>
15    <name>springboot-websocket-queue-security-example</name>
16    <description>Spring Boot WebSocket Example</description>
17
18    <properties>
19        <java.version>1.8</java.version>
20    </properties>
21
22    <dependencies>
23        <!-- SpringBoot WebSocket -->
24        <dependency>
25            <groupId>org.springframework.boot</groupId>
26            <artifactId>spring-boot-starter-websocket</artifactId>
27        </dependency>
28        <!-- Lombok -->
29        <dependency>
30            <groupId>org.projectlombok</groupId>
31            <artifactId>lombok</artifactId>
32            <scope>provided</scope>
33        </dependency>
34    </dependencies>
35
36    <build>
37        <plugins>
38            <plugin>
39                <groupId>org.springframework.boot</groupId>
40                <artifactId>spring-boot-maven-plugin</artifactId>
41            </plugin>
42        </plugins>
43    </build>
44
45</project>

2、创建测试实体类

MessageBody

 1import lombok.Data;
 2
 3@Data
 4public class MessageBody {
 5    /** 发送消息的用户 */
 6    private String from;
 7    /** 消息内容 */
 8    private String content;
 9    /** 目标用户（告知 STOMP 代理转发到哪个用户） */
10    private String targetUser;
11    /** 广播转发的目标地址（告知 STOMP 代理转发到哪个地方） */
12    private String destination;
13}

User

1import lombok.Data;
2
3@Data
4public class User {
5    private String username;
6    private String password;
7    private String name;
8    private String sex;
9}

3、WebSocket 握手请求拦截器

WebSocket 握手请求拦截器，可用于拦截握手请求和响应，以及将 HTTP 属性传递到目标 WebSocketHandler，这里主要是设置捂手时候，从 HTTP Session 中获取用户信息，传入 WebSocket 信息中。

 1import mydlq.club.model.User;
 2import org.springframework.http.server.ServerHttpRequest;
 3import org.springframework.http.server.ServerHttpResponse;
 4import org.springframework.http.server.ServletServerHttpRequest;
 5import org.springframework.web.socket.WebSocketHandler;
 6import org.springframework.web.socket.server.HandshakeInterceptor;
 7import javax.servlet.http.HttpSession;
 8import java.util.Map;
 9
10public class HttpHandshakeInterceptor implements HandshakeInterceptor {
11
12    /**
13     * 握手前拦截，从 HTTP 中参数传入 WebSocket Attributes 方便后续取出相关参数
14     *
15     * @param request    请求对象
16     * @param response   响应对象
17     * @param wsHandler  WebSocket 处理器
18     * @param attributes 从 HTTP 握手到与 WebSocket 会话关联的属性
19     * @return 如果返回 true 则继续握手，返回 false 则终止握手
20     */
21    @Override
22    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response,
23                                   WebSocketHandler wsHandler, Map<String, Object> attributes) {
24        // 将 request 对象转换为 ServletServerHttpRequest 对象
25        ServletServerHttpRequest serverRequest = (ServletServerHttpRequest) request;
26        // 获取 HTTP Session 对象
27        HttpSession session = serverRequest.getServletRequest().getSession();
28        if (session != null) {
29            // 从 HTTP Session 中获取用户信息
30            User user = (User) session.getAttribute("user");
31            if (user != null) {
32                // 将从 HTTP Session 中获取的用户信息存入 WebSocket 的 Attributes 对象中
33                attributes.put("user", user);
34                // 继续握手
35                return true;
36            }
37        }
38        // 终止握手
39        return false;
40    }
41
42    /**
43     * 握手完成后调用
44     *
45     * @param request   请求对象
46     * @param response  响应对象
47     * @param wsHandler WebSocket 处理器
48     * @param ex        异常信息
49     */
50    @Override
51    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response,
52                               WebSocketHandler wsHandler, Exception ex) {
53    }
54
55}

4、创建 WebSocket 握手处理器

创建用于与正在建立会话过程中的 WebSocket 的用户相关联的方法，可以在此处配置用于关联用户相关的信息。

 1import mydlq.club.model.User;
 2import org.springframework.http.server.ServerHttpRequest;
 3import org.springframework.web.socket.WebSocketHandler;
 4import org.springframework.web.socket.server.support.DefaultHandshakeHandler;
 5import java.security.Principal;
 6import java.util.Map;
 7
 8public class HttpHandshakeHandler extends DefaultHandshakeHandler {
 9
10    /**
11     * 用于与正在建立会话过程中的 WebSocket 的用户相关联的方法，可以在此处配置进行关联的用户信息。
12     *
13     * @param request    握手请求对象
14     * @param wsHandler  WebSocket 处理器
15     * @param attributes 从 HTTP 握手到与 WebSocket 会话关联的属性
16     * @return 对于 WebSocket 的会话的用户或者 null
17     */
18    @Override
19    protected Principal determineUser(ServerHttpRequest request, WebSocketHandler wsHandler, Map<String, Object> attributes) {
20        User user = (User) attributes.get("user");
21        return user::getUsername;
22    }
23
24}

5、创建 WebSocket 配置类

创建 WebSocket 配置类，配置进行连接注册的端点 /mydlq 和消息代理前缀 /queue 及接收客户端发送消息的前缀 /app，再加一个 configureWebSocketTransport 配置，用于设置 WebSocket 消息交互配置，这里可以将上面创建的。

 1import org.springframework.context.annotation.Configuration;
 2import org.springframework.messaging.simp.config.MessageBrokerRegistry;
 3import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
 4import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
 5import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
 6
 7/**
 8 * @author mydlq
 9 */
10@Configuration
11@EnableWebSocketMessageBroker
12public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
13
14    /**
15     * 配置消息代理选项
16     *
17     * @param registry 消息代理注册配置
18     */
19    @Override
20    public void configureMessageBroker(MessageBrokerRegistry registry) {
21        // 设置一个或者多个代理前缀，在 Controller 类中的方法里面发生的消息，会首先转发到代理从而发送到对应广播或者队列中。
22        registry.enableSimpleBroker("/queue");
23        // 配置客户端发送请求消息的一个或多个前缀，该前缀会筛选消息目标转发到 Controller 类中注解对应的方法里
24        registry.setApplicationDestinationPrefixes("/app");
25        // 服务端通知客户端的前缀，可以不设置，默认为user
26        registry.setUserDestinationPrefix("/user");
27    }
28
29    /**
30     * 配置 WebSocket 进入点，及开启使用 SockJS，这些配置主要用配置连接端点，用于 WebSocket 连接
31     *
32     * @param registry STOMP 端点
33     */
34    @Override
35    public void registerStompEndpoints(StompEndpointRegistry registry) {
36        // 配置 websocket 进入点
37        registry.addEndpoint("/mydlq")
38                .addInterceptors(new HttpHandshakeInterceptor())
39                .setHandshakeHandler(new HttpHandshakeHandler())
40                .withSockJS();
41    }
42
43}

6、创建测试 Controller 类

跟上面介绍广播模式一样，作用也是根据 WebSocket 配置类中 /app 前缀匹配后进入 Controller 类进行逻辑处理操作。

 1import mydlq.club.model.MessageBody;
 2import org.springframework.beans.factory.annotation.Autowired;
 3import org.springframework.messaging.handler.annotation.MessageMapping;
 4import org.springframework.messaging.simp.SimpMessageSendingOperations;
 5import org.springframework.stereotype.Controller;
 6import java.security.Principal;
 7
 8/**
 9 * @author mydlq
10 */
11@Controller
12public class MessageController {
13
14    @Autowired
15    private SimpMessageSendingOperations simpMessageSendingOperations;
16
17    /**
18     * 点对点发送消息，将消息发送到指定用户
19     */
20    @MessageMapping("/test")
21    public void sendUserMessage(Principal principal, MessageBody messageBody) {
22        messageBody.setFrom(principal.getName());
23        simpMessageSendingOperations.convertAndSendToUser(messageBody.getTargetUser(), messageBody.getDestination(), messageBody);
24    }
25
26}

7、创建登录 Controller 类

创建一个用于登录的 Controller 类，这里提前创建了两个模拟用户 mydlq1 与 mydlq2 与登录接口 /login，这里设置登录成功后就将用户信息存储到 Session 中，类内容如下：

  • 用户：mydlq1，密码：123456
  • 用户：mydlq2，密码：123456
 1import mydlq.club.model.User;
 2import org.springframework.web.bind.annotation.PostMapping;
 3import org.springframework.web.bind.annotation.RequestParam;
 4import org.springframework.web.bind.annotation.RestController;
 5import javax.annotation.PostConstruct;
 6import javax.servlet.http.HttpServletRequest;
 7import javax.servlet.http.HttpServletResponse;
 8import java.io.IOException;
 9import java.util.HashMap;
10import java.util.Map;
11
12@RestController
13public class LoginController {
14
15    /**
16     * 设置存储的 SESSION 名称
17     */
18    public static final String USER_SESSION_NAME = "user";
19    /**
20     * 设置模拟用户信息
21     */
22    private static final Map<String, User> USER_MAP = new HashMap<>();
23
24    /**
25     * 初始化
26     */
27    @PostConstruct
28    public void init() {
29        User user1 = new User();
30        user1.setUsername("mydlq1");
31        user1.setPassword("123456");
32        User user2 = new User();
33        user2.setUsername("mydlq2");
34        user2.setPassword("123456");
35        USER_MAP.put(user1.getUsername(), user1);
36        USER_MAP.put(user2.getUsername(), user2);
37    }
38
39    @PostMapping("/login")
40    public String login(@RequestParam String username, @RequestParam String password,
41                              HttpServletRequest request, HttpServletResponse response) throws IOException {
42        // 检测是否已经登录
43        if (request.getSession().getAttribute(USER_SESSION_NAME) != null) {
44            return "已经登陆，无需重复登陆";
45        }
46        // 验证用户名密码，如果验证失败则返回错误信息
47        if (validate(username, password)) {
48            // 验证成功则将登录用户信息存储
49            request.getSession().setAttribute(USER_SESSION_NAME, USER_MAP.get(username));
50            // 跳转到首页
51            response.sendRedirect("/index.html");
52        }
53        return "用户名或密码不正确";
54    }
55
56    /**
57     * 验证用户名、密码是否正确
58     *
59     * @param username 用户名
60     * @param password 密码
61     */
62    private boolean validate(String username, String password) {
63        // 验证用户名密码，如果验证失败则返回错误信息
64        for (User user : USER_MAP.values()) {
65            if (user.getUsername().equals(username) && user.getPassword().equals(password)) {
66                return true;
67            }
68        }
69        return false;
70    }
71
72}

8、SpringBoot 启动类

 1import org.springframework.boot.SpringApplication;
 2import org.springframework.boot.autoconfigure.SpringBootApplication;
 3
 4@SpringBootApplication
 5public class Application {
 6
 7    public static void main(String[] args) {
 8        SpringApplication.run(Application.class, args);
 9    }
10
11}

9、创建测试的前端 html 和 js

(1)、创建 WebSocket JS

创建用于操作 WebSocket 的 JS 文件 app-websocket.js，内容如下：

 1// 设置 STOMP 客户端
 2var stompClient = null;
 3
 4// 设置 WebSocket 进入端点
 5var SOCKET_ENDPOINT = "/mydlq";
 6// 设置订阅消息的请求地址前缀
 7var SUBSCRIBE_PREFIX  = "/queue";
 8// 设置订阅地址
 9var SUBSCRIBE = "";
10// 设置服务器端点，访问服务器中哪个接口
11var SEND_ENDPOINT = "/app/test";
12
13/* 进行连接 */
14function connect() {
15    // 设置 SOCKET
16    var socket = new SockJS(SOCKET_ENDPOINT);
17    // 配置 STOMP 客户端
18    stompClient = Stomp.over(socket);
19    // STOMP 客户端连接
20    stompClient.connect({}, function (frame) {
21        alert("连接成功");
22    });
23}
24
25/* 订阅信息 */
26function subscribeSocket(){
27    // 设置订阅地址
28    SUBSCRIBE = SUBSCRIBE_PREFIX + $("#subscribe").val();
29    // 输出订阅地址
30    alert("设置订阅地址为：" + SUBSCRIBE);
31    // 执行订阅消息
32    stompClient.subscribe("/user" + SUBSCRIBE, function (responseBody) {
33        var receiveMessage = JSON.parse(responseBody.body);
34        console.log(receiveMessage);
35        $("#information").append("<tr><td>" + receiveMessage.content + "</td></tr>");
36    });
37}
38
39/* 断开连接 */
40function disconnect() {
41    stompClient.disconnect(function() {
42        alert("断开连接");
43    });
44}
45
46/* 发送消息并指定目标地址 */
47function sendMessageNoParameter() {
48    // 设置发送的内容
49    var sendContent = $("#content").val();
50    // 设置发送的用户
51    var sendUser = $("#targetUser").val();
52    // 设置待发送的消息内容
53    var message = '{"targetUser":"' + sendUser + '", "destination": "' + SUBSCRIBE + '", "content": "' + sendContent + '"}';
54    // 发送消息
55    stompClient.send(SEND_ENDPOINT, {}, message);
56}

(2)、创建 WebSocket HTML

index.html

创建用于展示 WebSocket 相关功能的 WEB HTML 页面 index.html，内容如下：

 1<!DOCTYPE html>
 2<html>
 3<head>
 4    <title>Hello WebSocket</title>
 5    <link href="https://cdn.bootcdn.net/ajax/libs/twitter-bootstrap/3.4.1/css/bootstrap.min.css" rel="stylesheet">
 6    <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
 7    <script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.5.1/jquery.js"></script>
 8    <script src="https://cdn.bootcdn.net/ajax/libs/sockjs-client/1.4.0/sockjs.min.js"></script>
 9    <script src="https://cdn.bootcdn.net/ajax/libs/stomp.js/2.3.3/stomp.min.js"></script>
10    <script src="app-websocket.js"></script>
11</head>
12<body>
13    <div id="main-content" class="container" style="margin-top: 10px;">
14        <div class="row">
15            <form class="navbar-form" style="margin-left:0px">
16                <div class="col-md-12">
17                    <div class="form-group">
18                        <label>WebSocket 连接：</label>
19                        <button class="btn btn-primary" type="button" onclick="connect();">进行连接</button>
20                        <button class="btn btn-danger" type="button" onclick="disconnect();">断开连接</button>
21                    </div>
22                    <label>订阅地址：</label>
23                    <div class="form-group">
24                        <input type="text" id="subscribe" class="form-control" placeholder="订阅地址">
25                    </div>
26                    <button class="btn btn-warning" onclick="subscribeSocket();" type="button">订阅</button>
27                </div>
28            </form>
29        </div>
30        </br>
31        <div class="row">
32            <div class="form-group">
33                <label>发送的目标用户：</label>
34                <input type="text" id="targetUser" class="form-control" placeholder="发送的用户">
35                <label for="content">发送的消息内容：</label>
36                <input type="text" id="content" class="form-control" placeholder="消息的内容">
37            </div>
38            <button class="btn btn-info" onclick="sendMessageNoParameter();" type="button">发送</button>
39        </div>
40        </br>
41        <div class="row">
42            <div class="col-md-12">
43                <h5 class="page-header" style="font-weight:bold">接收到的消息：</h5>
44                <table class="table table-striped">
45                    <tbody id="information"></tbody>
46                </table>
47            </div>
48        </div>
49    </div>
50</body>
51</html>

login.html

创建用登录能的 WEB HTML 页面 login.html，内容如下：

 1<!DOCTYPE html>
 2<html>
 3<head>
 4    <title>Hello WebSocket</title>
 5    <link href="https://cdn.bootcdn.net/ajax/libs/twitter-bootstrap/3.4.1/css/bootstrap.min.css" rel="stylesheet">
 6    <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
 7    <script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.5.1/jquery.js"></script>
 8    <script src="https://cdn.bootcdn.net/ajax/libs/sockjs-client/1.4.0/sockjs.min.js"></script>
 9    <script src="https://cdn.bootcdn.net/ajax/libs/stomp.js/2.3.3/stomp.min.js"></script>
10</head>
11<body>
12    <div class="Container">
13        <div class="row">
14            <div class="col-md-4 col-md-offset-4">
15                <h1 class="page-header">用户登录</h1>
16                <form role="form" method="post" action="/login">
17                    <div class="form-group">
18                        <label for="username">用户名：</label>
19                        <input type="text" class="form-control" placeholder="请输入账号" id="username" name="username">
20                    </div>
21                    <div class="form-group">
22                        <label for="password">密 码：</label>
23                        <input type="password" class="form-control" placeholder="请输入密码" id="password" name="password">
24                    </div>
25                    <div class="checkbox" style="text-align:center;">
26                        <br/>
27                        <input type="submit" value="登录" class="btn"/>
28                    </div>
29                </form>
30            </div>
31        </div>
32    </div>
33</body>
34</html>

10、启动并进行测试

为了方便测试，需要打开两个不同类型浏览器（因为用户登录后会存 Session，如果一个浏览器不同用户登录会使之前 Session 失效）来进行测试，两个浏览器同时输入地址 http://localhost:8080/index.html 访问测试的前端页面，然后可以看到并没有进入 /index.html 页面，而是跳转到 Spring Security 提供的登录的 /login 页面，如下：

  • 浏览器1登录的用户：mydlq1，密码：123456
  • 浏览器2登录的用户：mydlq2，密码：123456

两个浏览器中都输入用户名/密码 mydlq1/123456 与 mydlq2/123456 进行登录，然后会回到 /index.html 页面，然后执行下面步骤进行测试：

  • (1)、"浏览器1"和"浏览器2"点击"进行连接"按钮，连接 WebSocket 服务端；
  • (2)、"浏览器1"和"浏览器2"中同时设置订阅地址为"/abc"，然后点击订阅按钮进行消息订阅；
  • (3)、"浏览器1"(用户 mydlq1)设置发送目标用户为"/mydlq2"，"浏览器2"(用户 mydlq2)设置发送目标用户为"/mydlq1"；
  • (4)、"浏览器1"(用户 mydlq1)设置发送消息为"Hi, I'm mydlq1"，"浏览器2"(用户 mydlq2)设置发送消息为"Hi, I'm mydlq2"；
  • (5)、点击"发送"按钮发送消息；

执行完上面步骤成后，可以在两个不同浏览器中观察到如下内容：

十三、SpringBoot 结合 WebSocket 的常用方法示例

1、WebSocket 开启跨域选项

WebSocket 配置类，里面设置允许跨域，内容如下：

 1import org.springframework.context.annotation.Configuration;
 2import org.springframework.messaging.simp.config.MessageBrokerRegistry;
 3import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
 4import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
 5import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
 6import org.springframework.web.socket.config.annotation.WebSocketTransportRegistration;
 7
 8@Configuration
 9@EnableWebSocketMessageBroker
10public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
11
12    @Override
13    public void configureMessageBroker(MessageBrokerRegistry registry) {
14        registry.enableSimpleBroker("/queue");
15        registry.setApplicationDestinationPrefixes("/app");
16    }
17
18    @Override
19    public void registerStompEndpoints(StompEndpointRegistry registry) {
20        registry.addEndpoint("/mydlq")
21        // 设置允许跨域，设置为"*"则为允许全部域名
22        .setAllowedOrigins("*")
23        .withSockJS();
24    }
25
26}

2、WebSocket 用户上、下线监听

创建 WebSocket 用户上线、下线处理器，内容如下：

 1import org.springframework.context.annotation.Configuration;
 2import org.springframework.web.socket.CloseStatus;
 3import org.springframework.web.socket.WebSocketHandler;
 4import org.springframework.web.socket.WebSocketSession;
 5import org.springframework.web.socket.handler.WebSocketHandlerDecorator;
 6import org.springframework.web.socket.handler.WebSocketHandlerDecoratorFactory;
 7
 8@Configuration
 9public class HttpWebSocketHandlerDecoratorFactory implements WebSocketHandlerDecoratorFactory {
10
11    /**
12     * 配置 webSocket 处理器
13     *
14     * @param webSocketHandler webSocket 处理器
15     * @return webSocket 处理器
16     */
17    @Override
18    public WebSocketHandler decorate(WebSocketHandler webSocketHandler) {
19        return new WebSocketHandlerDecorator(webSocketHandler) {
20            /**
21             * websocket 连接时执行的动作
22             * @param session    websocket session 对象
23             * @throws Exception 异常对象
24             */
25            @Override
26            public void afterConnectionEstablished(final WebSocketSession session) throws Exception {
27                // 输出进行 websocket 连接的用户信息
28                if (session.getPrincipal() != null) {
29                    String username = session.getPrincipal().getName();
30                    System.out.println("用户:" + username + "上线");
31                    super.afterConnectionEstablished(session);
32                }
33            }
34
35            /**
36             * websocket 关闭连接时执行的动作
37             * @param session websocket session 对象
38             * @param closeStatus 关闭状态对象
39             * @throws Exception 异常对象
40             */
41            @Override
42            public void afterConnectionClosed(final WebSocketSession session, CloseStatus closeStatus) throws Exception {
43                // 输出关闭 websocket 连接的用户信息
44                if (session.getPrincipal() != null) {
45                    String username = session.getPrincipal().getName();
46                    System.out.println("用户:" + username + "下线");
47                    super.afterConnectionClosed(session, closeStatus);
48                }
49            }
50        };
51    }
52
53}

WebSocket 配置类中实现 configureWebSocketTransport() 方法，将上面 WebSocket 处理器加到其中，如下：

 1import org.springframework.context.annotation.Configuration;
 2import org.springframework.messaging.simp.config.MessageBrokerRegistry;
 3import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
 4import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
 5import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
 6import org.springframework.web.socket.config.annotation.WebSocketTransportRegistration;
 7
 8@Configuration
 9@EnableWebSocketMessageBroker
10public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
11
12    @Override
13    public void configureMessageBroker(MessageBrokerRegistry registry) {
14        registry.enableSimpleBroker("/queue");
15        registry.setApplicationDestinationPrefixes("/app");
16    }
17
18    @Override
19    public void registerStompEndpoints(StompEndpointRegistry registry) {
20        registry.addEndpoint("/mydlq").withSockJS();
21    }
22
23    /**
24     * 添加 WebSocket 用户上、下线监听器
25     */
26    @Override
27    public void configureWebSocketTransport(WebSocketTransportRegistration registry) {
28        registry.addDecoratorFactory(new HttpWebSocketHandlerDecoratorFactory());
29    }
30
31}

如果对你有帮助，别忘了 Github 点颗星嘞~

---END---

  !版权声明：本博客内容均为原创,每篇博文作为知识积累,写博不易,转载请注明出处。