Java实现简单任务调度器

实现一个简单的任务调度器

在现代软件开发中，任务调度器是一个常见的功能模块。它用于在特定的时间或周期性地执行任务。本文将介绍如何在Java中实现一个简单的任务调度器。

功能需求

1. 任务注册：允许用户注册需要调度的任务。
2. 定时执行：支持任务在指定的时间间隔内执行。
3. 任务取消：允许用户取消已经注册的任务。
4. 任务状态查询：能够查询任务的当前状态（运行中、已取消等）。

方案设计

为了实现上述功能，我们可以使用Java的ScheduledExecutorService。这是一个强大的工具，提供了灵活的任务调度能力。

1. 任务注册与执行：使用ScheduledExecutorService的scheduleAtFixedRate方法来注册和执行任务。这个方法允许我们以固定的速率执行任务。

2. 任务取消：通过保存ScheduledFuture对象，用户可以调用其cancel方法来取消任务。

3. 任务状态查询：可以通过检查ScheduledFuture对象的状态来确定任务是否已经完成或被取消。

代码实现

以下是一个简单的实现示例：

import java.util.concurrent.*;

public class TaskScheduler {

    private final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);

    public ScheduledFuture<?> scheduleTask(Runnable task, long initialDelay, long period, TimeUnit unit) {
        return scheduler.scheduleAtFixedRate(task, initialDelay, period, unit);
    }

    public void cancelTask(ScheduledFuture<?> future) {
        future.cancel(false);
    }

    public boolean isTaskCancelled(ScheduledFuture<?> future) {
        return future.isCancelled();
    }

    public boolean isTaskDone(ScheduledFuture<?> future) {
        return future.isDone();
    }

    public void shutdown() {
        scheduler.shutdown();
    }

    public static void main(String[] args) {
        TaskScheduler taskScheduler = new TaskScheduler();

        Runnable task = () -> System.out.println("Task executed at: " + System.currentTimeMillis());

        ScheduledFuture<?> future = taskScheduler.scheduleTask(task, 0, 5, TimeUnit.SECONDS);

        // Simulate some delay before cancelling the task
        try {
            Thread.sleep(20000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        taskScheduler.cancelTask(future);

        System.out.println("Task cancelled: " + taskScheduler.isTaskCancelled(future));
        System.out.println("Task done: " + taskScheduler.isTaskDone(future));

        taskScheduler.shutdown();
    }
}

总结

通过使用Java的ScheduledExecutorService，我们可以轻松实现一个简单而有效的任务调度器。这个实现不仅满足了基本的功能需求，还可以根据实际需要进行扩展，例如增加任务优先级、任务依赖等功能。希望这篇文章能为你在Java开发中实现任务调度器提供一些启发。