通过上一节分析源码我们知道,执行任务的是TaskScheduler的实现类,在ScheduledAnnotationBeanPostProcessor#finishRegistration()方法中会设置TaskScheduler。
首先如果注册器ScheduledTaskRegistrar中的scheduler成员为空才会从Spring上下文中加载
我们只需要为ScheduledTaskRegistrar设置我们需要的TaskScheduler就好了。那么问题来了,这个registrar从哪里来?我们怎么才能访问到registrar并设置taskScheduler呢?
从这里我们可以看到,registrar对象在ScheduledAnnotationBeanPostProcessor初始化时创建,到这里断了思路不知如何下手。再仔细翻看finishRegistration()方法有了新的发现
这里获取了SchedulingConfigurer的实现类,并且把执行configureTasks方法传入了ScheduledTaskRegistrar,我们可以从这里做文章。
我们只需要给传入的taskRegistrar设置一个TaskScheduler就好可以了,好了大功告成~