From 40e13ce1360a54ab51bb5c516c0a9c54a44fffd5 Mon Sep 17 00:00:00 2001
From: lilinxiang <970259858@qq.com>
Date: Thu, 16 Oct 2025 16:11:49 +0800
Subject: [PATCH] =?UTF-8?q?update=20=E4=BC=98=E5=8C=96=20sse=20=E5=BF=83?=
 =?UTF-8?q?=E8=B7=B3=E6=A3=80=E6=B5=8B=E5=86=99=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../common/sse/core/SseEmitterManager.java    | 29 ++++++++-----------
 1 file changed, 12 insertions(+), 17 deletions(-)

diff --git a/ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/core/SseEmitterManager.java b/ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/core/SseEmitterManager.java
index 289a6ed97..e00c5930f 100644
--- a/ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/core/SseEmitterManager.java
+++ b/ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/core/SseEmitterManager.java
@@ -116,24 +116,19 @@ public class SseEmitterManager {
      * SSE心跳检测，关闭无效连接
      */
     public void sseMonitor() {
-        log.info("开始执行 SSE 心跳检测");
-        for (Map.Entry<Long, Map<String, SseEmitter>> userEmitters : USER_TOKEN_EMITTERS.entrySet()) {
-            Map<String, SseEmitter> emitters = userEmitters.getValue();
-            if (MapUtil.isNotEmpty(emitters)) {
-                for (Map.Entry<String, SseEmitter> entry : emitters.entrySet()) {
-                    try {
-                        // 向客户端发送心跳
-                        entry.getValue().send(SseEmitter.event().comment("heartbeat"));
-                    } catch (Exception e) {
-                        log.warn("心跳发送失败, 关闭连接: userId={}, token={}", userEmitters.getKey(), entry.getKey());
-                        SseEmitter remove = emitters.remove(entry.getKey());
-                        if (remove != null) {
-                            remove.complete();
-                        }
-                    }
+        log.info("开始 SSE 心跳");
+        USER_TOKEN_EMITTERS.forEach((userId, map) ->
+            map.entrySet().removeIf(e -> {
+                try {
+                    e.getValue().send(SseEmitter.event().comment("heartbeat"));
+                    return false;
+                } catch (Exception ex) {
+                    log.warn("心跳失败，移除连接: userId={}, token={}", userId, e.getKey());
+                    e.getValue().complete();
+                    return true;
                 }
-            }
-        }
+            })
+        );
     }
 
     /**