Spring 定时任务

SegmentFault

共 6376字,需浏览 13分钟

 ·

2020-10-02 22:29

作者:techstay
来源:SegmentFault 思否社区




在程序中常常有定时任务的需求,例如每隔一周生成一次报表、每个月月末清空用户积分等等。Spring也提供了相应的支持,我们可以非常方便的按时执行任务。




项目准备


这里我使用Gradle来建立项目,然后在build.gradle中添加下面一行。springVersion的值是目前最新的Spring版本'4.3.7.RELEASE'。使用Maven的话也添加相应的行。spring-context会自动引入spring-core等几个最基本的依赖。


compile group: 'org.springframework', name: 'spring-context', version: springVersion



定时任务属于Spring的核心支持部分,所以我们不需要再添加其他的依赖了。所以定时任务功能既可以在命令行程序中使用,也可以在Java Web程序中使用。当然后者可能使用的更广泛一些(毕竟Web程序需要一直运行的嘛)。


这里我们定义两个任务,后面会让它们可以定时执行。


public interface IService {
    void doService();
}

public class SimpleService implements IService {
    @Override
    public void doService() {
        LocalTime time = LocalTime.now();
        System.out.println("This is a simple service:" + time);
    }
}

public class ExpensiveTaskService implements IService {
    @Override
    public void doService() {
        try {
            Thread.sleep(TimeUnit.SECONDS.toMillis(1));
            LocalTime time = LocalTime.now();
            System.out.println("This is an expensive task:" + time);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}




Spring的任务抽象


TaskExecutor


TaskExecutor接口是任务执行接口,类似于java.util.concurrent.Executor ,该接口只有一个方法execute(Runnable task),用于执行任务。


Spring提供了一组TaskExecutor的实现,详细列表可以看这里34.2.1. TaskExecutor types。要使用它们也很简单,直接注册为Spring Bean,然后注入到程序中即可使用。


TaskScheduler


TaskScheduler接口是定时器的抽象,它的源代码如下。可以看到,该接口包含了一组方法用于指定任务执行的时间。


public interface TaskScheduler {

    ScheduledFuture schedule(Runnable task, Trigger trigger);

    ScheduledFuture schedule(Runnable task, Date startTime);

    ScheduledFuture scheduleAtFixedRate(Runnable task, Date startTime, long period);

    ScheduledFuture scheduleAtFixedRate(Runnable task, long period);

    ScheduledFuture scheduleWithFixedDelay(Runnable task, Date startTime, long delay);

    ScheduledFuture scheduleWithFixedDelay(Runnable task, long delay);

}


Spring提供了两个实现,一是TimerManagerTaskScheduler,会将任务代理到CommonJ TimerManager实例。第二个是ThreadPoolTaskScheduler,当我们不需要管理线程的时候就可以使用该类。而且它还同时实现了TaskExecutor接口,所以一个ThreadPoolTaskScheduler实例即可同时用于执行定时任务。


Trigger


在定时器接口的方法中我们可以发现一个方法接受Trigger接口, 而Trigger也是一个接口,抽象了触发任务执行的触发器。


Trigger接口有两个实现,先说说比较简单的一个PeriodicTrigger。它直接按照给定的时间间隔触发任务执行。更常用的一个触发器是CronTrigger,它使用Cron表达式指定何时执行任务。下面是Spring官方的一个例子。


scheduler.schedule(task, new CronTrigger("0 15 9-17 * * MON-FRI"));



关于Cron表达式的信息可以参考这篇博客QuartZ Cron表达式。另外还有一个可以在线生成Cron表达式的网站:CroMaker,不过好像需要XX才能访问。而且好像Spring不支持第二个星期一这样的定时器设置,所以如果有这样的需求,需要使用Quartz。




配置任务


任务配置既可以使用Java配置,也可以使用XML配置。不管使用哪种方法,首先需要将要执行的方法所在的类配置为Spring Bean。例如下面就用XML配置注册了两个要执行的任务。


    "simpleService" class="yitian.study.service.SimpleService"/>
    "expensiveTaskService"
          class="yitian.study.service.ExpensiveTaskService"/>


Java配置


定时任务


首先看看Java配置。我们需要在配置类上添加@EnableScheduling,如果需要异步的定时任务,还需要添加@Async。


@Configuration
@EnableAsync
@EnableScheduling
public class TaskConfiguration {
}


然后在要执行的方法上添加@Scheduled注解。@Scheduled注解有几个参数,任务会在相应参数的时间下执行。cron参数指定Cron表达式;fixedDelay指定任务执行的间隔,单位是毫秒;initialDelay指定当程序启动后多长时间开始执行第一次任务,单位是毫秒;zone指定任务执行时间所在的时区。下面的例子简单的指定了每隔一秒重复执行一次任务。


public class SimpleService implements IService {
    @Scheduled(fixedDelay = 1000)
    @Override
    public void doService() {
        LocalTime time = LocalTime.now();
        System.out.println("This is a simple service:" + time);
    }
}


异步任务


然后是异步任务,如果任务执行时间比较长的话,我们可以考虑使用异步的任务。当调用异步任务的时候,异步方法直接返回,异步任务会交由相应的任务执行器来执行。在Spring中标记异步方法很简单,直接在方法上使用@Async注解。如果需要指定异步方法使用的执行器,可以向注解传递执行器的名称。异步方法可以返回空值。


@Async("otherExecutor")
void doSomething(String s) {
    // this will be executed asynchronously by "otherExecutor"
}


但是如果异步方法想返回其他值的话,就必须使用Future。不过不仅是java.util.concurrent.Future,异步方法还可以返回Spring的org.springframework.util.concurrent.ListenableFuture和JDK8的java.util.concurrent.CompletableFuture类型。


@Async
Future returnSomething(int i) {
    // this will be executed asynchronously
}


异步方法不仅可以用于定时任务中,在Spring的其他地方也可以使用。例如Spring Data JPA可以使用@Async编写异步的查询方法。


需要注意,异步方法没有对应的XML配置,如果我们想让方法是异步的,只能使用注解。当然也不是完全不行,不过就比较麻烦了,你需要使用AsyncExecutionInterceptor和AOP配合才能达到类似的效果。


如果需要处理异步方法的异常,我们需要实现一个AsyncUncaughtExceptionHandler。下面的异步异常处理器简单的打印异常信息。


public class MyAsyncUncaughtExceptionHandler implements AsyncUncaughtExceptionHandler {
    @Override
    public void handleUncaughtException(Throwable ex, Method method, Object... params) {
        ex.printStackTrace();
    }
}


然后通过实现AsyncConfigurer接口(Java配置方式)或者task:annotation-driven(XML配置方式)的exception-handler元素来配置。


XML配置


Spring提供了task命名空间,让配置定时任务非常简单。


定时器


task:scheduler会注册一个ThreadPoolTaskScheduler 定时器,它只有一个属性线程池大小。默认是1,我们需要根据任务的数量指定一个合适的大小。


    "threadPoolTaskScheduler"
                    pool-size="10"/>


执行器


task:executor会注册一个ThreadPoolTaskExecutor执行器,我们可以使用它的相关属性来配置该执行器。默认情况下执行队列是无限的,可能会导致JVM使用完所有内存。因此我们最好指定一个确定的数值。还有一个rejection-policy属性,指定执行器队列满时的执行策略:默认是AbortPolicy,直接抛出异常;如果当系统忙时丢弃某些任务是可接受的,可以使用DiscardPolicy或DiscardOldestPolicy策略;当系统负载较重时还可以使用CallerRunsPolicy,它不会将任务交给执行器线程,而是让调用者线程来执行该任务。最后一个就是keep-alive属性,也就是超出线程池数量 线程完成任务之后的存活时间,单位是秒。


    "threadPoolTaskExecutor"
                   pool-size="10"
                   queue-capacity="10"/>


执行任务


执行任务很简单,使用指定要执行的Bean和方法即可。

    

        "simpleService" method="doService"
                        cron="*/1 * * * * *"/>
        "expensiveTaskService" method="doService"
                        cron="*/2 * * * * *"/>
    


要设置定时的话,只需要指定相应的属性即可。


"myScheduler">
    "beanA" method="methodA" fixed-delay="5000" initial-delay="1000"/>
    "beanB" method="methodB" fixed-rate="5000"/>
    "beanC" method="methodC" cron="*/5 * * * * MON-FRI"/>


"myScheduler" pool-size="10"/>




Quartz集成


Quartz是一个定时任务的库。Spring也提供了它的支持。Quartz的使用方法请查阅相应文档。这里只简单介绍一下。


Spring的Quartz集成在spring-context-support包中,它还需要Spring事务的支持。因此我们需要下面这样的依赖声明。


 compile group: 'org.springframework', name: 'spring-tx', version: springVersion
    compile group: 'org.springframework', name: 'spring-context-support', version: springVersion
    compile group: 'org.quartz-scheduler', name: 'quartz', version: '2.2.3'



定义任务


Quartz的任务需要继承Quartz的Job接口。所以一个典型的任务可以写成这样。


public class QuartzService implements IService, Job {
    @Override
    public void doService() {
        System.out.println("This is a quartz service");
    }

    @Override
    public void execute(JobExecutionContext context) throws JobExecutionException {
        System.out.println("Do something in execute method of quartz");
    }
}


JobDetailFactoryBean


JobDetailFactoryBean用来定义实现了Job接口的任务。如果需要添加更多信息,可以使用jobDataAsMap属性设置。


"jobDetail"
      class="org.springframework.scheduling.quartz.JobDetailFactoryBean">
    "jobClass" value="yitian.study.service.QuartzService"/>
    "jobDataAsMap">
        
            "timeout" value="10"/>
        

    


MethodInvokingJobDetailFactoryBean


如果任务没有实现Job接口,也可以执行,这时候需要使用MethodInvokingJobDetailFactoryBean。如果存在任务对象,使用targetObject属性,如果有任务类,使用targetClass属性。


"methodJobDetail" class="org.springframework.scheduling.quartz.MethodInvokingJobDetailFactoryBean">
    "targetObject" ref="quartzService"/>
    "targetMethod" value="doService"/>
    "concurrent" value="true"/>


触发器


有了任务,就可以定义触发器了。触发器有两个:SimpleTriggerFactoryBean,以指定的间隔重复执行任务;CronTriggerFactoryBean,以给定的Cron表达式执行任务。Quartz的Cron表达式比Spring 的强大,它支持第几个星期几这样的Cron表达式。


"simpleTrigger" class="org.springframework.scheduling.quartz.SimpleTriggerFactoryBean">
    "jobDetail" ref="jobDetail"/>
    "startDelay" value="0"/>
    "repeatInterval" value="1000"/>

"cronTrigger" class="org.springframework.scheduling.quartz.CronTriggerFactoryBean">
    "jobDetail" ref="methodJobDetail"/>
    "cronExpression" value="*/2 * * * * ?"/>


执行任务


有了触发器,我们就可以执行任务了。注册一个SchedulerFactoryBean,然后将触发器的Bean引用传入即可。


"quartzScheduler" class="org.springframework.scheduling.quartz.SchedulerFactoryBean">
    "triggers">
        
            "cronTrigger"/>
            "simpleTrigger"/>
        

    




本文参考自:

  • Spring官方文档 34. Task Execution and Scheduling,详细网址:
  • https://docs.spring.io/spring-framework/docs/current/spring-framework-reference/#scheduling





点击左下角阅读原文,到 SegmentFault 思否社区 和文章作者展开更多互动和交流。



- END -

浏览 18
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报