ly
/
Stwzhj-Cloud-Plus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix 修复 同一个用户不同token连接不同服务导致发送不到问题(改为全局发送)

2.X
疯狂的狮子Li 2024-10-08 14:44:45 +08:00
parent 0db17bac6c
commit 0e7c7e8033
1 changed files with 7 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -124,25 +124,13 @@ public class SseEmitterManager {
* @param sseMessageDto 要发布的SSE消息对象 * @param sseMessageDto 要发布的SSE消息对象
*/ */
public void publishMessage(SseMessageDto sseMessageDto) { public void publishMessage(SseMessageDto sseMessageDto) {
List<Long> unsentUserIds = new ArrayList<>(); SseMessageDto broadcastMessage = new SseMessageDto();
// 当前服务内用户,直接发送消息 broadcastMessage.setMessage(sseMessageDto.getMessage());
for (Long userId : sseMessageDto.getUserIds()) { broadcastMessage.setUserIds(sseMessageDto.getUserIds());
if (USER_TOKEN_EMITTERS.containsKey(userId)) { RedisUtils.publish(SSE_TOPIC, broadcastMessage, consumer -> {
sendMessage(userId, sseMessageDto.getMessage()); log.info("SSE发送主题订阅消息topic:{} session keys:{} message:{}",
continue; SSE_TOPIC, sseMessageDto.getUserIds(), sseMessageDto.getMessage());
} });
unsentUserIds.add(userId);
}
// 不在当前服务内用户,发布订阅消息
if (CollUtil.isNotEmpty(unsentUserIds)) {
SseMessageDto broadcastMessage = new SseMessageDto();
broadcastMessage.setMessage(sseMessageDto.getMessage());
broadcastMessage.setUserIds(unsentUserIds);
RedisUtils.publish(SSE_TOPIC, broadcastMessage, consumer -> {
log.info("SSE发送主题订阅消息topic:{} session keys:{} message:{}",
SSE_TOPIC, unsentUserIds, sseMessageDto.getMessage());
});
}
} }
/** /**