From 1a403361c951b91f6e3a33da1d8c0ee77e07bb86 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子Li <15040126243@163.com>
Date: 星期四, 19 十二月 2024 09:19:12 +0800
Subject: [PATCH] update 优化 关闭sse后 使用工具报错

---
 ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/utils/SseMessageUtils.java |   13 +++++++++++++
 1 files changed, 13 insertions(+), 0 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 c6abdc8..ff79a15 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
@@ -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 (SSE_ENABLE) {
+            return;
+        }
         MANAGER.sendMessage(userId, message);
     }
 
@@ -34,6 +38,9 @@
      * @param message 瑕佸彂閫佺殑娑堟伅鍐呭
      */
     public static void sendMessage(String message) {
+        if (SSE_ENABLE) {
+            return;
+        }
         MANAGER.sendMessage(message);
     }
 
@@ -43,6 +50,9 @@
      * @param sseMessageDto 瑕佸彂甯冪殑SSE娑堟伅瀵硅薄
      */
     public static void publishMessage(SseMessageDto sseMessageDto) {
+        if (SSE_ENABLE) {
+            return;
+        }
         MANAGER.publishMessage(sseMessageDto);
     }
 
@@ -52,6 +62,9 @@
      * @param message 瑕佸彂甯冪殑娑堟伅鍐呭
      */
     public static void publishAll(String message) {
+        if (SSE_ENABLE) {
+            return;
+        }
         MANAGER.publishAll(message);
     }
 

--
Gitblit v1.9.3