golang实现定时任务调度
package main
import (
"fmt"
"time"
"github.com/gorhill/cronexpr"
)
// 任务结构体--封装
type CronJob struct {
expr *cronexpr.Expression // 时间表达式
nextTime time.Time // 下次调度时间
}
// 声明一些全局变量
var (
cronJob *CronJob // job名字
expr *cronexpr.Expression // 时间表达式
now time.Time // 当前时间
scheduleTable map[string]*CronJob //key：任务名字, 调度表 存放所有定时任务
)
func main() {
// 需要有一个调度协程,让他定时检测所有的Corn任务,谁过期了就执行谁
// 当前时间
now = time.Now()
// 创建内存,map不可以直接使用,这里不给分配内存
scheduleTable = make(map[string]*CronJob)
// 定义两个cornjob: 每5s执行一次; 每57执行一次
expr = cronexpr.MustParse("*/5 * * * * * *")
// 将调度器CronJob初始化且赋值给cronJob
cronJob = &CronJob{
expr: expr,
nextTime: expr.Next(now),
}
// 任务注册到调度表
scheduleTable["job1"] = cronJob
// 定义两个cornjob: 每5s执行一次; 每57执行一次
expr = cronexpr.MustParse("*/5 * * * * * *")
// 将调度器CronJob初始化且赋值给cronJob
cronJob = &CronJob{
expr: expr,
nextTime: expr.Next(now),
}
// 任务注册到调度表
scheduleTable["job2"] = 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)
}
}
// 睡眠100 毫秒（不让它占用过多cpu）
select {
case <-time.NewTimer(100 * time.Millisecond).C: //将在100 毫秒可读，返回
}
// 简单的 time.Sleep(100 *time.Second)
}
}()
time.Sleep(100 * time.Second)
}