版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/danielzhou888/article/details/84074366
带日志和计时等功能的线程池
欢迎关注作者博客
简书传送门
前言
我们先来聊聊ThreadPoolExecutor的拓展。
扩展ThreadPoolExecutor
在执行任务的线程中将调用beforeExecute和afterExecute等方法,在这些方法中还可以添加日志、计时、监视或者统计信息收集的功能。无论任务是从run中正常返回,还是抛出一个异常而返回,afterExecute都会被调用。如果任务在完成后带有一个Error,那么就不会调用afterExecute。如果beforeExecute抛出一个RuntimeException,那么任务将不被执行,并且afterExecute也不会被调用。
在线程池完成关闭时调用terminated,也就是在所有任务都已经完成并且所有工作者线程也已经关闭后,terminated可以用来释放Executor在其生命周期里分配的各种资源,此外还可以执行发送通知、记录日志或者手机finalize统计等操作。
ThreadPoolExecutor使用了模板方法模式,提供了beforeExecute、afterExecute和terminated扩展方法。
- 线程执行前调用beforeExecute(如果beforeExecute抛出了一个RuntimeException,那么任务将不会被执行)
- 线程执行后调用afterExecute(抛出异常也会调用,如果任务在完成后带有一个Error,那么就不会调用afterExecute)
- 在线程池完成关闭操作时调用terminated,也就是所有任务都已经完成并且所有工作者线程也已经关闭后
源码
/**
* @program:
* @description: 具备日志和计时等功能的线程池
* @author: zhouzhixiang
* @create: 2018-11-14 20:40
*/
public class TimingThreadPool extends ThreadPoolExecutor{
private final ThreadLocal<Long> starttime = new ThreadLocal<>();
private final AtomicLong numtask = new AtomicLong();
private final AtomicLong totaltime = new AtomicLong();
private final Logger logger = Logger.getLogger("TimingThreadPool");
public TimingThreadPool(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue<Runnable> workQueue) {
super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue);
}
@Override
protected void beforeExecute(Thread t, Runnable r) {
super.beforeExecute(t, r);
logger.fine(String.format("Thread %s: start %s", t, r));
starttime.set(System.nanoTime());
}
@Override
protected void afterExecute(Runnable r, Throwable t) {
try {
long endtime = System.nanoTime() - starttime.get();
totaltime.addAndGet(endtime);
numtask.incrementAndGet();
logger.fine(String.format("Thread %s: end %s, time=%dns", t, r, totaltime));
}finally {
super.afterExecute(r, t);
}
}
@Override
protected void terminated() {
try{
logger.info(String.format("Terminated: avg time=%dns", totaltime.get() / numtask.get()));
}finally {
super.terminated();
}
}
}
欢迎加入Java猿社区