Spring统一异常处理捕获不到CompletableFuture异步编排中的异常的问题

Spring统一异常处理捕获不到CompletableFuture异步编排中的异常的问题

Spring统一异常处理

我们常常使用@RestControllerAdvice或@ControllerAdvice配合@ExceptionHandler来做全局的统一异常处理，这种方式被认为是十分高效的

简单例子

/**
 * @author Syf200208161018
 * @date 2022/12/3 15:21
 * @ClassName:ExcpHandler
 * @Effect:ExcpHandler is used for
 */
@RestControllerAdvice(annotations = {RestController.class})
public class ExcpHandler {

    @ExceptionHandler(value = {Exception.class})
    public String deal(Exception e){
        System.out.println(e.getMessage());
        return e.getMessage();
    }
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

在这个例子中，只要Controller中（带有@RestController注解）出现任意异常就会被捕捉到！
但是我们需要明确的是，我上面这句话只在主线程的情况下成立！

多线程无法捕获场合

正常场合（不使用异步编排）代码

ServiceIm

在这个类中我们看到有一个int i = 1/0;表明一定会引发除0异常！

/**
 * @author Syf200208161018
 * @date 2022/12/3 15:23
 * @ClassName:Service
 * @Effect:Service is used for
 */
@Service
public class ServiceIm {
    public void test(){
        int i = 1/0;
    }
}

1
2
3
4
5
6
7
8
9
10
11
12
13

ExcpHandler

/**
 * @author Syf200208161018
 * @date 2022/12/3 15:21
 * @ClassName:ExcpHandler
 * @Effect:ExcpHandler is used for
 */
@RestControllerAdvice(annotations = {RestController.class})
public class ExcpHandler {

    @ExceptionHandler(value = {Exception.class})
    public String deal(Exception e){
        System.out.println(e.getMessage());
        return e.getMessage();
    }
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

Controller

/**
 * @author Syf200208161018
 * @date 2022/12/3 15:22
 * @ClassName:Controller
 * @Effect:Controller is used for
 */
@RestController
public class Controller {
    @Autowired
    private ServiceIm serviceIm;

    @GetMapping("/test")
    public String test(){
         serviceIm.test();
        return "ok";
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

结果

可以看到结果很正常，确实把错误返回了

异常场合（使用异步编排）代码

我们只要把Controller中的代码改成如下：

Controller

/**
 * @author Syf200208161018
 * @date 2022/12/3 15:22
 * @ClassName:Controller
 * @Effect:Controller is used for
 */
@RestController
public class Controller {
    @Autowired
    private ServiceIm serviceIm;

    @GetMapping("/test")
    public String test(){
        CompletableFuture.runAsync(() -> {
            serviceIm.test();
        }).exceptionally((e) -> {
            throw new RuntimeException("cuowu");
        });
        
        return "ok";
    }
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

结果

可以看到结果是ok，而且查看控制台并没有进行异常捕获，有些人可能说是因为已经使用了异步编排的exceptionally做了异常处理，可是我们要知道我在里面是直接抛出RuntimeException的！，那去除异步编排的异常处理可以吗？答案是也无法捕获到，更本原因就是并不发生在主线程上，或者说，主线程已经执行到了return但是实际我们的业务并没有结束

使用join方法阻塞

我们在学习异步编排的时候有一个join方法不知道大家记不记得
用于对任务线进行阻塞,join和get都可以，看场景，单纯阻塞无返回值建议join

修改

@RestController
public class Controller {
    @Autowired
    private ServiceIm serviceIm;

    @GetMapping("/test")
    public String test(){
        final CompletableFuture<Void> thread1 = CompletableFuture.runAsync(() -> {
            serviceIm.test();
        }).exceptionally((e) -> {
            throw new RuntimeException("cuowu");
        });
        thread1.join();

        return "ok";
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

结果

结果我们发现成功了！当然这是意料之中的！因为这时候主线程就阻塞等待异步编排结束，异常也就被捕获到了

使用join的问题（不使用exceptionally捕获）

虽然join可以解决我们的问题，但是阻塞带来的问题就是性能变低，效率变低，我们使用异步编排的目的就是为了提升效率，所以若不用exceptionally进行捕获就会导致效率降低的问题

结论

如此我们得到一个结论，使用异步编排的正确姿势：

1. 构建线程池
2. 任务结束使用exceptionally进行捕获
3. 使用join阻塞防止异常捕获失效
4. 使用统一异常处理捕获

但是注意这只是对于时效性强必须有返回值的方法才能这样做！！！
也就是对于不要求及时返回的不要这样干，这里我们建议使用RabbitMQ处理！！！或者简单的使用@Async