fix 修复 sse 关闭接口无法断连问题

1 year ago · b95dc22403
parent 54471dfab0
commit b95dc22403
1 changed files with 3 additions and 1 deletions
--- 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
@ -70,7 +70,9 @@ public class SseEmitterManager {
        Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.get(userId);
        if (emitters != null) {
            try {
-                emitters.get(token).send(SseEmitter.event().comment("disconnected"));
+                SseEmitter sseEmitter = emitters.get(token);
+                sseEmitter.send(SseEmitter.event().comment("disconnected"));
+                sseEmitter.complete();
            } catch (Exception ignore) {
            }
            emitters.remove(token);