Spring中实现的三种异步流式接口方法

@GetMapping("/bodyEmitter")
public ResponseBodyEmitter handle() {
    // 创建一个ResponseBodyEmitter，-1代表不超时
    ResponseBodyEmitter emitter = new ResponseBodyEmitter(-1L);

    // 异步执行耗时操作
    CompletableFuture.runAsync(() -> {
        try {
            for (int i = 0; i < 10000; i++) {
                System.out.println("bodyEmitter " + i);
                // 发送数据
                emitter.send("bodyEmitter " + i + " @ " + new Date() + "\n");
                Thread.sleep(2000);
            }
            // 完成
            emitter.complete();
        } catch (Exception e) {
            // 发生异常时结束接口
            emitter.completeWithError(e);
        }
    });

    return emitter;
}

@GetMapping("/subSseEmitter/{userId}")
public SseEmitter sseEmitter(@PathVariable String userId) {
    log.info("sseEmitter: {}", userId);
    SseEmitter emitterTmp = new SseEmitter(-1L);
    // 将SseEmitter对象进行持久化，以便在消息产生时直接取出对应的发送器
    EMITTER_MAP.put(userId, emitterTmp);

    CompletableFuture.runAsync(() -> {
        try {
            // 模拟发送数据
            SseEmitter.SseEventBuilder event = SseEmitter.event()
                .data("sseEmitter" + userId + " @ " + LocalTime.now());
            emitterTmp.send(event);
            // 这里可以添加更多的发送逻辑
        } catch (Exception e) {
            emitterTmp.completeWithError(e);
        }
    });

    return emitterTmp;
}

@GetMapping("/streaming")
public StreamingResponseBody streaming() {
    return outputStream -> {
        // 异步执行耗时操作
        CompletableFuture.runAsync(() -> {
            try {
                for (int i = 0; i < 10000; i++) {
                    String data = "Streaming data " + i + "\n";
                    outputStream.write(data.getBytes());
                    outputStream.flush();
                    Thread.sleep(1000);
                }
                outputStream.close();
            } catch (Exception e) {
                // 处理异常
            }
        });
    };
}