From a9e798f2035c484bfebebe354e7004bccdafd310 Mon Sep 17 00:00:00 2001 From: bjdgyc Date: Wed, 26 Apr 2023 21:05:18 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E7=89=88=E6=9C=AC=E6=89=93?= =?UTF-8?q?=E5=8D=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/base/cmd.go | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/server/base/cmd.go b/server/base/cmd.go index 75d5f25..ca4d095 100644 --- a/server/base/cmd.go +++ b/server/base/cmd.go @@ -69,6 +69,11 @@ func initCmd() { Run: func(cmd *cobra.Command, args []string) { // fmt.Println("cmd：", cmd.Use, args) runSrv = true + + if rev { + printVersion() + os.Exit(0) + } }, } @@ -92,6 +97,7 @@ func initCmd() { // viper.SetDefault(v.Name, v.Value) } + rootCmd.Flags().BoolVarP(&rev, "version", "v", false, "display version info") rootCmd.AddCommand(initToolCmd()) cobra.OnInitialize(func() { @@ -127,8 +133,7 @@ func initToolCmd() *cobra.Command { toolCmd.Run = func(cmd *cobra.Command, args []string) { switch { case rev: - 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) + printVersion() case secret: s, _ := utils.RandSecret(40, 60) s = strings.Trim(s, "=") @@ -145,3 +150,8 @@ func initToolCmd() *cobra.Command { 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) +}