From 239d59c864af365d2b61dc99b2cfa21bf813896b Mon Sep 17 00:00:00 2001
From: 疯狂的狮子Li <15040126243@163.com>
Date: 星期一, 29 七月 2024 12:44:45 +0800
Subject: [PATCH] update 优化 sse发送消息 增加token有效期判断
---
ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/core/SseEmitterManager.java | 8 +++++++-
1 files changed, 7 insertions(+), 1 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 4b56b69..91dd71d 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
@@ -1,5 +1,6 @@
package org.dromara.common.sse.core;
+import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.collection.CollUtil;
import lombok.extern.slf4j.Slf4j;
import org.dromara.common.redis.utils.RedisUtils;
@@ -30,6 +31,7 @@
emitter.onCompletion(() -> emitters.remove(token));
emitter.onTimeout(() -> emitters.remove(token));
+ emitter.onError((e) -> emitters.remove(token));
try {
emitter.send(SseEmitter.event().comment("connected"));
@@ -70,9 +72,13 @@
if (emitters != null) {
for (Map.Entry<String, SseEmitter> entry : emitters.entrySet()) {
try {
+ // token 鏃犳晥鎴栧凡杩囨湡
+ if (StpUtil.stpLogic.getTokenActiveTimeoutByToken(entry.getKey()) < -1) {
+ emitters.remove(entry.getKey());
+ continue;
+ }
entry.getValue().send(SseEmitter.event()
.name("message")
- .reconnectTime(10000L)
.data(message));
} catch (Exception e) {
emitters.remove(entry.getKey());
--
Gitblit v1.9.3