等待取消的未来真正完成

Question

我有一个 SwingWorker，它调用一些不检查线程中断的代码。调用 worker.cancel(true) 后，worker.get() 方法将立即抛出 CancellationException （正如预期的那样） 。但是，由于后台任务的代码从不检查其线程是否被中断，因此它会愉快地继续执行。

是否有等待后台任务实际完成的标准方法？我希望显示“正在取消...”消息或类似的消息并阻止，直到任务终止。 （我确信如果有必要，我总是可以在工人类中使用标志来完成此任务，只需寻找任何其他解决方案即可。）

Answer 1

我对此进行了一些尝试，这就是我的想法。我使用 CountDownLatch 并基本上将其 await() 方法公开为我的 SwingWorker 对象上的方法。仍在寻找更好的解决方案。

final class Worker extends SwingWorker<Void, Void> {

    private final CountDownLatch actuallyFinishedLatch = new CountDownLatch(1);

    @Override
    protected Void doInBackground() throws Exception {
        try {
            System.out.println("Long Task Started");

            /* Simulate long running method */
            for (int i = 0; i < 1000000000; i++) {
                double d = Math.sqrt(i);
            }

            return null;
        } finally {
            actuallyFinishedLatch.countDown();
        }
    }

    public void awaitActualCompletion() throws InterruptedException {
        actuallyFinishedLatch.await();
    }

    public static void main(String[] args) {
        Worker worker = new Worker();
        worker.execute();

        try {
            TimeUnit.SECONDS.sleep(1);
        } catch (InterruptedException e) {

        }

        System.out.println("Cancelling");
        worker.cancel(true);

        try {
            worker.get();
        } catch (CancellationException e) {
            System.out.println("CancellationException properly thrown");
        } catch (InterruptedException e) {

        } catch (ExecutionException e) {

        }

        System.out.println("Awaiting Actual Completion");
        try {
            worker.awaitActualCompletion();
            System.out.println("Done");
        } catch (InterruptedException e) {

        }
    }

}

Answer 2

最接近标准或现成方法的是 progress 属性和/或由 SwingWorker 提供的发布/处理方法对。您可以在方法末尾将其设置为“我已完成”值，以指示后台工作已完成。等待 Swing Worker 的线程可以发出“正在取消...”消息，并定期检查进度以查看是否已完成。如果等待线程是 swing EDT，那么您将需要使用计时器定期检查进度属性并在完成后清除取消消息。

下面是一些运行顽固后台线程的示例代码，该线程被取消，然后等待进度达到 100。

@Test
public void testSwingWorker()
{
    SwingWorker worker = new SwingWorker() {

        @Override
        protected void process(List chunks)
        {
            for (Object chunk : chunks)
            {
                System.out.println("process: "+chunk.toString());
            }
        }

        @Override
        protected void done()
        {
            System.out.println("done");
        }

        @Override
        protected Object doInBackground() throws Exception
        {
            // simulate long running method
            for (int i=0; i<1000000000; i++)
            {
                double d = Math.sqrt(i);
            }
            System.err.println("finished");
            publish("finished");
            setProgress(100);
            return null;
        }
    };
    Thread t = new Thread(worker);
    t.start();

    try
    {
        worker.get(1, TimeUnit.SECONDS);
    }
    catch (InterruptedException e)        {
    }
    catch (ExecutionException e)        {
    }
    catch (TimeoutException e)        {
    }

    worker.cancel(true);

    // now wait for finish.
    int progress = 0;
    do
    {
        try
        {
            Thread.sleep(1000);
        }
        catch (InterruptedException e)
        {
        }
        progress = worker.getProgress();
        System.out.println(String.format("progress %d", progress));
    }
    while (progress<100);
}

另一种方法是使用 publish\process 方法对推送一个特殊值，该值指示后台线程已完成进入 EDT。然后，SwingWorker 中的 process 重写方法会获取此特殊值并隐藏“正在取消...”消息。这样做的优点是不需要轮询或计时器。示例代码显示，尽管任务取消后立即调用 done，但即使任务取消，发布/处理方法对仍然可以工作。

Answer 3

受到 Paul Blessing 解决方案的启发，我对其进行了一些改进，使其成为一个类，您可以子类化以获得所需的功能：

class AwaitingWorker<T,V> extends SwingWorker<T, V> {

    private final CountDownLatch actuallyFinishedLatch = new CountDownLatch(1);

    /**
     * Override this to do something useful
     */
    protected abstract void performOperation();

    @Override
    protected final T doInBackground() throws Exception {
        try {
            return performOperation();
        } finally {
            actuallyFinishedLatch.countDown();
        }
    }

    public void awaitActualCompletion() throws InterruptedException {
        actuallyFinishedLatch.await();
    }

}