Golang Cron 的使用

关于动态添加cron 命令执行可以参考api

• cron.Remove
• cron.AddFunc
• cron.Run

即可

见代码

package schedule

import (
    "encoding/json"
    "fmt"
    "github.com/blinkbean/dingtalk"
    "github.com/go-redis/redis"
    "github.com/robfig/cron/v3"
    "gocms/pkg/config"
    "gocms/pkg/logger"
    "time"
)

var (
    SCHEDULE_KEY = "SCHEDULE:JOBS"

    STATUS_RUNING   = 1
    STATUS_STARTING = 2
    STATUS_STOPPING = 3
    STATUS_STOPPED  = 4
)

type Process struct {
    Name    string       `json:"name"`
    Content string       `json:"content"`
    Spec    string       `json:"spec"`
    TimeAt  string       `json:"time_at"`
    EntryId cron.EntryID `json:"entry_id"`
    Status  int          `json:"status"`
    StopAt  string       `json:"stop_at"`
}

type Schedule struct {
    cron    *cron.Cron
    client  *redis.Client
    Content string `json:"content"`
}

// InitSchedule 初始化队列管理器
func InitSchedule() {
    var schedule Schedule
    schedule.cron = cron.New()
    schedule.client = config.Redis

    schedule.RunJobs()
    schedule.cron.AddFunc("* * * * *", schedule.ManangeJob)
    schedule.cron.Run()
}

func DispatchTestProcess() {
    process := Process{
        Name:    "每3分钟运行一次",
        Content: "每3分钟运行一次",
        Spec:    "*/3 * * * *",
        Status:  STATUS_STARTING,
    }
    Dispatch(process)

    process = Process{
        Name:    "每2分钟运行一次",
        Content: "每2分钟运行一次",
        Spec:    "*/2 * * * *",
        Status:  STATUS_STARTING,
    }

    Dispatch(process)

    process = Process{
        Name:    "每4分钟运行一次",
        Content: "每4分钟运行一次",
        Spec:    "*/4 * * * *",
        Status:  STATUS_STARTING,
    }

    Dispatch(process)

    process = Process{
        Name:    "每1分钟运行一次",
        Content: "每1分钟运行一次",
        Spec:    "* * * * *",
        Status:  STATUS_STARTING,
    }

    Dispatch(process)
}

// ManangeJob 维护和管理每个任务的执行停止删除
func (that Schedule) ManangeJob() {
    that.RunJobs()
}

// GetJobs 从 redis 获取当前已有的任务
func (that Schedule) GetJobs() []Process {
    var jobs []Process
    if r, err := that.client.HGetAll(SCHEDULE_KEY).Result(); err == nil {
        for _, item := range r {
            var process Process
            json.Unmarshal([]byte(item), &process)
            jobs = append(jobs, process)
        }
    }

    return jobs
}

// RunJobs 从 redis 执行当前已有的任务
func (that Schedule) RunJobs() {
    jobs := that.GetJobs()
    timeAt := time.Now().Format("2006-01-02 15:04:05")
    for _, item := range jobs {
        switch item.Status {
        case STATUS_STOPPING:
            that.cron.Remove(item.EntryId)
            item.Status = STATUS_STOPPED
            item.StopAt = timeAt
        case STATUS_RUNING:
            logger.Info(item.Name, "正常运行中... | Time: "+timeAt)
        case STATUS_STARTING:
            entry_id := that.StartJob(item)
            logger.Info(item.Name, "启动完成... | Time: "+timeAt)
            item.Status = STATUS_RUNING
            item.EntryId = entry_id
            item.TimeAt = timeAt
        case STATUS_STOPPED:
            logger.Info(item.Name, "检测到已停止... | Time: "+timeAt)
        default:
            logger.Error(item.Name, "检测异常... | Time: "+timeAt)
        }

        // 同步到redis中
        Dispatch(item)
    }
}

// 添加一个运行队列
func (that Schedule) StartJob(process Process) cron.EntryID {
    that.Content = process.Content
    entry_id, _ := that.cron.AddFunc(process.Spec, that.DingTalk)

    return entry_id
}

// 用来执行队列做的事情
func (that Schedule) DingTalk() {
    var dingToken = []string{"b960b8c2240b7d0f05b1ffbf26b4a7807efa2fb22603127dcbdc618ea48607ea"}
    cli := dingtalk.InitDingTalk(dingToken, "任务")
    fmt.Println(that.Content + " | TimeAt:" + time.Now().Format("2006-01-02 15:04:05"))
    cli.SendTextMessage(that.Content + " | TimeAt:" + time.Now().Format("2006-01-02 15:04:05"))
}

// 分发任务
func Dispatch(process Process) {
    r, _ := json.Marshal(process)
    fmt.Println(string(r))
    if _, err := config.Redis.HSet(SCHEDULE_KEY, process.Name, string(r)).Result(); err != nil {
        logger.Error("REDIS ERROR", err.Error())
    }
}

分析

这里用到的包为

• github.com/robfig/cron/v3
• github.com/go-redis/redis
• github.com/blinkbean/dingtalk

可以动态通过修改status 值，使其终止队列job的运行和启动

演示效果

本文由 邓尘锋 创作，采用 知识共享署名4.0 国际许可协议进行许可
本站文章除注明转载/出处外，均为本站原创或翻译，转载前请务必署名
最后编辑时间为: Apr 2, 2021 at 01:18 pm