如何用Go内置flag包实现子命令后带参数的功能？

Go内置的flag包在解析命令行参数时，会在遇到第一个非flag参数时停止解析，这就导致像git status --short这种子命令后跟随flag参数的场景无法直接实现。以下是问题复现和解决方案：

问题复现 #

示例程序 #

package main

import (
    "flag"
    "fmt"
)

func main() {
    alpha := flag.String("alpha", "", "全局参数1")
    beta := flag.String("beta", "", "全局参数2")
    flag.Parse()
    fmt.Println("ALPHA:", *alpha)
    fmt.Println("BETA:", *beta)
}

不同调用方式的输出 #

# 子命令在所有flag之后，能正常解析所有全局flag
$ ./app --alpha 1 --beta 2 hi
ALPHA: 1
BETA: 2

# 子命令在中间，后续flag无法被解析
$ ./app --alpha 1 hi --beta 2 
ALPHA: 1
BETA: 

# 子命令在最前面，所有全局flag都无法被解析
$ ./app hi --alpha 1 --beta 2 
ALPHA: 
BETA: 

解决方案 #

核心思路是手动拆分命令行参数，为每个子命令创建独立的flag.FlagSet实例，分别解析全局flag和子命令专属flag：

1. 用默认flag包解析子命令之前的全局参数；
2. 提取第一个非flag参数作为子命令；
3. 为子命令初始化专属的flag.FlagSet，解析子命令之后的所有参数。

实现示例（模拟git status --short） #

package main

import (
    "flag"
    "fmt"
    "os"
)

func main() {
    // 定义全局flag
    globalVerbose := flag.Bool("verbose", false, "全局详细输出")
    flag.Parse()

    // 获取剩余未解析的参数（第一个就是子命令）
    args := flag.Args()
    if len(args) == 0 {
        fmt.Println("请指定子命令，如status")
        os.Exit(1)
    }

    // 根据子命令分支处理
    switch args[0] {
    case "status":
        // 为status子命令创建专属FlagSet
        statusFlags := flag.NewFlagSet("status", flag.ExitOnError)
        short := statusFlags.Bool("short", false, "精简输出状态")
        // 解析子命令之后的所有参数
        if err := statusFlags.Parse(args[1:]); err != nil {
            os.Exit(1)
        }

        // 执行子命令逻辑
        fmt.Println("全局verbose:", *globalVerbose)
        if *short {
            fmt.Println("精简状态输出：已修改1个文件，已暂存0个文件")
        } else {
            fmt.Println("完整状态输出：\n- 已修改：README.md\n- 未暂存")
        }
    default:
        fmt.Printf("未知子命令：%s\n", args[0])
        os.Exit(1)
    }
}

测试调用输出 #

# 全局flag+子命令带flag，正常解析
$ ./app --verbose status --short
全局verbose: true
精简状态输出：已修改1个文件，已暂存0个文件

# 子命令flag在前，全局flag在后，全局flag属于子命令参数，不会被解析为全局flag
$ ./app status --short --verbose
全局verbose: false
精简状态输出：已修改1个文件，已暂存0个文件

# 无flag的子命令调用
$ ./app status
全局verbose: false
完整状态输出：
- 已修改：README.md
- 未暂存

内容的提问来源于stack exchange，提问作者Thomas Hunter II