线程池任务统计应用

最新推荐文章于 2024-04-10 01:00:36 发布

原创最新推荐文章于 2024-04-10 01:00:36 发布 · 4.5k 阅读

1 ·

CC 4.0 BY-SA版权

JAVA 专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种自定义线程池的方法，通过继承ThreadPoolExecutor并覆盖其计数功能，实现了任务计数的重置。示例代码展示了如何创建自定义线程池并使用它来执行多个任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

ThreadPoolExecutor没有重置计数的功能，简单实现一下

public class NewThreadPoolExecutor extends ThreadPoolExecutor {

	private long tempCompletedTaskCount;

	private long tempTaskCount;

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

	public NewThreadPoolExecutor(int nThreads) {
		super(nThreads, nThreads, 0L, TimeUnit.MILLISECONDS, new LinkedBlockingQueue<Runnable>());
	}

	public long getCompletedTaskCount() {
		return super.getCompletedTaskCount() - tempCompletedTaskCount;
	}

	public long getTaskCount() {
		return super.getTaskCount() - tempTaskCount;
	}

	public void resetTaskCount() {
		if (super.getCompletedTaskCount() == super.getTaskCount()) {
			tempCompletedTaskCount = super.getCompletedTaskCount();
			tempTaskCount = super.getTaskCount();
		}
	}
}

public class Test {
	private static NewThreadPoolExecutor executorService = new NewThreadPoolExecutor(5);
	
	public static void main(String[] args) throws Exception {
		final Future<Integer> f1 = executorService.submit(new Callable<Integer>() {
			@Override
			public Integer call() throws Exception {
				TimeUnit.SECONDS.sleep(2);
				int temp = 0;
				while (temp++ < 30)
					System.out.println("F1-" + temp + Thread.currentThread().getName());
				return Integer.valueOf(temp);
			}
		});
		final Future<Integer> f2 = executorService.submit(new Callable<Integer>() {
			@Override
			public Integer call() throws Exception {
				int temp = 0;
				while (temp++ < 30)
					System.out.println("F2-" + temp + Thread.currentThread().getName());
				;
				return Integer.valueOf(temp);
			}
		});
		final Future<Integer> f3 = executorService.submit(new Callable<Integer>() {
			@Override
			public Integer call() throws Exception {
				int temp = 0;
				while (temp++ < 30)
					System.out.println("F3-" + temp + Thread.currentThread().getName());
				;
				return Integer.valueOf(temp);
			}
		});
		executorService.execute(new Runnable() {
			@Override
			public void run() {
				try {
					System.out.println(Thread.currentThread().getName());
					System.out.println(f1.get() + f2.get() + f3.get());
				} catch (InterruptedException e) {
					e.printStackTrace();
				} catch (ExecutionException e) {
					e.printStackTrace();
				}
			}
		});

		System.out.println("【线程活跃数】" + executorService.getActiveCount());
		System.out.println("【总任务数】" + executorService.getTaskCount());
		System.out.println("【已完成任务数】" + executorService.getCompletedTaskCount());
		TimeUnit.SECONDS.sleep(8);
		System.out.println("【线程活跃数】" + executorService.getActiveCount());
		System.out.println("【总任务数】" + executorService.getTaskCount());
		System.out.println("【已完成任务数】" + executorService.getCompletedTaskCount());

		// 重新计数
		executorService.resetTaskCount();
		final Future<Integer> f4 = executorService.submit(new Callable<Integer>() {
			@Override
			public Integer call() throws Exception {
				TimeUnit.SECONDS.sleep(3);
				int temp = 0;
				while (temp++ < 30)
					System.out.println("F4-" + temp + Thread.currentThread().getName());
				;
				return Integer.valueOf(temp);
			}
		});
		f4.get();
		System.out.println("【线程活跃数】" + executorService.getActiveCount());
		System.out.println("【总任务数】" + executorService.getTaskCount());
		System.out.println("【已完成任务数】" + executorService.getCompletedTaskCount());
	}
}