本文共 3676 字,大约阅读时间需要 12 分钟。
ExecutorService关闭机制
在多线程编程中,ExecutorService是处理任务的核心工具之一。它能够管理多个线程任务的执行和终止过程。在某些情况下,我们需要提前关闭ExecutorService,这通常涉及到两种主要方法:shutdown和shutdownNow。了解这两种方法的区别以及它们的适用场景,是掌握 ExecutorService 使用的关键。
Shutdown和ShutdownNow的区别
ExecutorService提供了两种终止方法:shutdown和shutdownNow。两者的主要区别在于它们的安全性和响应速度。Shutdown方法在关闭时会等待所有正在执行的任务完成,这确保了操作的安全性,但响应速度较慢。ShutdownNow方法则强行关闭所有执行中和未执行的任务,响应速度更快,但可能导致正在执行的任务中断,这增加了操作的风险。
使用Shutdown
Shutdown方法的典型使用场景是当我们希望确保所有任务都能安全完成时。以下是一个使用Shutdown的示例:
public void useShutdown() throws InterruptedException { ExecutorService executor = Executors.newFixedThreadPool(10); Runnable runnableTask = () -> { try { TimeUnit.MILLISECONDS.sleep(300); } catch (InterruptedException e) { e.printStackTrace(); } }; executor.submit(runnableTask); executor.shutdown(); executor.awaitTermination(800, TimeUnit.MILLISECONDS);}
在这个示例中,awaitTermination
方法会阻塞直到所有任务完成,或者在指定的等待时间内超时。这种方式保证了任务的安全终止。
使用ShutdownNow
ShutdownNow方法适用于需要立即停止所有任务的情况。它会取消所有正在执行的任务,并返回尚未开始执行的任务。以下是一个使用ShutdownNow的示例:
public void useShutdownNow() throws InterruptedException { TrackingExecutor trackingExecutor = new TrackingExecutor(Executors.newCachedThreadPool()); Runnable runnableTask = () -> { try { TimeUnit.MILLISECONDS.sleep(300); } catch (InterruptedException e) { e.printStackTrace(); } }; trackingExecutor.submit(runnableTask); ListnotRunList = trackingExecutor.shutdownNow(); if (trackingExecutor.awaitTermination(800, TimeUnit.SECONDS)) { List runButCancelledList = trackingExecutor.getCancelledTask(); }}
这里,trackingExecutor
是一个自定义实现的ExecutorService,它能够跟踪被取消的任务。getCancelledTask
方法返回那些在关闭时被中断的任务。这种方式适用于需要立即停止所有正在执行的任务,而不必等待所有任务完成的情况。
自定义ExecutorService实现
有时,我们可能需要自定义ExecutorService以满足特定的需求。以下是一个自定义实现的示例:
public class TrackingExecutor extends AbstractExecutorService { private final ExecutorService executorService; private final SettaskCancelledAtShutdown = Collections.synchronizedSet(new HashSet<>()); public TrackingExecutor(ExecutorService executorService) { this.executorService = executorService; } @Override public void shutdown() { executorService.shutdown(); } @Override public List shutdownNow() { return executorService.shutdownNow(); } @Override public boolean isShutdown() { return executorService.isShutdown(); } @Override public boolean isTerminated() { return executorService.isTerminated(); } @Override public boolean awaitTermination(long timeout, TimeUnit unit) throws InterruptedException { return executorService.awaitTermination(timeout, unit); } @Override public void execute(Runnable command) { executorService.execute(() -> { try { command.run(); } finally { if (isShutdown() && Thread.currentThread().isInterrupted()) { taskCancelledAtShutdown.add(command); } } }); } @Override public List getCancelledTask() { if (!executorService.isTerminated()) { throw new IllegalStateException("executorService is not terminated"); } return new ArrayList<>(taskCancelledAtShutdown); }}
这个自定义实现在执行任务时,检查是否已经关闭。如果关闭且当前线程被中断,它会将任务记录下来。通过getCancelledTask
方法,可以获取这些被中断的任务。
结论
ExecutorService的shutdown和shutdownNow方法提供了不同的任务终止机制。Shutdown方法确保所有任务安全完成,而ShutdownNow方法允许立即停止所有任务。选择哪种方法取决于具体的应用场景和需求。在某些情况下,自定义ExecutorService实现可能会提供额外的功能,如跟踪中断的任务。理解这些机制是有效管理多线程任务的关键。
转载地址:http://rhnuz.baihongyu.com/