45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
package console
|
|
|
|
import (
|
|
"bbs/app/console/command/demo"
|
|
"github.com/Superdanda/hade/framework"
|
|
"github.com/Superdanda/hade/framework/cobra"
|
|
"github.com/Superdanda/hade/framework/command"
|
|
"time"
|
|
)
|
|
|
|
func RunCommand(container framework.Container) error {
|
|
// 根Command
|
|
var rootCmd = &cobra.Command{
|
|
// 定义根命令的关键字
|
|
Use: "hade",
|
|
// 简短介绍
|
|
Short: "hade 命令",
|
|
// 根命令的详细介绍
|
|
Long: "hade 框架提供的命令行工具，使用这个命令行工具能很方便执行框架自带命令，也能很方便编写业务命令",
|
|
// 根命令的执行函数
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
cmd.InitDefaultHelpFlag()
|
|
return cmd.Help()
|
|
},
|
|
// 不需要出现cobra默认的completion子命令
|
|
CompletionOptions: cobra.CompletionOptions{DisableDefaultCmd: true},
|
|
}
|
|
// 为根Command设置服务容器
|
|
rootCmd.SetContainer(container)
|
|
// 绑定框架的命令
|
|
command.AddKernelCommands(rootCmd)
|
|
// 绑定业务的命令
|
|
AddAppCommand(rootCmd)
|
|
|
|
return rootCmd.Execute()
|
|
}
|
|
|
|
// 绑定业务的命令
|
|
func AddAppCommand(rootCmd *cobra.Command) {
|
|
// demo 例子
|
|
rootCmd.AddDistributedCronCommand("foo_func_for_test",
|
|
"* * * * * *",
|
|
demo.InitFoo(), 2*time.Second)
|
|
}
|