使用Viper访问类型为"StringToStringVar"的眼镜蛇标志如何?

Question

我正在尝试在 Go 中开发一个应用程序，该应用程序接受命令行上的输入作为键值对字符串。为此，我正在使用眼镜蛇库中的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) {
}

可以看出，即使我正确设置了输入(我相信(，他的输出也不包含任何项目。我一直在尝试使用公关 https://github.com/spf13/pflag/pull/133 来研究如何做到这一点，但我没有任何运气。

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

Answer 1

就绑定部分而言，它是正确的。但是，在处理之前，您可以使用 vipver 验证特定密钥是否存在。IsSet((，使用 viper 打印所有可用的键。AllKeys(( 或使用 viper 打印所有内容。全部设置((。

下面是一个示例(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 解组调用。

对于地图转换，您可以像这样编写自己的函数(实时(：

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
}