修改版本打印

This commit is contained in:
bjdgyc 2023-04-26 21:05:18 +08:00
parent 165d4ef8a0
commit a9e798f203
1 changed files with 12 additions and 2 deletions

View File

@ -69,6 +69,11 @@ func initCmd() {
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
// fmt.Println("cmd", cmd.Use, args) // fmt.Println("cmd", cmd.Use, args)
runSrv = true runSrv = true
if rev {
printVersion()
os.Exit(0)
}
}, },
} }
@ -92,6 +97,7 @@ func initCmd() {
// viper.SetDefault(v.Name, v.Value) // viper.SetDefault(v.Name, v.Value)
} }
rootCmd.Flags().BoolVarP(&rev, "version", "v", false, "display version info")
rootCmd.AddCommand(initToolCmd()) rootCmd.AddCommand(initToolCmd())
cobra.OnInitialize(func() { cobra.OnInitialize(func() {
@ -127,8 +133,7 @@ func initToolCmd() *cobra.Command {
toolCmd.Run = func(cmd *cobra.Command, args []string) { toolCmd.Run = func(cmd *cobra.Command, args []string) {
switch { switch {
case rev: case rev:
fmt.Printf("%s v%s build on %s [%s, %s] commit_id(%s) \n", printVersion()
APP_NAME, APP_VER, runtime.Version(), runtime.GOOS, runtime.GOARCH, CommitId)
case secret: case secret:
s, _ := utils.RandSecret(40, 60) s, _ := utils.RandSecret(40, 60)
s = strings.Trim(s, "=") s = strings.Trim(s, "=")
@ -145,3 +150,8 @@ func initToolCmd() *cobra.Command {
return toolCmd return toolCmd
} }
func printVersion() {
fmt.Printf("%s v%s build on %s [%s, %s] commit_id(%s) \n",
APP_NAME, APP_VER, runtime.Version(), runtime.GOOS, runtime.GOARCH, CommitId)
}