从0实现分布式任务调度系统--实现cron表达式定时任务

在Linux下会经常使用cron表达式完成一些定时任务。下面通过git上的一个项目来达到类似的效果。

获取资源：

https://github.com/gorhill/cronexpr

该包一共支持如下7个时间粒度设置，分别是：秒级，分，小时，日期，月份，星期，年份。

各个支持的符号说明如下：

“*” 字符代表所有可能的值。

“-” 表示指定范围。

“,” 表示列出枚举值。

“/” 被用于指定增量。

“L” 表示last，放在“星期”字段中，“FRIL”可以用来表示当月的最后一个周五；放在日期字段中“L”代表该月的最后一天。

“W”只能用在日期字段，是“Weekday”的缩写，用来描叙最接近指定天的工作日（周一到周五）。例如：在day-of-month字段用“15W”指“最接近这个月第15天的工作日”。

“#” 只能在星期字段中使用，后面必须跟一个 1-5 的数字。例如“MON#2”，表示第2个周一。

其他的细节，请参阅该项目的README文档。

 

下面使用Go语言来实现一个简单的定时调用程序：

package main

import(
   "fmt"
   "github.com/gorhill/cronexpr"
   "time"
)

//定义一个任务
type CronJob struct {
   expr *cronexpr.Expression
   nextTime time.Time //通过expr.Next(now)来获得
}

func main() {
   //需要有一个协程，他定时检查所有的Cron任务，谁过期了就执行谁
   var(
      expr *cronexpr.Expression
      now time.Time
      cronJob *CronJob
      scheduleTable map[string]*CronJob // key:job_name
   )

   scheduleTable = make(map[string]*CronJob)

   //当前时间
   now = time.Now()

   //定义2个cronjob
   expr = cronexpr.MustParse("*/5 * * * * * *")
   cronJob = &CronJob{
      expr:expr,
      nextTime:expr.Next(now),
   }
   //任务注册到调度表
   scheduleTable["5S_JOB"] = cronJob

   expr = cronexpr.MustParse("*/2 * * * * * *")
   cronJob = &CronJob{
      expr:expr,
      nextTime:expr.Next(now),
   }
   //任务注册到调度表
   scheduleTable["2S_JOB"] = cronJob

   //启动一个调度协程
   go func() {
      var(
         jobName string
         cronJob *CronJob
         now time.Time
      )
      //定时检查任务调度表
      for {
         now = time.Now()

         for jobName, cronJob = range scheduleTable{
            //判断是否过期
            if cronJob.nextTime.Before(now) || cronJob.nextTime.Equal(now){
               //启动一个协程，执行这个任务
               go func(jobName string) {
                  fmt.Println("执行：", jobName)
               }(jobName)

               //计算下一次执行时间
               cronJob.nextTime = cronJob.expr.Next(now)
               fmt.Println(jobName, "下次执行时间：", cronJob.nextTime)
            }
         }
         //睡眠100ms
         select {
         case <- time.NewTimer(100 * time.Millisecond).C://将在100毫秒后返回可读

         }
      }
   }()

//调用select阻塞主线程，防止进程退出
   select {

   }
}

执行效果如下：

通过类似的方法就可以很方便地实现各个时间粒度的定时任务调度。