From 4ac9edd4c0030d2daafd712cd14ef9080b2f1e43 Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Tue, 10 Feb 2026 11:52:40 +0800
Subject: [PATCH] 优化
---
ard-work/src/main/java/com/ard/work/component/websocket/utils/WebSocketUtil.java | 14 ++++++++++++++
1 files changed, 14 insertions(+), 0 deletions(-)
diff --git a/ard-work/src/main/java/com/ard/work/component/websocket/utils/WebSocketUtil.java b/ard-work/src/main/java/com/ard/work/component/websocket/utils/WebSocketUtil.java
index 0bf9553..dbf54b1 100644
--- a/ard-work/src/main/java/com/ard/work/component/websocket/utils/WebSocketUtil.java
+++ b/ard-work/src/main/java/com/ard/work/component/websocket/utils/WebSocketUtil.java
@@ -3,6 +3,7 @@
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.ard.work.component.websocket.WebSocketSessionManager;
+import com.ard.work.domain.message.BaseMessage;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
@@ -125,4 +126,17 @@
WebSocketSessionManager.broadcast(JSON.toJSONString(message));
}
+ // 发送消息到所有 WebSocket 客户端
+ public static <T> void sendToAll(BaseMessage<T> baseMessage) {
+ WebSocketUtil.sendMessageAll(baseMessage); // 你已有的 WebSocket 发送逻辑
+ }
+
+ // 根据消息类型和内容创建消息并发送
+ public static <T> void sendMessage(String type, T payload) {
+ BaseMessage<T> baseMessage = BaseMessage.<T>builder()
+ .type(type)
+ .payload(payload)
+ .build();
+ sendToAll(baseMessage);
+ }
}
\ No newline at end of file
--
Gitblit v1.9.3