增加版本显示

iiiusky · Oct 27, 2020 · 9204a2e · 9204a2e
1 parent 3ba05a9
commit 9204a2e
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -34,6 +34,10 @@ var rootCmd = &cobra.Command{
 	Short: "阿里云API利用工具",
 	Long:  `该工具主要是方便快速使用阿里云api执行一些操作`,
 	PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+		if cmd.Use == "version" {
+			return nil
+		}
+
 		if accessKey == "" || secretKey == "" {
 			return errors.New("请设置ak以及sk的值")
 		}

diff --git a/common/common.go b/common/common.go
@@ -25,6 +25,7 @@ import (
 var AccessKey string
 var SecretKey string
 var ECSRegions []ecs.Region
+var APPVersion string
 
 // 初始化区域信息表
 func InitEcsRegions() bool {

diff --git a/main.go b/main.go
@@ -15,8 +15,12 @@ limitations under the License.
 */
 package main
 
-import "alicloud-tools/cmd"
+import (
+	"alicloud-tools/cmd"
+	"alicloud-tools/common"
+)
 
 func main() {
-  cmd.Execute()
+	common.APPVersion = "1.0.0"
+	cmd.Execute()
 }