线程池中如何寻找堆栈？如何打印出堆栈的异常信息？

先由一个案例引入，来解释为什么要寻找线程池中的堆栈。

请看下面的一个简单案例，有一个Runable接口，用它来计算两个数的商。

public class DivTask implements Runnable {
    int a,b;
    public DivTask(int a, int b) {
        this.a = a;
        this.b = b;
    }
    @Override
    public void run() {
        double re = a/b;
        System.out.println(re);
    }
}

如果程序运行了这个任务，那么我们期望他打印出给定的结果。现在构造几个这样的任务，希望程序可以给定出预期的一组结果：

public static void main(String[] args) {

    ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(0,Integer.MAX_VALUE, 0L,TimeUnit.SECONDS,
            new SynchronousQueue<Runnable>());

    for(int i = 0; i<5; i++){
        threadPoolExecutor.submit(new DivTask(100,i));
    }
}

上述代码将DivTask提交到线程池，从for循环来看，应该是得到五个结果，分别是100除以给定的i后的商。但如果你真的运行程序，得到的结果是：

 

只有四个输出，也就是说程序漏算了一个？并且没有给出任何日志信息，就好像一切正一样。

再来看程序其实你会发现，第一个传过去的i是0，应该出现异常才对。也就是说线程池将抛出的异常吃掉了？而此时如果没有堆栈信息输出的话，就会导致我们对程序的错误一无所知。

那么怎么向线程池讨回异常堆栈的信息就是接下来要考虑的事情了

 

1、一种最简单的办法就是，将threadPoolExecutor.submit(new DivTask(100,i));改为threadPoolExecutor.execute(new DivTask(100,i)); 或者对submit()方法进行改造。这两种方法都可以得到部分堆栈信息，请看执行结果：

但是，这种方式打印的堆栈信息很有限。如果想知道这个出现异常的任务是哪里提交的？而任务的具体提交位置已经被线程池完全淹没了，怎么办？这就用到了第二种方式

 

2、扩展ThreadPoolExecuort线程池

让他在调度任务之前，先保存一下提交任务线程的堆栈信息。 

public class TraceThreadPoolExecutor extends ThreadPoolExecutor {

    public TraceThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue<Runnable> workQueue) {
        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("client stack trace");
    }

    private Runnable wrap(final Runnable task,final Exception clientStack,String clientThreadName){
        return new Runnable() {

            @Override
            public void run() {
                try {
                    task.run();
                } catch (Exception e) {
                    clientStack.printStackTrace();
                    throw e;
                }
            }
        };
    }
}

wrap()方法的第2个参数为一个异常，保存着提交任务的线程的堆栈信息。该方法将传入的Runable任务进行一次包装，使之能处理异常信息。当任务发生异常时，这个异常会被打印。

现在用新成员TraceThreadPoolExecutor来尝试执行这段代码：

public static void main(String[] args) {
    ThreadPoolExecutor threadPoolExecutor = new TraceThreadPoolExecutor(0,Integer.MAX_VALUE, 0L,TimeUnit.SECONDS,
            new SynchronousQueue<Runnable>());

    for(int i = 0; i<5; i++){
        threadPoolExecutor.submit(new DivTask(100,i));
    }
}

执行结果：

 

现在熟悉的异常又回来了，不仅可以得到异常发生的Runable实现的信息，也知道了任务是在哪里提交的。可以帮助我们瞬间找到问题所在。