如何确定“flag.VisitAll”是否已设置标志？

使用 Go 的原生 "flag" 包时，无法直接确定标志是否已设置。但可以通过获取 `flag.VisitAll` 和 `flag.Visit` 返回的标志之间的差异来实现此功能。`flag.VisitAll` 遍历所有标志，而 `flag.Visit` 遍历已设置的标志。通过比较这两个集合，可以识别出未设置的标志。

我使用的是 go 的原生 "flag" 包。

它内置了使用 flag.VisitAll 访问所有当前定义的标志的能力。

我正在尝试构建一个片段，尝试从环境变量中获取该标志的值（如果存在）并且如果未设置该标志，并且我找不到一种方法来确定特定的值是否存在标志是否已设置。

有没有办法在不实现新参数类型的情况下实现这一目标？

解决方案

没有函数可以遍历未设置的命令行标志。但是，可以通过获取 VisitAll 和 Visit 返回的标志之间的差异来实现此功能；前者遍历所有标志，而后者遍历设置的标志：

func unsetflags(fs *flag.flagset) []*flag.flag {
    var unset []*flag.flag
    fs.visitall(func(f *flag.flag) {
        unset = append(unset, f)
    })
    fs.visit(func(f *flag.flag) {
        for i, h := range unset {
            if f == h {
                unset = append(unset[:i], unset[i+1:]...)
            }
        }
    })
    return unset
}

您可以在 flag.parse 调用之后使用该函数将任何未设置的标志设置为其环境值：

for _, f := range unsetflags(flag.commandline) {
    v := os.getenv(f.name)
    f.value.set(v)
}

使用 flag.visitall 听起来有点复杂；我建议使用合理的默认值获取环境变量并将其用作标志的默认值 - 这意味着如果未设置标志，环境变量将是后备：

package main

import (
    "flag"
    "fmt"
    "os"
)

func GetEnvDefault(key, def string) string {
    v := os.Getenv(key)

    if v == "" {
        return def
    }

    return v
}

func main() {
    // Uncomment to test behaviour
    // os.Setenv("SERVER_NAME", "donaldduck")

    var serverName string

    flag.StringVar(&serverName, "n", GetEnvDefault("SERVER_NAME", "mickeymouse"), "The human name for the server")
    flag.Parse()

    fmt.Println(serverName)
}

参见：https://play.golang.org/p/ixDsXH31cBF

本篇关于《如何确定“flag.VisitAll”是否已设置标志？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！