74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
|
package cobra
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"github.com/Superdanda/hade/framework/contract"
|
|||
|
|
"github.com/robfig/cron/v3"
|
|||
|
|
"log"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// AddDistributedCronCommand 实现一个分布式定时器
|
|||
|
|
// serviceName 这个服务的唯一名字，不允许带有空格
|
|||
|
|
// spec 具体的执行时间
|
|||
|
|
// cmd 具体的执行命令
|
|||
|
|
// holdTime 表示如果我选择上了，这次选择持续的时间，也就是锁释放的时间
|
|||
|
|
func (c *Command) AddDistributedCronCommand(serviceName string,
|
|||
|
|
spec string,
|
|||
|
|
cmd *Command,
|
|||
|
|
holdTime time.Duration) {
|
|||
|
|
|
|||
|
|
root := c.Root()
|
|||
|
|
|
|||
|
|
// 初始化cron
|
|||
|
|
if root.Cron == nil {
|
|||
|
|
root.Cron = cron.New(cron.WithParser(cron.NewParser(cron.SecondOptional | cron.Minute | cron.Hour | cron.Dom | cron.Month | cron.Dow | cron.Descriptor)))
|
|||
|
|
root.CronSpecs = []CronSpec{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// cron命令的注释，这里注意Type为distributed-cron，ServiceName需要填写
|
|||
|
|
root.CronSpecs = append(root.CronSpecs, CronSpec{
|
|||
|
|
Type: "distributed-cron",
|
|||
|
|
Cmd: cmd,
|
|||
|
|
Spec: spec,
|
|||
|
|
ServiceName: serviceName,
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
appService := root.MustMakeApp()
|
|||
|
|
distributeService := root.GetContainer().MustMake(contract.DistributedKey).(contract.Distributed)
|
|||
|
|
appID := appService.AppId()
|
|||
|
|
|
|||
|
|
// 复制要执行的command为cronCmd，并且设置为rootCmd
|
|||
|
|
var cronCmd Command
|
|||
|
|
ctx := root.Context()
|
|||
|
|
cronCmd = *cmd
|
|||
|
|
cronCmd.args = []string{}
|
|||
|
|
cronCmd.SetParentNull()
|
|||
|
|
|
|||
|
|
// cron增加匿名函数
|
|||
|
|
root.Cron.AddFunc(spec, func() {
|
|||
|
|
// 防止panic
|
|||
|
|
defer func() {
|
|||
|
|
if err := recover(); err != nil {
|
|||
|
|
log.Println(err)
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
// 节点进行选举，返回选举结果
|
|||
|
|
selectedAppID, err := distributeService.Select(serviceName, appID, holdTime)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果自己没有被选择到，直接返回
|
|||
|
|
if selectedAppID != appID {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果自己被选择到了，执行这个定时任务
|
|||
|
|
err = cronCmd.ExecuteContext(ctx)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Println(err)
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|