Java时间轮调度算法的代码实现

public class TimeSlot {
    private List<Runnable> tasks = new ArrayList<>();

    public void addTask(Runnable task) {
        tasks.add(task);
    }

    public List<Runnable> getTasks() {
        return tasks;
    }

    public void clearTasks() {
        tasks.clear();
    }
}

public class TimeWheel {
    private TimeSlot[] slots;
    private int currentIndex = 0;
    private final int slotCount;
    private final long tickDuration;

    public TimeWheel(int slotCount, long tickDuration) {
        this.slotCount = slotCount;
        this.tickDuration = tickDuration;
        this.slots = new TimeSlot[slotCount];
        for (int i = 0; i < slotCount; i++) {
            slots[i] = new TimeSlot();
        }
    }

    public void addTask(Runnable task, long delay) {
        int slotIndex = (int) ((currentIndex + delay / tickDuration) % slotCount);
        slots[slotIndex].addTask(task);
    }

    public void tick() {
        TimeSlot slot = slots[currentIndex];
        for (Runnable task : slot.getTasks()) {
            task.run();
        }
        slot.clearTasks();
        currentIndex = (currentIndex + 1) % slotCount;
    }
}

public class TimeWheelExample {
    public static void main(String[] args) throws InterruptedException {
        TimeWheel timeWheel = new TimeWheel(10, 1000); // 10 个槽，每个槽间隔 1 秒

        timeWheel.addTask(() -> System.out.println("Task 1 executed!"), 3000);
        
        timeWheel.addTask(() -> System.out.println("Task 2 executed!"), 5000);
      	timeWheel.addTask(() -> System.out.println("Task 3 executed!"), 4000);
        while (true) {
            timeWheel.tick();
            Thread.sleep(1000); // 每秒执行一次 tick
        }
    }
}