62 lines
1.2 KiB
Go
62 lines
1.2 KiB
Go
|
package cobra
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/Superdanda/hade/framework"
|
||
|
|
"github.com/robfig/cron/v3"
|
||
|
|
"log"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (c *Command) SetContainer(container framework.Container) {
|
||
|
|
c.container = container
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Command) GetContainer() framework.Container {
|
||
|
|
return c.Root().container
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Command) SetParentNull() {
|
||
|
|
c.parent = nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// AddCronCommand 是用来创建一个Cron任务的
|
||
|
|
func (c *Command) AddCronCommand(spec string, cmd *Command) {
|
||
|
|
root := c.Root()
|
||
|
|
|
||
|
|
if root.Cron == nil {
|
||
|
|
// 初始化cron
|
||
|
|
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{}
|
||
|
|
}
|
||
|
|
|
||
|
|
root.CronSpecs = append(root.CronSpecs, CronSpec{
|
||
|
|
Type: "normal-cron",
|
||
|
|
Spec: spec,
|
||
|
|
Cmd: cmd,
|
||
|
|
})
|
||
|
|
|
||
|
|
var cronCmd Command
|
||
|
|
ctx := root.Context()
|
||
|
|
cronCmd = *cmd
|
||
|
|
cronCmd.args = []string{}
|
||
|
|
cronCmd.SetParentNull()
|
||
|
|
cronCmd.SetContainer(root.GetContainer())
|
||
|
|
root.Cron.AddFunc(spec, func() {
|
||
|
|
defer func() {
|
||
|
|
if err := recover(); err != nil {
|
||
|
|
log.Println(err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
err := cronCmd.ExecuteContext(ctx)
|
||
|
|
if err != nil {
|
||
|
|
log.Println(err)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
type CronSpec struct {
|
||
|
|
Type string
|
||
|
|
Cmd *Command
|
||
|
|
Spec string
|
||
|
|
ServiceName string
|
||
|
|
}
|