使用ExecutorService来停止线程服务-白红宇

使用ExecutorService来停止线程服务

阅读量：427 次

发布时间：2019-03-06

本文共 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);
    List
   
     notRunList = 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 Set
   
     taskCancelledAtShutdown = 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/

你可能感兴趣的文章