2023-03-08 11:45:54 +08:00
|
|
|
package cmd
|
|
|
|
|
|
|
|
|
|
import "github.com/spf13/cobra"
|
|
|
|
|
|
|
|
|
|
type CronTaskCmd struct {
|
|
|
|
|
*RootCmd
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func NewCronTaskCmd() *CronTaskCmd {
|
2023-03-13 12:34:56 +08:00
|
|
|
return &CronTaskCmd{NewRootCmd("cronTask")}
|
2023-03-08 11:45:54 +08:00
|
|
|
}
|
|
|
|
|
|
2023-03-08 13:34:12 +08:00
|
|
|
func (c *CronTaskCmd) addRunE(f func() error) {
|
2023-03-08 11:45:54 +08:00
|
|
|
c.Command.RunE = func(cmd *cobra.Command, args []string) error {
|
|
|
|
|
return f()
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-03-08 13:34:12 +08:00
|
|
|
|
|
|
|
|
func (c *CronTaskCmd) Exec(f func() error) error {
|
|
|
|
|
c.addRunE(f)
|
|
|
|
|
return c.Execute()
|
|
|
|
|
}
|
