你可以为 Cobra 中的标志指定可选参数吗？

Question

假设我的程序中有这个标志，它只打印正数：

c.PersistentFlags().IntVar(&SomeFlag, optionSomeFlag, 0, "do something (range: x-y)")

默认值为 0，因此如果用户不切换标志，则不会打印任何内容。

如何使标志接受参数但本身有默认值？ 即如果默认值为 5

./program --someflag

输出将是

5

但如果我这么做了

./program --someflag=1

输出将是

1

我尝试遵循 Cobra 的用户指南，并期待一种命令类型，该命令类型允许我仅在用户触发该标志时指定默认值，而不仅仅是完全指定默认值。我可能误解了这一点或错过了一些东西。

Answer 1

可以使用NoOptDefVal来完成

rootCmd.PersistentFlags().Lookup("someflag").NoOptDefVal = "5"

在下面的代码中，您可以找到具有您描述的行为的带有 cobra 的命令行应用程序的完整示例

package main

import (
    "fmt"
    "github.com/spf13/cobra"
)

func main() {
    var someFlag int
    var defaultSomeFlag = "5"

    // Create the root command.
    rootCmd := &cobra.Command{
        Use:   "program",
        Short: "A brief description of your application",
        Long:  "A longer description of your application",
        Run: func(cmd *cobra.Command, args []string) {
            // Check whether the flag was explicitly set.
            if cmd.Flags().Lookup("someflag").Changed {
                fmt.Printf("someflag: %d\n", someFlag)
            } else {
                // If the flag was not explicitly set don't print a value.
                fmt.Printf("someflag is not set\n")
            }
        },
    }

    // Define the flag and set its default value.
    rootCmd.PersistentFlags().IntVar(&someFlag, "someflag", 0, "do something (range: x-y)")
    rootCmd.PersistentFlags().Lookup("someflag").NoOptDefVal = defaultSomeFlag

    // Execute the root command.
    if err := rootCmd.Execute(); err != nil {
        fmt.Println(err)
    }
}

下面是不同标志值的执行结果。

$ ./test
someflag is not set
$ ./test --someflag
someflag: 5
$ ./test --someflag=3
someflag: 3

Answer 2

我计划通过自己解析

os.Args[1:]

来查找不带参数的标志，并将默认值插入到参数的修改副本中，并将修改后的副本传递给

rootCommand.SetArgs

来解决此限制。

感觉有点黑客且难以维持，但我没有看到其他解决方案。