| | |
| | | |
| | | import java.util.List; |
| | | |
| | | /** |
| | | * SSE 控制器 |
| | | * |
| | | * @author Lion Li |
| | | */ |
| | | @RestController |
| | | @ConditionalOnProperty(value = "sse.enabled", havingValue = "true") |
| | | @RequiredArgsConstructor |
| | |
| | | |
| | | private final SseEmitterManager sseEmitterManager; |
| | | |
| | | /** |
| | | * 建立 SSE 连接 |
| | | */ |
| | | @GetMapping(value = "${sse.path}", produces = MediaType.TEXT_EVENT_STREAM_VALUE) |
| | | public SseEmitter connect() { |
| | | String tokenValue = StpUtil.getTokenValue(); |
| | |
| | | return sseEmitterManager.connect(userId, tokenValue); |
| | | } |
| | | |
| | | /** |
| | | * 关闭 SSE 连接 |
| | | */ |
| | | @SaIgnore |
| | | @GetMapping(value = "${sse.path}/close") |
| | | public R<Void> close() { |
| | |
| | | return R.ok(); |
| | | } |
| | | |
| | | /** |
| | | * 向特定用户发送消息 |
| | | * |
| | | * @param userId 目标用户的 ID |
| | | * @param msg 要发送的消息内容 |
| | | */ |
| | | @GetMapping(value = "${sse.path}/send") |
| | | public R<Void> send(Long userId, String msg) { |
| | | SseMessageDto dto = new SseMessageDto(); |
| | |
| | | return R.ok(); |
| | | } |
| | | |
| | | /** |
| | | * 向所有用户发送消息 |
| | | * |
| | | * @param msg 要发送的消息内容 |
| | | */ |
| | | @GetMapping(value = "${sse.path}/sendAll") |
| | | public R<Void> send(String msg) { |
| | | sseEmitterManager.publishAll(msg); |
| | | return R.ok(); |
| | | } |
| | | |
| | | /** |
| | | * 清理资源。此方法目前不执行任何操作,但避免因未实现而导致错误 |
| | | */ |
| | | @Override |
| | | public void destroy() throws Exception { |
| | | // 销毁时不需要做什么 此方法避免无用操作报错 |