zhuxinru/cid
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix:SSE添加日志&获取用户token改成username接收

Browse Source
This commit is contained in:
zhuxinru
2026-03-30 15:50:15 +08:00
parent e146368ff9
commit dd254ce1f4
3 changed files with 38 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
honeycom-msg/honeycom-msg-biz/src/main/java/com/honeycombis/honeycom/msg/service/impl/MessageServiceImpl.java
View File

@@ -78,6 +78,7 @@ public class MessageServiceImpl extends ServiceImpl<MessageMapper, Message> impl
R<Object> sendSSEMessage(Message message) {
log.info("发送消息通知！");
Long userId = message.getUserId();
// 这里去获取真实的消息参数信息
String realMessage = messageTaskNotifyService.getMessageJson(message);

43
honeycom-msg/honeycom-msg-biz/src/main/java/com/honeycombis/honeycom/msg/service/impl/SSEServiceImpl.java
View File

@@ -8,6 +8,8 @@ import org.springframework.stereotype.Service;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import java.io.IOException;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
@@ -72,15 +74,42 @@ public class SSEServiceImpl implements SSEService {
@Override
public void sendMessage(Long userId, String eventName, String message) {
// 这里根据用户ID获取所有的SSE链接信息
Map<String, SseEmitter> emitters = userEmitters.entrySet().stream().filter(res -> res.getKey().startsWith(userId + ":")).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
for (String key : emitters.keySet()) {
log.info("[SSEServiceImpl] sendMessage: userId={}, event={}, message={}", userId, eventName, message);
// 安全获取该用户所有 SSE 连接
List<Map.Entry<String, SseEmitter>> userEmittersList = userEmitters.entrySet().stream()
.filter(res -> res.getKey().startsWith(userId + ":"))
.collect(Collectors.toList());
// 用迭代器遍历，避免并发修改异常
Iterator<Map.Entry<String, SseEmitter>> iterator = userEmittersList.iterator();
while (iterator.hasNext()) {
Map.Entry<String, SseEmitter> entry = iterator.next();
String key = entry.getKey();
SseEmitter emitter = entry.getValue();
try {
emitters.get(key).send(SseEmitter.event().name(eventName).data(message));
} catch (Exception e) {
log.info("###Error sending " + eventName + " message to user: " + userId);
// 关键：先判断 emitter 是否可用
if (emitter == null) {
userEmitters.remove(key);
continue;
}
// 发送 SSE 事件
emitter.send(SseEmitter.event().name(eventName).data(message));
log.info("[SSE] 发送成功: key={}", key);
} catch (IOException e) {
// 客户端断开连接（最常见）
log.warn("[SSE] 客户端断开，自动移除: key={}, error={}", key, e.getMessage());
userEmitters.remove(key);
} catch (IllegalStateException e) {
// emitter 已完成/超时
log.warn("[SSE] Emitter 已关闭，自动移除: key={}", key);
userEmitters.remove(key);
} catch (Exception e) {
log.error("[SSE] 发送消息异常: key=" + key, e); // 必须打印堆栈
userEmitters.remove(key);
log.info("###Removed userId=" + userId + " from userEmitters");
}
}
}

2
honeycom-spdm/src/main/java/com/honeycombis/honeycom/spdm/controller/SpdmUserController.java
View File

@@ -290,7 +290,7 @@ public class SpdmUserController {
Long tenantId = 1979091834410176514L;
// 利元亨传工号
R<List<SysUserVO>> sysUserVOR = remoteUserServiceFeign.getUserListByUserName(String.valueOf(userParamDto.getUserId()));
R<List<SysUserVO>> sysUserVOR = remoteUserServiceFeign.getUserListByUserName(userParamDto.getUserName());
if (CollectionUtils.isNotEmpty(sysUserVOR.getData())) {
userParamDto.setUserId(sysUserVOR.getData().get(0).getUserId());
}