admin/lx-admin-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update 优化 sse发送消息 增加token有效期判断

Browse Source
This commit is contained in:
疯狂的狮子Li 2024-07-29 12:44:45 +08:00
parent 7297053dd6
commit 239d59c864
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ruoyi-common/ruoyi-common-sse/src/main/java/org/dromara/common/sse/core/SseEmitterManager.java
View File

@ -1,5 +1,6 @@
package org.dromara.common.sse.core; package org.dromara.common.sse.core;
import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollUtil;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.dromara.common.redis.utils.RedisUtils; import org.dromara.common.redis.utils.RedisUtils;
@ -71,6 +72,11 @@ public class SseEmitterManager {
if (emitters != null) { if (emitters != null) {
for (Map.Entry<String, SseEmitter> entry : emitters.entrySet()) { for (Map.Entry<String, SseEmitter> entry : emitters.entrySet()) {
try { try {
// token 无效或已过期
if (StpUtil.stpLogic.getTokenActiveTimeoutByToken(entry.getKey()) < -1) {
emitters.remove(entry.getKey());
continue;
}
entry.getValue().send(SseEmitter.event() entry.getValue().send(SseEmitter.event()
.name("message") .name("message")
.data(message)); .data(message));