Golang的flag命令行参数 解析与使用技巧

Go语言flag包提供命令行参数解析，支持基础类型与自定义类型，通过flag.Type定义参数，flag.Parse解析，可实现短选项与子命令。

Go语言内置的flag包提供了简洁高效的命令行参数解析功能，适合大多数CLI程序的需求。它支持布尔、整型、字符串等基础类型，并能自动生成帮助信息。合理使用flag可以提升程序的可配置性和用户体验。

基本参数定义与解析

使用flag前需导入"flag"包。通过flag.Type系列函数定义参数，如flag.String、flag.Int、flag.Bool等。

每个参数包含名称、默认值和说明。调用flag.Parse()后，程序开始解析os.Args[1:]。

var (
    name = flag.String("name", "world", "姓名")
    age  = flag.Int("age", 18, "年龄")
    vip  = flag.Bool("vip", false, "是否VIP")
)
func main() {
    flag.Parse()
    fmt.Printf("Hello %s, age %d, VIP: %v\n", *name, *age, *vip)
}

支持短选项与自定义类型

flag默认不支持短选项（如-n），但可通过重复定义实现。例如：

flag.StringVar(name, "name", "world", "姓名")
flag.StringVar(name, "n", "world", "姓名（简写）")

若需支持复杂类型（如切片、时间），可实现flag.Value接口：

type sliceValue []string
func (s *sliceValue) Set(v string) error {
    *s = append(*s, v)
    return nil
}
func (s *sliceValue) String() string {
    return strings.Join(*s, ",")
}
// 使用
var tags sliceValue
flag.Var(&tags, "tag", "标签列表")

子命令与高级控制

对于复杂CLI工具，可结合flag手动实现子命令。思路是先读取第一个参数判断命令，再针对不同命令初始化各自的FlagSet。

if len(os.Args) < 2 {
    flag.PrintDefaults()
    return
}
switch os.Args[1] {
case "serve":
    serveFlags := flag.NewFlagSet("serve", flag.ExitOnError)
    port := serveFlags.Int("port", 8080, "服务端口")
    serveFlags.Parse(os.Args[2:])
    startServer(*port)
case "fetch":
    // 解析fetch专用参数
}

这种方式能实现类似git clone、git push的子命令结构。

实用技巧与注意事项

以下是一些提升使用体验的小技巧：

• 调用flag.Usage = func(){...}可自定义帮助输出格式
• 使用flag.CommandLine.SetOutput(ioutil.Discard)可关闭默认错误输出，便于测试
• 参数定义顺序不影响解析，但重复定义同一名称会panic
• 布尔参数支持-debug（true）和-debug=false两种写法
• 非选项参数（位置参数）可通过flag.Args()获取，flag.NArg()获取数量

基本上就这些。flag包虽简单，但配合良好设计足以应对多数场景。不复杂但容易忽略。