温馨提示×

java时间轮如何实现

小亿
81
2024-12-05 11:01:40
栏目: 编程语言

Java中的时间轮算法通常用于实现高效的定时任务调度。它通过将时间分成若干个时间间隔(称为“刻度”),并将任务存储在这些刻度上,从而实现快速查找和调度任务。以下是一个简单的Java时间轮实现示例:

import java.util.ArrayList;
import java.util.List;

public class TimingWheel {
    private final int wheelSize;
    private final long interval;
    private List<Task> tasks;
    private int currentIndex;

    public TimingWheel(int wheelSize, long interval) {
        this.wheelSize = wheelSize;
        this.interval = interval;
        this.tasks = new ArrayList<>(wheelSize);
        this.currentIndex = 0;
    }

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

    public void tick() {
        currentIndex = (currentIndex + 1) % wheelSize;
        List<Task> expiredTasks = tasks.subList(currentIndex, wheelSize);
        for (Task task : expiredTasks) {
            task.run();
        }
    }

    public static void main(String[] args) {
        TimingWheel timingWheel = new TimingWheel(60, 1000); // 每秒一个刻度,每个刻度1000毫秒

        for (int i = 0; i < 10; i++) {
            final int taskId = i;
            timingWheel.addTask(new Task() {
                @Override
                public void run() {
                    System.out.println("Task " + taskId + " executed at " + System.currentTimeMillis());
                }
            });
        }

        while (true) {
            timingWheel.tick();
            try {
                Thread.sleep(500); // 每500毫秒执行一次tick方法
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}

abstract class Task {
    abstract void run();
}

这个示例中,我们创建了一个名为TimingWheel的类,它包含一个固定大小的轮询数组tasks,以及一个表示当前刻度的索引currentIndex。我们还定义了一个名为Task的抽象类,用于表示要执行的任务。

addTask方法用于向轮询数组中添加任务,tick方法用于处理过期任务。在main方法中,我们创建了一个TimingWheel实例,并向其中添加了10个任务。然后,我们使用一个无限循环来定期执行tick方法,以便在指定的时间间隔内执行这些任务。

请注意,这个示例仅用于演示目的,实际应用中可能需要根据具体需求进行调整。例如,你可能需要实现一个更复杂的任务调度策略,或者使用多个时间轮来处理不同时间粒度的任务。

0