关于线程池处理任务之后还需要return自定义数据

文章讨论了在使用线程池处理Excel导入数据库任务时,如何确保在数据插入完成后返回主键的问题。提出了两种解决方案:一是使用`executor.shutdown()`使主线程阻塞,但可能导致线程池关闭;二是使用`CountDownLatch`计数器来控制任务执行顺序,但需要预先知道任务次数。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

问题:在使用线程池处理导入的excel文件信息,并存到数据库,然后就需要返回这一批刚导入数据库的excel数据的主键,并通过主键查询数据

问题发现:在执行过程中,线程池属于异步多线程执行,与主线程无关,此时excel的信息还没有导入数据库,但是主键id已经返回了,前端此时根据主键查询信息必然为null。

解决思路:此次问题,最大的原因就是因为使用线程池,加快导入信息的速度,但是因为是异步任务,才导致的。所以应该是主线程等待线程池内的任务处理完毕后再返回主键。

解决方案一、

思路:让线程池任务执行完之前主线程要一直处于阻塞状态。

具体操作:使用下面两个方法,executor.shutdown();是停止线程池,此时线程池不能接受新任务,把原有的任务执行完毕后就会停止。executor.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS);因为主线程一直在阻塞,所以要设置一个最长等待时间,不可能一直处于阻塞状态,超过等待时间则抛出异常

缺点:因为我们使用了shutdown()方法,所以当前的线程池关闭了,当你下次再次使用的时候就不会再执行任务,所以我们需要每次执行任务都new 一个线程池出来。

解决方案二、

思路:根据要执行的任务的数量,分配一个计数器,每执行一次,计数器就减一,如果是0,就代表当前任务全部执行完毕,开始执行主线程。

具体操作:使用CountDownLatch 作为计数器,每次执行完一次run方法就调用一次countDown()方法,计数器减一

缺点:你得知道具体的你要执行几次任务

public class 线程池异步问题 {
 public static void main(String[] args) {
        ThreadPoolExecutor executor = new  ThreadPoolExecutor(20,25,100L,
            TimeUnit.SECONDS,new LinkedBlockingQueue<>(),new ThreadPoolExecutor.CallerRunsPolicy());
   
        CountDownLatch latch = new CountDownLatch(1);
        ThreadTask threadTask = new ThreadTask(latch);

        try {
            executor.execute(threadTask);
            latch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        System.out.println("123");
    }
}
------------线程方法-------------
public class ThreadTask implements Runnable {
    private  CountDownLatch latch;

    public ThreadTask(CountDownLatch latch) {
        this.latch = latch;
    }

    public ThreadTask() {
    }

    @Override
    public void run() {
        try {
            System.out.println("hello");
            System.out.println("world");
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
            latch.countDown();
        }
    }
}

正确结果应该是 :

hello

world

123 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

刘个Java

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值