检查Golang CLI 整数标志是否设置

一分耕耘，一分收获！既然打开了这篇文章《检查Golang CLI 整数标志是否设置》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

当标志为布尔值时，我能够查看该标志是否存在。我试图弄清楚如何查看标志是否存在（如果是整数），如果确实存在，则使用该值（如果不忽略）。

例如，如果使用任一 bool 标志，下面的代码就会执行某些操作。但我想为 -month 设置一个标志，即 go run appname.go -month=01 应该保存 01 的值，但如果不使用 -month ，那么它应该像 bool 标志一样被忽略。在下面的代码中，我通过设置默认值 0 来解决这个问题，因此如果没有使用 -month 标志，则该值为 0。有没有更好的方法来做到这一点？

package main

import (
    "flag"
    "fmt"
    "time"
)

func main() {
    //Adding flags
    useCron := flag.Bool("cron", false, "-cron, uses cron flags and defaults to true")
    useNow := flag.Bool("now", false, "-now, uses cron flags and defaults to true")
    useMonth := flag.Int("month", 0, "-month, specify a month")
    flag.Parse()

    //If -cron flag is used
    if *useCron {
        fmt.Printf("Cron set using cron run as date")
        return
    } else if *useNow {
        fmt.Printf("Use Current date")
        return
    }

    if *useMonth != 0 {
        fmt.Printf("month %v\n", *useMonth)
    }
}

正确答案

请参阅 flag.Value 接口的文档和示例：

您可以定义一个自定义类型，它实现 flag.value，例如：

type cliint struct {
    isset bool
    val   int
}

func (i *cliint) string() string {
    if !i.isset {
        return ""
    }
    return strconv.itoa(i.val)
}

func (i *cliint) set(value string) error {
    v, err := strconv.atoi(value)
    if err != nil {
        return err
    }

    i.isset = true
    i.val = v
    return nil
}

然后使用 flag.var() 进行绑定：

flag.var(&i1, "i1", "1st int to parse")
    flag.var(&i2, "i2", "2nd int to parse")

调用 flag.parse() 后，您可以检查 .isset 标志以查看该标志是否已设置。

playground

另一种方法是在调用 flag.parse() 后调用 flag.Visit() :

var isset1, isset2 bool

    i1 := flag.int("i1", 0, "1st int to parse")
    i2 := flag.int("i2", 0, "2nd int to parse")

    flag.parse([]string{"-i1=123"})

    // flag.visit will only visit flags which have been explicitly set :
    flag.visit(func(fl *flag.flag) {
        if fl.name == "i1" {
            isset1 = true
        }
        if fl.name == "i2" {
            isset2 = true
        }
    })

playground

在 bool 标志中，您也无法检查是否存在，因为您正在检查默认的 false 值。如果您将其更改为 true，您将始终按现有方式获得它。

useCron := flag.Bool("cron", true, "-cron, uses cron flags and defaults to true")
useNow := flag.Bool("now", true, "-now, uses cron flags and defaults to true")

对于任何其他数据类型也是如此，我们只能检查默认值。

您可以使用 flag.nflag()。
此函数返回 cli 设置的许多标志。
当需要所有选项时，此选项才有意义。

@legec 答案满足您的要求。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。