go flag库使用

Flag库是golang自带的命令行参数库。可以通过 -name caixukun的方式 或者-name=caixukun 的方式来给命令行参数name传入 caixukun这个值

1. 简单使用命令行参数的情况

1. 需要特别注意 一定不要忘记调用 flag.Parse() 否则你的参数都是默认值。
2. 另外需要注意StringVar 中第一个参数一定是变量的指针 这样才能将解析到的值添加到变量中。

package main

import (
   "flag"
   "fmt"
)

// 适用于最简单的 只有一组命令行参数的情况

// 带有等号的格式
// eg： go run main.go -name=ikun -id=29238 -male

// 不带有等号的格式
// eg： go run main.go -name ikun -id 29238 -male
func main() {
   var name string
   var studentId int
   var isMale bool

   // 定义 flag.xxxVar(&变量, 参数名, 默认值, 帮助信息)
   flag.StringVar(&name, "name", "caixukun.666", "姓名")
   flag.IntVar(&studentId, "id", 128, "学号")
   flag.BoolVar(&isMale, "male", false, "是男性")

   // 注意一定要有解析 否则变量都是默认值
   flag.Parse()

   // 查看解析结果
   fmt.Printf("Name=%s, StudentId=%d, IsMale=%t\n", name, studentId, isMale)
}

2. 有子命令的情况

类似于docker container -ls 中container 就是一个子命令
1. 这种情况下需要首先解析一次。否则flag.Args是无效的。
2. 使用flag.Args()[0] 来判断使用的是哪个子命令
3. 然后重新创建一个命名的flagSet 进一步解析 flag.Args[1:] 部分的命令行参数作为子命令的命令行参数

package main

import (
   "flag"
   "fmt"
)

// 适用于有子命令的情况

// 不带有等号的格式
// eg:  student子命令: go run main.go student -name ikun -id 29238 -male
// eg:  student子命令: go run main.go teacher -name mayun -salary 20000

// 带有等号的格式
// eg:  student子命令: go run main.go student -name=ikun -id=29238 -male
// eg:  student子命令: go run main.go teacher -name=mayun -salary=20000
func main() {
   //首先要用全局默认FlagSet解析一遍 否则flag.Args是无效的。
   flag.Parse()
   flagArgs := flag.Args()
   switch flagArgs[0] {
   case "student":// student子命令解析
      var name string
      var studentId int
      var isMale bool
      stuendtFlags := flag.NewFlagSet("student", flag.ExitOnError)
      stuendtFlags.StringVar(&name, "name", "caixukun.666", "姓名")
      stuendtFlags.IntVar(&studentId, "id", 128, "学号")
      stuendtFlags.BoolVar(&isMale, "male", false, "是否男性")
      
      // 索引从1开始 索引0是子命令 "student" 字符串
      err := stuendtFlags.Parse(flagArgs[1:])
      if err != nil {
         fmt.Printf("Student Invalid cmd option\n")
         return
      }
      fmt.Printf("Student: Name=%s, StudentId=%d, IsMale=%t\n", name, studentId, isMale)
   case "teacher":
      var name string
      var salary int64
      teacherFlags := flag.NewFlagSet("teacher", flag.ExitOnError)
      teacherFlags.StringVar(&name, "name", "caixukun.666", "姓名")
      teacherFlags.Int64Var(&salary, "salary", 128, "薪资")
      err := teacherFlags.Parse(flagArgs[1:])
      if err != nil {
         fmt.Printf("Teacher Invalid cmd option\n")
         return
      }
      fmt.Printf("Teacher: Name=%s, Salary=%d\n", name, salary)
   default:
      fmt.Printf("Invalid cmd option\n")
   }
}