扩展线程池捕获线程异常

如何捕获线程池中的线程异常

简单案例

我们先看一个案例，实现Runnable接口，计算两个数的商 创建一个线程池，参数如下 提交任务到线程池，查看执行结果 执行结果

我们发现，我们代码中，提交了5个线程，但最终只打印了4个结果， 而且没有报何错误，很明显100/0的那个任务没有打印 100/0会报除零异常，但是显然这次没有报任何错误

改用execute()提交线程

为了获得线程报错信息，我们可以改用execute()提交线程

从这里的控制台，我们得到了部分的报错信息，但是我们只能知道异常是在哪里抛出来的，ThreadPoolTest.java的第51行； 但是我们还是希望得到其他的更重要的信息，这个任务在哪里提交的？

扩展线程池ThreadPoolExecutor

我们可以扩展一下ThreadPoolExecutor这个线程池，让他在任务调度前，先保存一下提交任务线程的堆栈信息

package com.stream.juc;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.Future;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * 扩展线程池
 * 捕获线程池的异常
 *
 * @author stream
 * @since 2021/9/15 7:51
 */
class ExtenThreadPoolExecutor extends ThreadPoolExecutor {
          
   

    public ExtenThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue<Runnable> workQueue, CallerRunsPolicy callerRunsPolicy) {
          
   
        super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue);
    }

    @Override
    public void execute(Runnable command) {
          
   
        super.execute(wrap(command,clientTrace(),Thread.currentThread().getName()));
    }

    @Override
    public Future<?> submit(Runnable task) {
          
   
        return super.submit(wrap(task,clientTrace(),Thread.currentThread().getName()));
    }

	// 抛出异常信息
    private Exception clientTrace(){
          
   
        return new Exception("Thread Stack");
    }

    private Runnable wrap(Runnable task, Exception exception , String name) {
          
   
        return new Runnable() {
          
   
            @Override
            public void run() {
          
   
                try {
          
   
                    task.run(); // 捕获异常
                }catch (Exception e){
          
   
                    exception.printStackTrace();
                    throw e;
                }
            }
        };
    }
}

我们用扩展后的线程池提交任务，还是刚才的代码 这样，我们就得到了异常发生的内部堆栈信息，帮助我们快速定位问题