聊聊dkron的Scheduler

本文主要研究一下dkron的Scheduler

Scheduler

// Scheduler represents a dkron scheduler instance, it stores the cron engine
// and the related parameters.
type Scheduler struct {
    Cron        *cron.Cron
    Started     bool
    EntryJobMap sync.Map
}

// NewScheduler creates a new Scheduler instance
func NewScheduler() *Scheduler {
    schedulerStarted.Set(0)
    return &Scheduler{
        Cron:        nil,
        Started:     false,
        EntryJobMap: sync.Map{},
    }
}

// Start the cron scheduler, adding its corresponding jobs and
// executing them on time.
func (s *Scheduler) Start(jobs []*Job, agent *Agent) error {
    s.Cron = cron.New(cron.WithParser(extcron.NewParser()))

    metrics.IncrCounter([]string{"scheduler", "start"}, 1)
    for _, job := range jobs {
        job.Agent = agent
        s.AddJob(job)
    }
    s.Cron.Start()
    s.Started = true
    schedulerStarted.Set(1)

    return nil
}

// Stop stops the scheduler effectively not running any job.
func (s *Scheduler) Stop() {
    if s.Started {
        log.Debug("scheduler: Stopping scheduler")
        s.Cron.Stop()
        s.Started = false
        // Keep Cron exists and let the jobs which have been scheduled can continue to finish,
        // even the node's leadership will be revoked.
        // Ignore the running jobs and make s.Cron to nil may cause whole process crashed.
        //s.Cron = nil

        // expvars
        cronInspect.Do(func(kv expvar.KeyValue) {
            kv.Value = nil
        })
    }
    schedulerStarted.Set(0)
}

// Restart the scheduler
func (s *Scheduler) Restart(jobs []*Job, agent *Agent) {
    s.Stop()
    s.ClearCron()
    s.Start(jobs, agent)
}

// Clear cron separately, this can only be called when agent will be stop.
func (s *Scheduler) ClearCron() {
    s.Cron = nil
}
Scheduler定义了Cron、Started、EntryJobMap属性;NewScheduler方法创建默认的Scheduler;Start方法遍历jobs,挨个设置job.Agent,然后添加到Scheduler中,之后执行Scheduler.Cron.Start();Stop方法执行Scheduler.Cron.Stop();Restart方法执行Stop、ClearCron、Start方法;ClearCron设置Cron为nil

AddJob

// AddJob Adds a job to the cron scheduler
func (s *Scheduler) AddJob(job *Job) error {
    // Check if the job is already set and remove it if exists
    if _, ok := s.EntryJobMap.Load(job.Name); ok {
        s.RemoveJob(job)
    }

    if job.Disabled || job.ParentJob != "" {
        return nil
    }

    log.WithFields(logrus.Fields{
        "job": job.Name,
    }).Debug("scheduler: Adding job to cron")

    // If Timezone is set on the job, and not explicitly in its schedule,
    // AND its not a descriptor (that don't support timezones), add the
    // timezone to the schedule so robfig/cron knows about it.
    schedule := job.Schedule
    if job.Timezone != "" &&
        !strings.HasPrefix(schedule, "@") &&
        !strings.HasPrefix(schedule, "TZ=") &&
        !strings.HasPrefix(schedule, "CRON_TZ=") {
        schedule = "CRON_TZ=" + job.Timezone + " " + schedule
    }

    id, err := s.Cron.AddJob(schedule, job)
    if err != nil {
        return err
    }
    s.EntryJobMap.Store(job.Name, id)

    cronInspect.Set(job.Name, job)
    metrics.IncrCounterWithLabels([]string{"scheduler", "job_add"}, 1, []metrics.Label{{Name: "job", Value: job.Name}})

    return nil
}
AddJob方法先移除EntryJobMap中的同名job,然后执行Cron.AddJob(schedule, job),最后存储到EntryJobMap

RemoveJob

// RemoveJob removes a job from the cron scheduler
func (s *Scheduler) RemoveJob(job *Job) {
    log.WithFields(logrus.Fields{
        "job": job.Name,
    }).Debug("scheduler: Removing job from cron")
    if v, ok := s.EntryJobMap.Load(job.Name); ok {
        s.Cron.Remove(v.(cron.EntryID))
        s.EntryJobMap.Delete(job.Name)

        cronInspect.Delete(job.Name)
        metrics.IncrCounterWithLabels([]string{"scheduler", "job_delete"}, 1, []metrics.Label{{Name: "job", Value: job.Name}})
    }
}
RemoveJob方法先从EntryJobMap移除同名job,然后执行cronInspect.Delete(job.Name)

小结

dkron的Scheduler定义了Cron、Started、EntryJobMap属性;NewScheduler方法创建默认的Scheduler;它提供了Start、Stop、Restart、ClearCron、AddJob、RemoveJob方法。

doc


code-craft
spring boot , docker and so on 欢迎关注微信公众号: geek_luandun

当一个代码的工匠回首往事时,不因虚度年华而悔恨,也不因碌碌无为而羞愧,这样,当他老的时候,可以很...

11.8k 声望
2k 粉丝
0 条评论
推荐阅读
2022年终总结
最近两年开始陷入颓废中,博客也写的越来越少了。究其原因,主要还是陷入了职业倦怠期,最近一次跳槽感觉颇为失败,但是碍于给的薪资高,为了五斗米折腰,又加上最近行情不好,想要往外跳也跳不了,就这样子一直...

codecraft阅读 724

Golang 中 []byte 与 string 转换
string 类型和 []byte 类型是我们编程时最常使用到的数据结构。本文将探讨两者之间的转换方式,通过分析它们之间的内在联系来拨开迷雾。

机器铃砍菜刀22阅读 55.2k评论 1

年度最佳【golang】map详解
这篇文章主要讲 map 的赋值、删除、查询、扩容的具体执行过程,仍然是从底层的角度展开。结合源码,看完本文一定会彻底明白 map 底层原理。

去去100214阅读 11.1k评论 2

年度最佳【golang】GMP调度详解
Golang最大的特色可以说是协程(goroutine)了, 协程让本来很复杂的异步编程变得简单, 让程序员不再需要面对回调地狱, 虽然现在引入了协程的语言越来越多, 但go中的协程仍然是实现的是最彻底的. 这篇文章将通过分析...

去去100213阅读 11.3k评论 4

【已结束】SegmentFault 思否技术征文丨浅谈 Go 语言框架
亲爱的开发者们:我们的 11 月技术征文如期而来,这次主题围绕 「 Go 」 语言,欢迎大家来参与分享~征文时间11 月 4 日 - 11 月 27 日 23:5911 月 28 日 18:00 前发布中奖名单参与条件新老思否作者均可参加征文...

SegmentFault思否11阅读 4.7k评论 11

封面图
【Go微服务】开发gRPC总共分三步
之前我也有写过RPC相关的文章:《 Go RPC入门指南:RPC的使用边界在哪里?如何实现跨语言调用?》,详细介绍了RPC是什么,使用边界在哪里?并且用Go和php举例,实现了跨语言调用。不了解RPC的同学建议先读这篇文...

王中阳Go8阅读 3.7k评论 6

封面图
【golang】sync.WaitGroup详解
上一期中,我们介绍了 sync.Once 如何保障 exactly once 语义,本期文章我们介绍 package sync 下的另一个工具类:sync.WaitGroup。

去去100213阅读 30.2k评论 2

当一个代码的工匠回首往事时,不因虚度年华而悔恨,也不因碌碌无为而羞愧,这样,当他老的时候,可以很...

11.8k 声望
2k 粉丝
宣传栏