mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-05-06 01:55:58 +08:00
cobra
This commit is contained in:
@@ -10,8 +10,13 @@ func NewCronTaskCmd() *CronTaskCmd {
|
||||
return &CronTaskCmd{NewRootCmd()}
|
||||
}
|
||||
|
||||
func (c *CronTaskCmd) AddRunE(f func() error) {
|
||||
func (c *CronTaskCmd) addRunE(f func() error) {
|
||||
c.Command.RunE = func(cmd *cobra.Command, args []string) error {
|
||||
return f()
|
||||
}
|
||||
}
|
||||
|
||||
func (c *CronTaskCmd) Exec(f func() error) error {
|
||||
c.addRunE(f)
|
||||
return c.Execute()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user