有没有一种方法可以确定在使用“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