如何使用 Viper 访问类型为“StringToStringVar”的 Cobra Flag？

Question

我正在尝试用 Go 开发一个应用程序，它接受命令行上的输入作为键值对字符串。为此，我使用了 Cobra 库中的 StrngToStringVar。

我也在使用 Viper 将这些标志绑定到配置，但是无论我设置什么值，我似乎都无法从 Viper 获得它。

这是代码

package main

import (
    "fmt"

    "github.com/spf13/cobra"
    "github.com/spf13/viper"

    "github.com/davecgh/go-spew/spew"
)

func main() {
    var items map[string]string
    var name string

    rootCmd := &cobra.Command{
        Use:   "cobra",
        Short: "Test options on the command line",
        Long:  ``,
        Run: func(ccmd *cobra.Command, args []string) {

            spew.Dump(viper.GetString("name"))
            spew.Println("")

            fmt.Println("GetStringMap")
            spew.Dump(viper.GetStringMap("items"))

            fmt.Println("")

            fmt.Println("GetStringMapString")
            spew.Dump(viper.GetStringMapString("items"))
        },
    }

    rootCmd.Flags().StringVar(&name, "name", "", "Name of the list")
    rootCmd.Flags().StringToStringVar(&items, "item", nil, "Map stating the items to be included")

    viper.BindPFlag("name", rootCmd.Flags().Lookup("name"))
    viper.BindPFlag("items", rootCmd.Flags().Lookup("item"))

    rootCmd.Execute()
}

如果我运行使用命令 go run .\main.go --item shopping=apple,banana --name foobar 我会得到以下结果

(string) (len=6) "foobar"

GetStringMap
(map[string]interface {}) {
}

GetStringMapString
(map[string]string) {
}

可以看出，即使我（我相信）正确设置了输入，他的输出也不包含任何项目。我一直在尝试使用 PR https://github.com/spf13/pflag/pull/133 来弄清楚如何去做，但我没有任何运气。

我想知道绑定是否不正确，但我在其他项目中成功使用了 CObra，所以我对如何引用 Cobra 生成的 map[string]string 缺乏理解。

Answer 1

就绑定部分而言，是正确的。但是，在处理之前，您可以使用 vipver.IsSet(), print all the available keys using viper.AllKeys() or print everything using viper.AllSettings().

验证特定键是否存在

这是一个例子 (test.go):

package main

import (
    "fmt"

    "github.com/spf13/cobra"
    "github.com/spf13/viper"
)

func main() {
    var items map[string]string

    rootCmd := &cobra.Command{
        Use: "app",
        Run: func(ccmd *cobra.Command, args []string) {

            fmt.Println("item exists?", viper.IsSet("item"))
            fmt.Println("GetString  :", viper.GetString("item"))
            fmt.Println("Keys       :", viper.AllKeys())
            fmt.Println("Settings   :", viper.AllSettings())
        },
    }

    rootCmd.Flags().StringToStringVarP(&items, "item", "i", nil, "Map stating the items to be included")
    viper.BindPFlag("item", rootCmd.Flags().Lookup("item"))
    rootCmd.Execute()
}

输出：

$ ./test.exe -i 'k=v,"a=b,c"'
item exists? true
GetString  : [k=v,"a=b,c"]
Keys       : [item]
Settings   : map[item:[k=v,"a=b,c"]]

GetStringMap() 和 GetStringMapString() 的用法对于 JSON、YAML 等其他格式更有意义。您可以跟踪这些函数的执行到 JSON在调试模式下解组调用。

地图转换，你可以这样写自己的函数(live):

func getStringMap(s string) map[string]string {
    entries := strings.Split(s, ",")

    m := make(map[string]string)
    for _, e := range entries {
        tokens := strings.Split(e, "=")
        k := strings.TrimSpace(tokens[0])
        v := strings.TrimSpace(tokens[1])
        m[k] = v
    }

    return m
}

如何使用 Viper 访问类型为“StringToStringVar”的 Cobra Flag？

How is a Cobra Flag of type `StringToStringVar` access using Viper?

go

go-cobra

viper-go