当前位置： > > > > 您可以为 Cobra 中的标志指定可选参数吗？

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

来源：stackoverflow
2024-04-29 16:39:38
0浏览
收藏

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《您可以为 Cobra 中的标志指定可选参数吗？》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

问题内容

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

c.PersistentFlags().IntVar(&SomeFlag, optionSomeFlag, 0, "做某事(范围: x-y)")

默认值为 0，因此如果用户不切换该标志，则不会打印任何内容。 如何使标志接受参数但本身有默认值？ 即如果默认值为 5

./program --someflag 输出将是 5

但如果我这样做了

./program --someflag=1 输出将为 1

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

正确答案

可以使用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

以上就是《您可以为 Cobra 中的标志指定可选参数吗？》的详细内容，更多关于的资料请关注米云公众号！