SpringMVC异步处理的 5 种方式

时间:2021-03-02 程序之心人气:0

作者：丁仪

来源：https://chengxuzhixin.com/blog/post/SpringMVC-yi-bu-chu-li-de-5-zhong-fang-shi.html

前段时间研究了下 diamond 的原理，其中有个重要的知识点是长连接的实现，用到了 servlet 的异步处理。异步处理最大的好处是可以提高并发量，不阻塞当前线程。其实 Spring MVC 也支持了异步处理，本文记录下相关的技术点。

异步处理 demo

如果要启用异步返回，需要开启 @EnableAsync。如下的代码中，使用 DeferredResult 进行异步处理。

请求进来后，首先创建 DeferredResult 对象，设置超时时间为 60 秒。然后指定 DeferredResult 在异步完成和等待超时时的回调。同步的处理只需要创建异步任何，然后返回 DeferredResult 即可。这样 Spring MVC 处理完此次请求后，不会立即返回 response 给客户端，会一直等待 DeferredResult 处理完成。如果 DeferredResult 没有在 60 秒内处理完成，就会触发超时，然后返回 response 给客户端。

@RequestMapping(value = "/async/demo")
public DeferredResult<String> async(){
    // 创建 DeferredResult，设置超时时间 60s
    DeferredResult<String> deferredResult = new DeferredResult<>((long)60 * 1000);

    String uuid = UUID.randomUUID().toString();
    Runnable callback = () -> manager.remove(deferredResult, uuid);
    // 设置完成和超时的回调
    deferredResult.onCompletion(callback);
    deferredResult.onTimeout(callback);

    // 创建异步任务
    manager.addAsyncTask(deferredResult, uuid);

    // 同步返回 DeferredResult
    return deferredResult;
}

对于异步任务来说，需要持有 DeferredResult 对象。在异步处理结束时，需要手动调用 DeferredResult.setResult 完成输出。调用 setResult 时，数据输出写到客户端，然后触发异步完成事件执行回调。

task.getDeferredResult().setResult(ConfigJsonUtils.toJsonString(map));

使用 DeferredResult 进行异步处理

DeferredResult 这个类代表延迟结果。DeferredResult 可以用在异步任务中，其他线程能够获取 DeferredResult 并设置 DeferredResult 的返回数据。通常可以使用线程池、队列等配合 DeferredResult 实现异步处理。

根据官方描述，Spring MVC 处理流程如下：

把 controller 返回的 DeferredResult 保存在内存队列或集合当中；
Spring MVC 调用 request.startAsync()，开启异步；
DispatcherServlet 和所有的 Filter 退出当前请求线程；
业务应用在异步线程中设置 DeferredResult 的返回值，Spring MVC 会再次发送请求；
DispatcherServlet 再次被调用，并使用 DeferredResult 的返回值；

使用 Callable 进行异步处理

使用 Callable 进行异步处理与 DeferredResult 类似。不同的是，Callable 会交给系统指定的 TaskExecutor 执行。

根据官方描述，Spring MVC 处理流程如下：

controller 返回 Callable；
Spring MVC 调用 request.startAsync()，开启异步，提交 Callable 到一个任务线程池；
DispatcherServlet 和所有的 Filter 退出当前请求线程；
业务应用在异步线程中返回值，Spring MVC 会再次发送请求；
DispatcherServlet 再次被调用，并使用 Callable 的返回值；

@RequestMapping(value = "/async/demo")
public Callable<String> async(){
    Callable<String> callable = () -> String.valueOf(System.currentTimeMillis());
    // 同步返回
    return callable;
}

使用 ListenableFuture 进行异步处理

ListenableFuture 作为返回值，与 DeferredResult 类似。也需要使用者自行处理异步线程，但不支持超时、完成回调，需要自行处理。

@RequestMapping(value = "/async/demo")
public ListenableFuture<String> async(){
    ListenableFutureTask<String> ListenableFuture= new ListenableFutureTask<>(() -> {
        return String.valueOf(System.currentTimeMillis());
    });
    Executors.newSingleThreadExecutor().submit(ListenableFuture);
    return ListenableFuture;
}

使用 ResponseBodyEmitter 进行异步处理

DeferredResult 和 Callable 都只能返回一个异步值。如果需要返回多个对象，就要使用 ResponseBodyEmitter。返回的每个对象都会被 HttpMessageConverter 处理并写回输出流。如果希望设置更多返回数据，如 header、status 等，可以把 ResponseBodyEmitter 作为 ResponseEntity 的实体数据返回。

@RequestMapping("/async/responseBodyEmitter")
public ResponseBodyEmitter responseBodyEmitter(){
    ResponseBodyEmitter responseBodyEmitter=new ResponseBodyEmitter();

    Executors.newSingleThreadExecutor().submit(() -> {
        try {
            responseBodyEmitter.send("demo");
            responseBodyEmitter.send("test");
            responseBodyEmitter.complete();
        } catch (Exception ignore) {}
    });

    return responseBodyEmitter;
}

使用 StreamingResponseBody 进行异步处理

如果希望跳过返回值的自动转换，直接把输出流写入 OutputStream，可以使用 StreamingResponseBody。也可以作为 ResponseEntity 的实体数据返回。

@RequestMapping("/async/streamingResponseBody")
public StreamingResponseBody streamingResponseBody(){
    StreamingResponseBody streamingResponseBody = outputStream -> {
        Executors.newSingleThreadExecutor().submit(() -> {
            try {
                outputStream.write("<html>streamingResponseBody</html>".getBytes());
            } catch (IOException ignore) {}
        });
    };
    return streamingResponseBody;
}

各种处理方式的对比

以上几种异步处理方式各有差异，需要按需取舍。对比如下。

	可返回次数	数据转换	回调	线程池
DeferredResult	1 次	有	完成、超时	自行处理
Callable	1 次	有	无	系统处理
ListenableFuture	1 次	有	无	自行处理
ResponseBodyEmitter	多次	有	无	自行处理
StreamingResponseBody	多次	无	无	自行处理