feat: 新增 websocket 功能模块

src/main/java/com/youlai/system/config/WebSocketConfig.java

@@ -0,0 +1,32 @@
+package com.youlai.system.config;
+
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.messaging.simp.config.MessageBrokerRegistry;
+import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
+import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
+import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
+
+/**
+ * WebSocket 配置类
+ *
+ * @author haoxr
+ * @since 2.3.0
+ */
+@Configuration
+@ConditionalOnProperty(name = "system.config.websocket-enabled")// system.config.websocket-enabled = true 才会自动装配
+@EnableWebSocketMessageBroker // 注解告诉Spring框架要开启WebSocket消息代理的功能，并配置相关的端点和消息代理
+public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
+
+
+    @Override
+    public void configureMessageBroker(MessageBrokerRegistry registry) {
+        registry.enableSimpleBroker("/topic", "/user");
+    }
+
+    @Override
+    public void registerStompEndpoints(StompEndpointRegistry registry) {
+        // 注册了一个 /ws 的端点，用于建立WebSocket连接。
+        registry.addEndpoint("/ws").setAllowedOriginPatterns("*");
+    }
+}

src/main/java/com/youlai/system/controller/WebsocketController.java

@@ -0,0 +1,61 @@
+package com.youlai.system.controller;
+
+import cn.hutool.json.JSONUtil;
+import com.youlai.system.common.result.Result;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.messaging.simp.SimpMessagingTemplate;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * WebSocket 测试控制器
+ *
+ * @author haoxr
+ * @since 2.3.0
+ */
+@RestController
+@RequestMapping("/websocket")
+@RequiredArgsConstructor
+@Slf4j
+public class WebsocketController {
+
+    private final SimpMessagingTemplate messagingTemplate;
+
+    /**
+     * 广播发送消息
+     *
+     * @param message 消息内容
+     */
+    @PostMapping("/sendToAll")
+    // @SendTo("/topic/all")  // 使用SimpMessagingTemplate发送消息给所有订阅了"/topic/all"目标的客户端
+    public Result sendToAll(String message) {
+        log.info("【广播消息请求接收】消息：{}", message);
+        // 处理接收到的消息逻辑
+        // ...
+
+        // 构造要发送的消息内容
+        String content = "服务端广播消息： " + message;
+
+        // 使用SimpMessagingTemplate发送消息给所有订阅了"/topic/all"目标的客户端
+        messagingTemplate.convertAndSend("/topic/all", content);
+
+        return Result.success("广播消息发送成功");
+    }
+
+    /**
+     * 点对点发送消息
+     *
+     * @param userId 用户ID
+     */
+    @PostMapping("/sendToUser/{userId}")
+    public Result sendToUser(@PathVariable Long userId, String message) {
+        log.info("【点对点请求接收】用户：{};消息：{}", userId, JSONUtil.toJsonStr(message));
+        // 发送主题目的(destination)= /user/{userId}/message
+        messagingTemplate.convertAndSendToUser(String.valueOf(userId), "/message", message);
+        return Result.success("点对点消息发送成功");
+    }
+
+}