如何中断ComppletableFuture的底层执行

concurrency java completable-future

2022-09-01 12:10:24

我知道设计不会通过中断来控制其执行，但我想你们中的一些人可能会遇到这个问题。s 是编写异步执行的好方法，但是考虑到您希望在取消 future 时中断或停止底层执行的情况，我们该怎么做呢？或者我们必须接受任何取消或手动完成不会影响线程在那里完成它？CompletableFutureCompletableFutureCompletableFuture

在我看来，这显然是一项无用的工作，需要执行者工人的时间。我想知道在这种情况下，什么方法或设计可能会有所帮助？

更新

这是一个简单的测试

public class SimpleTest {

  @Test
  public void testCompletableFuture() throws Exception {
    CompletableFuture<Void> cf = CompletableFuture.runAsync(()->longOperation());

    bearSleep(1);

    //cf.cancel(true);
    cf.complete(null);

    System.out.println("it should die now already");
    bearSleep(7);
  }

  public static void longOperation(){
    System.out.println("started");
    bearSleep(5);
    System.out.println("completed");
  }

  private static void bearSleep(long seconds){
    try {
      TimeUnit.SECONDS.sleep(seconds);
    } catch (InterruptedException e) {
      System.out.println("OMG!!! Interrupt!!!");
    }
  }
}

答案 1

可完成的未来与最终可能完成它的异步操作无关。

由于（与）这个类没有直接控制导致它完成的计算，取消被视为另一种形式的特殊完成。方法具有与相同的效果。FutureTaskcancelcompleteExceptionally(new CancellationException())

甚至可能没有一个单独的线程在完成它（甚至可能有许多线程在处理它）。即使有，也没有从 a 到任何引用它的线程的链接。CompletableFuture

因此，您无法通过任何方法来中断任何可能正在运行某些任务以完成它的任务的线程。您必须编写自己的逻辑来跟踪获取对的引用并打算完成它的任何实例。CompletableFutureThreadCompletableFuture

这是一个我认为你可以逃脱的执行类型的例子。

public static void main(String[] args) throws Exception {
    ExecutorService service = Executors.newFixedThreadPool(1);
    CompletableFuture<String> completable = new CompletableFuture<>();
    Future<?> future = service.submit(new Runnable() {
        @Override
        public void run() {
            for (int i = 0; i < 10; i++) {
                if (Thread.interrupted()) {
                    return; // remains uncompleted
                }
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    return; // remains uncompleted
                }
            }
            completable.complete("done");
        }
    });

    Thread.sleep(2000);

    // not atomic across the two
    boolean cancelled = future.cancel(true);
    if (cancelled)
        completable.cancel(true); // may not have been cancelled if execution has already completed
    if (completable.isCancelled()) {
        System.out.println("cancelled");
    } else if (completable.isCompletedExceptionally()) {
        System.out.println("exception");
    } else {
        System.out.println("success");
    }
    service.shutdown();
}

这假定正在执行的任务设置为正确处理中断。

答案 2

这又如何呢？

public static <T> CompletableFuture<T> supplyAsync(final Supplier<T> supplier) {

    final ExecutorService executorService = Executors.newFixedThreadPool(1);

    final CompletableFuture<T> cf = new CompletableFuture<T>() {
        @Override
        public boolean complete(T value) {
            if (isDone()) {
                return false;
            }
            executorService.shutdownNow();
            return super.complete(value);
        }

        @Override
        public boolean completeExceptionally(Throwable ex) {
            if (isDone()) {
                return false;
            }
            executorService.shutdownNow();
            return super.completeExceptionally(ex);
        }
    };

    // submit task
    executorService.submit(() -> {
        try {
            cf.complete(supplier.get());
        } catch (Throwable ex) {
            cf.completeExceptionally(ex);
        }
    });

    return cf;
}

简单测试：

    CompletableFuture<String> cf = supplyAsync(() -> {
        try {
            Thread.sleep(1000L);
        } catch (Exception e) {
            System.out.println("got interrupted");
            return "got interrupted";
        }
        System.out.println("normal complete");
        return "normal complete";
    });

    cf.complete("manual complete");
    System.out.println(cf.get());

我不喜欢每次都必须创建一个Executor服务的想法，但也许你可以找到一种方法来重用ForkJoinPool。