Golang使用Cobra如何创建支持带参、无参的命令行Flag

默认StringVarP定义的字符串Flag要求必须传入参数，要支持「仅传Flag不传参数」的场景，需要使用Cobra底层依赖的pflag库提供的NoOptDefVal属性，具体实现步骤如下：

1. 定义Flag时配置NoOptDefVal #

NoOptDefVal的作用是：当用户仅传入Flag标识、未传对应参数时，自动将该值赋值给Flag变量，不会抛出参数缺失错误，示例代码如下：

var flag string
// 原有Flag定义逻辑不变
cmd.Flags().StringVarP(&flag, "flag", "f", "", "print names")
// 新增：给flag设置NoOptDefVal，值可以根据业务需求自定义，比如你希望仅传-f时flag的值为某个固定值，就填对应内容
_ = cmd.Flags().Lookup("flag").NoOptDefVal = "default_no_arg"

2. 三种场景的效果说明 #

配置完成后三种场景的取值如下：

• 执行cmd（未传-f）：flag取定义时的默认值，也就是上述代码中的空字符串""
• 执行cmd -f（仅传-f无参数）：flag取你设置的NoOptDefVal值，即上面示例的default_no_arg
• 执行cmd -f "value"（传-f带参数）：flag取用户传入的value

3. 可选：区分「未传Flag」和「传了Flag但无参数」 #

如果你设置的NoOptDefVal和Flag的默认值相同，无法直接通过变量值区分两种场景，可以通过Flag的Changed属性判断：

f := cmd.Flags().Lookup("flag")
if f.Changed {
    // 用户主动传了-f标识
    if flag == "" {
        // 对应 cmd -f 场景，走无参数传入的逻辑
    } else {
        // 对应 cmd -f "value" 场景，走带参数的逻辑
    }
} else {
    // 对应 cmd 场景，用户未传-f，走无Flag的逻辑
}

内容的提问来源于stack exchange，提问作者user17561141