|
|
|
|
@ -44,9 +44,24 @@ public class SseEmitterManager {
|
|
|
|
|
emitters.put(token, emitter);
|
|
|
|
|
|
|
|
|
|
// 当 emitter 完成、超时或发生错误时,从映射表中移除对应的 token
|
|
|
|
|
emitter.onCompletion(() -> emitters.remove(token));
|
|
|
|
|
emitter.onTimeout(() -> emitters.remove(token));
|
|
|
|
|
emitter.onError((e) -> emitters.remove(token));
|
|
|
|
|
emitter.onCompletion(() -> {
|
|
|
|
|
SseEmitter remove = emitters.remove(token);
|
|
|
|
|
if (remove != null) {
|
|
|
|
|
remove.complete();
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
emitter.onTimeout(() -> {
|
|
|
|
|
SseEmitter remove = emitters.remove(token);
|
|
|
|
|
if (remove != null) {
|
|
|
|
|
remove.complete();
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
emitter.onError((e) -> {
|
|
|
|
|
SseEmitter remove = emitters.remove(token);
|
|
|
|
|
if (remove != null) {
|
|
|
|
|
remove.complete();
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
try {
|
|
|
|
|
// 向客户端发送一条连接成功的事件
|
|
|
|
|
@ -106,7 +121,10 @@ public class SseEmitterManager {
|
|
|
|
|
.name("message")
|
|
|
|
|
.data(message));
|
|
|
|
|
} catch (Exception e) {
|
|
|
|
|
emitters.remove(entry.getKey());
|
|
|
|
|
SseEmitter remove = emitters.remove(entry.getKey());
|
|
|
|
|
if (remove != null) {
|
|
|
|
|
remove.complete();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
|