From 447f1d7162a486e16d7bc56760ac3019431b350e Mon Sep 17 00:00:00 2001
From: yulinling <2712495353@qq.com>
Date: Sun, 22 Jun 2025 21:08:43 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20-=20=E6=B7=BB=E5=8A=A0WebSocket?=
 =?UTF-8?q?=E6=8E=A8=E9=80=81=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pom.xml                                       |  6 ++++
 .../workflow/config/WebSocketConfig.java      | 33 +++++++++++++++++++
 .../workflow/constant/WebSocketConst.java     | 14 ++++++++
 .../workflow/task/WebSocketTask.java          | 31 +++++++++++++++++
 4 files changed, 84 insertions(+)
 create mode 100644 src/main/java/asia/yulinling/workflow/config/WebSocketConfig.java
 create mode 100644 src/main/java/asia/yulinling/workflow/constant/WebSocketConst.java
 create mode 100644 src/main/java/asia/yulinling/workflow/task/WebSocketTask.java

diff --git a/pom.xml b/pom.xml
index fd862f0..f41f6cf 100644
--- a/pom.xml
+++ b/pom.xml
@@ -44,6 +44,12 @@
             <artifactId>spring-boot-starter-web</artifactId>
         </dependency>
 
+        <!-- Spring WebSocket-->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-websocket</artifactId>
+        </dependency>
+
         <!-- Spring Security -->
         <dependency>
             <groupId>org.springframework.boot</groupId>
diff --git a/src/main/java/asia/yulinling/workflow/config/WebSocketConfig.java b/src/main/java/asia/yulinling/workflow/config/WebSocketConfig.java
new file mode 100644
index 0000000..8d1db4a
--- /dev/null
+++ b/src/main/java/asia/yulinling/workflow/config/WebSocketConfig.java
@@ -0,0 +1,33 @@
+package asia.yulinling.workflow.config;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.messaging.simp.config.MessageBrokerRegistry;
+import org.springframework.web.socket.config.annotation.EnableWebSocket;
+import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
+import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
+import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
+
+/**
+ * <p>
+ * WebSocket配置
+ * </p>
+ *
+ * @author YLL
+ * @since 2025/6/22
+ */
+@Configuration
+@EnableWebSocket
+@EnableWebSocketMessageBroker
+public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
+    @Override
+    public void registerStompEndpoints(StompEndpointRegistry registry) {
+        // 注册一个 /notification 端点，前端通过这个端点进行连接
+        registry.addEndpoint("/websocket").setAllowedOrigins("*").withSockJS();
+    }
+
+    @Override
+    public void configureMessageBroker(MessageBrokerRegistry registry) {
+        // 定义了一个客户端订阅地址的前缀信息，也就是客户端接收服务端发送消息的前缀信息
+        registry.enableSimpleBroker("/topic");
+    }
+}
diff --git a/src/main/java/asia/yulinling/workflow/constant/WebSocketConst.java b/src/main/java/asia/yulinling/workflow/constant/WebSocketConst.java
new file mode 100644
index 0000000..622de20
--- /dev/null
+++ b/src/main/java/asia/yulinling/workflow/constant/WebSocketConst.java
@@ -0,0 +1,14 @@
+package asia.yulinling.workflow.constant;
+
+/**
+ * <p>
+ * WebSocketConst
+ * </p>
+ *
+ * @author YLL
+ * @since 2025/6/22
+ */
+public interface WebSocketConst {
+    String WEBSOCKET_PATH = "/websocket";
+    String PUSH_SERVER = "/topic/push";
+}
diff --git a/src/main/java/asia/yulinling/workflow/task/WebSocketTask.java b/src/main/java/asia/yulinling/workflow/task/WebSocketTask.java
new file mode 100644
index 0000000..860236e
--- /dev/null
+++ b/src/main/java/asia/yulinling/workflow/task/WebSocketTask.java
@@ -0,0 +1,31 @@
+package asia.yulinling.workflow.task;
+
+import cn.hutool.core.date.DateUtil;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.messaging.simp.SimpMessagingTemplate;
+import org.springframework.stereotype.Component;
+
+import java.util.Date;
+
+/**
+ * <p>
+ * WebSocketTask
+ * </p>
+ *
+ * @author YLL
+ * @since 2025/6/22
+ */
+@Component
+@Slf4j
+@RequiredArgsConstructor
+public class WebSocketTask {
+    private final SimpMessagingTemplate simpMessagingTemplate;
+
+//    @Scheduled(fixedRate = 1000)
+    public void sendMessage() throws Exception{
+        log.info("【推送消息】开始执行：{}", DateUtil.formatDateTime(new Date()));
+        simpMessagingTemplate.convertAndSend("test", "1");
+        log.info("【推送消息】执行结束：{}", DateUtil.formatDateTime(new Date()));
+    }
+}