如何创建可条件执行的自定义@MyScheduler注解替代Java的@Schedule

方法一：使用AOP切面统一拦截（推荐，无侵入） #

无需修改任何现有带@Schedule注解的方法，只需新增一个切面类，拦截所有被@Schedule标记的方法，在执行前判断isLeader状态，只有当状态为true时才允许方法执行。

示例代码（以Spring AOP为例）：

@Component
@Aspect
public class LeaderGuardScheduleAspect {
    // 假设全局isLeader状态存储在LeaderContext这个Bean中
    @Autowired
    private LeaderContext leaderContext;

    // 拦截所有带有@Schedule注解的方法
    @Around("@annotation(com.yourpackage.Schedule)")
    public Object interceptScheduledTask(ProceedingJoinPoint joinPoint) throws Throwable {
        if (leaderContext.isLeader()) {
            // 是leader，执行原方法
            return joinPoint.proceed();
        }
        // 非leader，跳过执行
        return null;
    }
}

优势 #

• 对现有代码零侵入，不用逐个修改定时任务方法
• 统一控制所有定时任务的执行条件，后续调整规则只需修改切面逻辑

方法二：自定义注解+切面（替代方案） #

如果你更倾向于用自定义注解标记需要受leader控制的任务，可以这样实现：

1. 定义自定义注解@LeaderControlledSchedule，复用原@Schedule的属性：

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
// 继承原@Schedule的属性，确保调度框架能识别调度规则
@Schedule
public @interface LeaderControlledSchedule {
    String minute() default "*";
    String hour() default "*";
    boolean persistent() default true;
    String info() default "";
}

2. 编写切面拦截该自定义注解：

@Component
@Aspect
public class LeaderControlledScheduleAspect {
    @Autowired
    private LeaderContext leaderContext;

    @Around("@annotation(com.yourpackage.LeaderControlledSchedule)")
    public Object interceptLeaderControlledTask(ProceedingJoinPoint joinPoint) throws Throwable {
        return leaderContext.isLeader() ? joinPoint.proceed() : null;
    }
}

优势 #

• 语义更清晰，一眼就能看出哪些任务受leader控制
• 可以灵活选择需要控制的任务，不需要全局拦截所有@Schedule任务

方法三：动态控制任务注册（适合底层调度框架可定制场景） #

如果你的调度框架支持动态注册/取消任务，可以在应用启动或isLeader状态变化时，动态添加或移除定时任务：

示例代码（Spring环境下）：

@Component
public class DynamicScheduleController {
    @Autowired
    private LeaderContext leaderContext;
    @Autowired
    private TaskScheduler taskScheduler;
    @Autowired
    private YourTaskBean taskBean;

    private ScheduledFuture<?> scheduledTask;

    // 监听leader状态变化事件（假设你有事件机制）
    @EventListener
    public void onLeaderStatusChanged(LeaderStatusChangedEvent event) {
        updateScheduleTask();
    }

    // 应用启动时初始化任务
    @PostConstruct
    public void initScheduleTask() {
        updateScheduleTask();
    }

    private void updateScheduleTask() {
        // 先取消已存在的任务
        if (scheduledTask != null && !scheduledTask.isCancelled()) {
            scheduledTask.cancel(false);
        }

        // 如果是leader，注册5分钟执行一次的任务
        if (leaderContext.isLeader()) {
            scheduledTask = taskScheduler.scheduleAtFixedRate(
                taskBean::newMethod,
                Duration.ofMinutes(5)
            );
        }
    }
}

注意事项 #

• 该方法需要针对每个任务单独处理，适合任务数量较少的场景
• 需要确保isLeader状态变化时有通知机制（比如事件、定时轮询），否则无法动态调整任务状态

内容的提问来源于stack exchange，提问作者Kavindu Nilshan