0

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

我还使用 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，所以我对如何引用map[string]string从 Cobra 生成的内容缺乏了解。

4

1 回答 1

0

就绑定部分而言，是正确的。但是，在处理之前，您可以使用 vipver.IsSet() 验证是否存在特定密钥，使用viper.AllKeys()打印所有可用密钥或使用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"]]

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

对于地图转换，您可以像这样（ 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
}
于 2019-10-20T07:20:13.847 回答