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

update 优化 sse 如果获取token列表为空 删除userid对应的存储

Browse Source
This commit is contained in:
疯狂的狮子Li 2024-12-27 11:17:09 +08:00
parent c2746c2392
commit a7b83672ba
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
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.hutool.core.map.MapUtil;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.dromara.common.redis.utils.RedisUtils; import org.dromara.common.redis.utils.RedisUtils;
import org.dromara.common.sse.dto.SseMessageDto; import org.dromara.common.sse.dto.SseMessageDto;
@ -65,7 +66,7 @@ public class SseEmitterManager {
*/ */
public void disconnect(Long userId, String token) { public void disconnect(Long userId, String token) {
Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.get(userId); Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.get(userId);
if (emitters != null) { if (MapUtil.isNotEmpty(emitters)) {
try { try {
SseEmitter sseEmitter = emitters.get(token); SseEmitter sseEmitter = emitters.get(token);
sseEmitter.send(SseEmitter.event().comment("disconnected")); sseEmitter.send(SseEmitter.event().comment("disconnected"));
@ -73,6 +74,8 @@ public class SseEmitterManager {
} catch (Exception ignore) { } catch (Exception ignore) {
} }
emitters.remove(token); emitters.remove(token);
} else {
USER_TOKEN_EMITTERS.remove(userId);
} }
} }
@ -93,7 +96,7 @@ public class SseEmitterManager {
*/ */
public void sendMessage(Long userId, String message) { public void sendMessage(Long userId, String message) {
Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.get(userId); Map<String, SseEmitter> emitters = USER_TOKEN_EMITTERS.get(userId);
if (emitters != null) { if (MapUtil.isNotEmpty(emitters)) {
for (Map.Entry<String, SseEmitter> entry : emitters.entrySet()) { for (Map.Entry<String, SseEmitter> entry : emitters.entrySet()) {
try { try {
entry.getValue().send(SseEmitter.event() entry.getValue().send(SseEmitter.event()
@ -103,6 +106,8 @@ public class SseEmitterManager {
emitters.remove(entry.getKey()); emitters.remove(entry.getKey());
} }
} }
} else {
USER_TOKEN_EMITTERS.remove(userId);
} }
} }