RetryExecutor :如何等待所有任务完成？

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class RetryExecutorExample {
    public static void main(String[] args) throws InterruptedException {
        int taskCount = 5; // 任务数量

        // 创建RetryExecutor
        RetryExecutor retryExecutor = new RetryExecutor();

        // 创建CountDownLatch，初始计数为任务数量
        CountDownLatch latch = new CountDownLatch(taskCount);

        // 创建线程池
        ExecutorService executorService = Executors.newFixedThreadPool(taskCount);

        for (int i = 0; i < taskCount; i++) {
            final int taskId = i;
            executorService.submit(() -> {
                try {
                    // 执行任务
                    retryExecutor.execute(() -> {
                        // 任务逻辑
                        System.out.println("Task " + taskId + " is executing");
                        // 模拟任务执行失败的情况
                        if (taskId % 2 == 0) {
                            throw new RuntimeException("Task " + taskId + " failed");
                        }
                    });
                } catch (Exception e) {
                    e.printStackTrace();
                } finally {
                    // 任务完成，计数减一
                    latch.countDown();
                }
            });
        }

        // 等待所有任务完成
        latch.await();

        // 关闭线程池
        executorService.shutdown();

        System.out.println("All tasks completed");
    }
}