Go Cobra MarkFlagsOneRequired 为什么不等于只能选一个

MarkFlagsOneRequired 表达的是“这一组参数至少设置一个”，并不限制最多设置几个。要让 Go Cobra 的一组参数恰好选一个，需要对同一组同时调用 MarkFlagsOneRequired 和 MarkFlagsMutuallyExclusive：前者排除零个，后者排除多个。

官方地址：https://github.com/spf13/cobra

我会把这类需求先写成一个简单的数量条件，再选择 API。英文名字里出现 One 容易让人联想到“只能一个”，但需要看完整约定。对于文件输入与标准输入二选一的命令，目标是恰好一个来源，而不是只要有来源就接受。

至少一个和只能一个是两个条件

我们做一个名为 sourcepick 的小命令：--file 指定文件来源，--stdin 指定标准输入来源。项目只展示来源选择，不负责打开文件或实际导入，便于把注意力集中在参数约束上。

这里“通过”只表示这一项参数组约束允许继续，不代表参数值一定有效。假如只加 OneRequired，同时传入 file 和 stdin 仍符合“至少一个”的条件；如果业务代码优先处理 file，看起来就像 stdin 被忽略了，实际是命令根本没有声明互斥。

构建一个完整的来源选择命令

下面的单文件代码依赖已经安装的 github.com/spf13/cobra。先创建参数，再标记分组，名称使用注册时的长参数名，不带前面的两个横线。把它放在命令构造函数中，也能让后续集成时保持定义位置集中。

package main

import (
    "fmt"
    "os"
    "strings"

    "github.com/spf13/cobra"
)

func newCommand() *cobra.Command {
    var file string
    var stdin bool
    cmd := &cobra.Command{
        Use: "sourcepick",
        Short: "选择一个输入来源并显示选择结果",
        Args: cobra.NoArgs,
        SilenceUsage: true,
        SilenceErrors: true,
        RunE: func(cmd *cobra.Command, args []string) error {
            // 参数组只判断是否设置；值是否有意义由这里处理。
            if cmd.Flags().Changed("file") {
                if strings.TrimSpace(file) == "" {
                    return fmt.Errorf("--file 不能是空路径")
                }
                // 这里只展示选择，不读取或导入文件。
                _, err := fmt.Fprintln(cmd.OutOrStdout(), "文件来源:", file)
                return err
            }
            if !stdin {
                return fmt.Errorf("--stdin 必须为 true 才表示使用标准输入")
            }
            _, err := fmt.Fprintln(cmd.OutOrStdout(), "标准输入来源")
            return err
        },
    }
    // 先定义成员，再标记同一个参数组。
    cmd.Flags().StringVar(&file, "file", "", "输入文件路径")
    cmd.Flags().BoolVar(&stdin, "stdin", false, "使用标准输入")
    cmd.MarkFlagsOneRequired("file", "stdin")
    cmd.MarkFlagsMutuallyExclusive("file", "stdin")
    return cmd
}

func main() {
    // 每次构建独立命令，避免复用之前解析留下的状态。
    if err := newCommand().Execute(); err != nil {
        fmt.Fprintln(os.Stderr, err)
        os.Exit(1)
    }
}

两条标记语句指向完全相同的成员集合，组合关系才符合本例目标。如果 OneRequired 标记 file 与 stdin，而互斥标记的是另两个参数，不能指望它们替你完成这组二选一规则。标记不存在的参数名属于构造错误，也不能当成运行时用户输入错误处理。

代码在 RunE 中才进行值检查与来源展示；没有在构造阶段执行导入，也没有把业务副作用塞进参数定义过程。接入真实项目时，可以把展示语句替换为调用已经定义好的输入处理逻辑，但应保留组约束和业务值检查。

显式设置不等于值有效

Cobra 的组约束根据参数的 Changed 状态计数，而不是根据字符串是否非空或布尔值是否为 true。因此 --file="" 仍是显式设置，--stdin=false 也是显式设置。上面的 RunE 额外拒绝空文件路径和 false 的标准输入选项，避免“数量条件满足，实际却没有来源”。

同样，给 file 配置一个默认值，并不等于调用者设置了 file。若产品希望“未提供参数时默认读取某文件”，那就不是本文的严格二选一输入协议，应重新考虑是否需要 OneRequired，或者先设计显式来源选择与配置默认值之间的优先级。

这也是我更倾向于让示例默认值保持空和 false 的原因：它让“用户选择”与“程序提供默认行为”保持清晰。这个取舍是本例的接口设计判断，并不是 Cobra 禁止给参数设置默认值。

用参数组合理解预期行为

以下命令用于读者自行尝试，属于使用示例，本文没有执行它们，也没有提供实际终端截图。这里假设程序已构建为 sourcepick。

# 选择文件来源，预期通过分组与非空检查。
./sourcepick --file records.csv

# 选择标准输入来源；本示例只展示选择，不读取数据。
./sourcepick --stdin

# 未选择来源，预期被至少一个约束拒绝。
./sourcepick

# 选择两个来源，预期被互斥约束拒绝。
./sourcepick --file records.csv --stdin

# 显式 false 仍计入参数组，再由业务值检查拒绝。
./sourcepick --stdin=false

不要把具体英文错误文本当作稳定接口协议。命令封装层只需处理 Execute 返回的错误并设置合适退出状态；如果还有结构化错误输出要求，可以在自己的应用层统一包装。示例设置 SilenceErrors 后由 main 输出一次，避免把同一错误重复打印。

集成时还应保留哪些约定

能不能只用 MarkFlagsMutuallyExclusive？可以用于“零个或一个都允许”的场景。本例要求必须选择来源，所以还需要 OneRequired。

MarkFlagsRequiredTogether 能替代它们吗？不能。RequiredTogether 表达组内成员一旦选择就要一起提供，更适合成套参数。本例的两个来源彼此替代，语义不同。

三个参数要三选一怎么办？对同一个三成员集合同时注册至少一个与互斥约束。数量关系没有变化，但每个成员的值合法性仍需单独处理。

已有单元调用能直接复用同一个 Command 吗？本例提供 newCommand，让每次构造有独立参数存储。若反复解析同一个实例，必须考虑已有值和 Changed 状态，不能把上次执行留下的状态当成新输入。把构造与执行分开，也方便命令被其他入口调用。

对于这个来源选择小项目，判断需求的关键句是“必须选择，而且不能同时选择”。把这两部分分别映射为组约束，再处理参数值，就不会因为 OneRequired 的名字误以为它已经包含互斥规则。

参考：Cobra 官方用户指南；参数分组约束实现。