您的位置:首页 > 其它

封装ThreadPoolExecutor 支持打印异常堆栈

2017-06-22 16:15 204 查看
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;
}
}
};
}
}


测试代码如下:

public class DoIT {

public static 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;
println(re);
}

}

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

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

}

static void println(Object str) {
System.out.println(str);
}

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: