From ad85fa201649695919b037b028e53d19932c4a1c Mon Sep 17 00:00:00 2001
From: 疯狂的狮子Li <15040126243@163.com>
Date: 星期四, 19 十二月 2024 10:54:08 +0800
Subject: [PATCH] update 优化 sse工具类
---
ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/utils/SseMessageUtils.java | 22 +++++++++++++++++++++-
1 files changed, 21 insertions(+), 1 deletions(-)
diff --git a/ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/utils/SseMessageUtils.java b/ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/utils/SseMessageUtils.java
index 4334e98..45f354c 100644
--- a/ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/utils/SseMessageUtils.java
+++ b/ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/utils/SseMessageUtils.java
@@ -8,7 +8,7 @@
import org.dromara.common.sse.dto.SseMessageDto;
/**
- * 宸ュ叿绫�
+ * SSE宸ュ叿绫�
*
* @author Lion Li
*/
@@ -16,6 +16,7 @@
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class SseMessageUtils {
+ private final static Boolean SSE_ENABLE = SpringUtils.getProperty("sse.enabled", Boolean.class, true);
private final static SseEmitterManager MANAGER = SpringUtils.getBean(SseEmitterManager.class);
/**
@@ -25,6 +26,9 @@
* @param message 瑕佸彂閫佺殑娑堟伅鍐呭
*/
public static void sendMessage(Long userId, String message) {
+ if (!isEnable()) {
+ return;
+ }
MANAGER.sendMessage(userId, message);
}
@@ -34,6 +38,9 @@
* @param message 瑕佸彂閫佺殑娑堟伅鍐呭
*/
public static void sendMessage(String message) {
+ if (!isEnable()) {
+ return;
+ }
MANAGER.sendMessage(message);
}
@@ -43,6 +50,9 @@
* @param sseMessageDto 瑕佸彂甯冪殑SSE娑堟伅瀵硅薄
*/
public static void publishMessage(SseMessageDto sseMessageDto) {
+ if (!isEnable()) {
+ return;
+ }
MANAGER.publishMessage(sseMessageDto);
}
@@ -52,7 +62,17 @@
* @param message 瑕佸彂甯冪殑娑堟伅鍐呭
*/
public static void publishAll(String message) {
+ if (!isEnable()) {
+ return;
+ }
MANAGER.publishAll(message);
}
+ /**
+ * 鏄惁寮�鍚�
+ */
+ public static Boolean isEnable() {
+ return SSE_ENABLE;
+ }
+
}
--
Gitblit v1.9.3