SpringBoot Test 多线程报错:dataSource already closed

2022-10-15,,,,

1:前言

最近在项目中使用多线程对大任务拆分处理时,进行数据库操作的时候报错了。

业务代码大概是这样的:

@Service
public calss TestServiceImpl implements TestService{
@Resource
private TestMapper testMapper; public void insert(Test test){
ThreadPoolExecutor executor = new ThreadPoolExecutor(...); executor.execute(() -> {
testMapper.insert(test);
});
} }

测试用例代码:

@RunWith(SpirngRunner.class)
@SpringBootTest
public class ApplicationTests{ @Resource
private TestService testService; @Test
public void insertTest(){
Test test = new Test();
testService.insert(test);
} }

2:排查思路

1:项目中使用了一个很老旧的定时器工具(LTS),由于配置未接入,启动时LTS一直会报错的,

我首先怀疑是LTS的问题,是不是内部某部分源码调用了DruidDataSource的close方法。

因此我在项目中先把LTS排除不让它启动。

结果:然并卵

2:后来怀疑是不是项目中的定时任务造成的,因为数据源关闭前打印了以下日志

可以看到 Thread-2 执行了ThreadPoolTaskExecutor 再关闭数据源的。

因此我在SpringBoot 的启动类上加上了以下代码

@SpringBootApplication(exclude = {TaskExecutionAutoConfiguration.class})

结果:然并卵

3:关电脑回家吃饭睡觉

第二天早上回到工位上缓了一会,思考还有什么原因会导致这种问题。

因为我们的线程池是一个封装的工具类封装过的(我封装的),里面有两行代码引起了我的注意:

我先把 threadPoolExecutor.awaitTermination(3, TimeUnit.MINUTES); 这行代码注释掉,重新跑测试用例。

发现并没有报错,并且程序很快就终止运行了。

在这里我就想到了,SpringBoot Test 中主线程退出,是不会管用户线程是否结束了任务的这个问题。

3:根本原因

1:SpringBoot Test 主线程退出,导致Spring 容器关闭。

2:Spring容器关闭,导致DruidDataSource 关闭

3:此时用户线程去访问已关闭的数据源,导致报错。

这样不行啊,那我测试用例怎么跑完呢?

可能这个时候有人会想到在测试用例最后面加以下代码:

TimeUnit.SECONDS.sleep(30);

可是本人觉得不够优雅,也不灵活。

4:解决方案

监听容器关闭,关闭前先判断线程池中是否存在未执行玩任务的线程。

ThreadPoolExecutor 的源码:

因此我们是可以知道是否存在未执行完任务的线程的。

所以我在项目中写了以下代码:

当线程池中存在未执行完任务的线程的时候,就先等待,直到所有任务完成或超过等待时间。

可把自己牛逼坏了(叉会腰)!

SpringBoot Test 多线程报错:dataSource already closed的相关教程结束。

《SpringBoot Test 多线程报错:dataSource already closed.doc》

下载本文的Word格式文档,以方便收藏与打印。