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

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

请看下面的一个简单案例,有一个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());

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

}
执行结果:

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

posted @ 2021-03-04 13:21  satire  阅读(782)  评论(0)    收藏  举报