go - 将带有用破折号分隔的单词的标志映射到 camel_case 配置？

Question

我有一个如下所示的配置格式:

type config struct {
   FooBar string `mapstructure:"foo_bar"`
   BazBot string `mapstructure:"baz_bot"`
}

我的基于 Cobra 的 CLI 有带破折号的标志(例如 --foo-bar=value1 --baz-bot=value2 )。

Viper 不知道如何将它们正确映射到配置中，当它们在 CLI 中设置以覆盖配置文件值时。这里有什么解决办法吗？ Viper 配置设置如下所示:

func loadConfig(cmd *cobra.Command) (*config, error) {
    res := &config{}

    if err := viper.BindPFlags(cmd.Flags()); err != nil {
        return nil, err
    }

    viper.SetEnvPrefix("K")
    viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_", "-", "_"))
    viper.AutomaticEnv()

    var useConfigFile bool

    if configFile, _ := cmd.Flags().GetString("config"); configFile != "" {
        useConfigFile = true
        viper.SetConfigFile(configFile)
    }

    if useConfigFile {
        if err := viper.ReadInConfig(); err != nil {
            return nil, err
        }

        if err := viper.Unmarshal(res); err != nil {
            return nil, err
        }
    }

    return res, nil
}

如果 viper 无法做到，也愿意为支持此功能的库提供建议。

Answer 1

首先，您应该定义标志。根据您的示例，我不知道您是否这样做，但是您应该做类似的事情( based on this ):

    cfg := &config{}
    val := reflect.ValueOf(cfg).Elem()
    for i := 0; i < val.NumField(); i++ {
        fieldType := val.Type().Field(i)
        name := fieldType.Tag.Get("mapstructure")
        rootCmd.PersistentFlags().String(strings.ReplaceAll(name, "_", "-"), "default", "usage")
    }

有了这个 --help命令将显示您可用的标志( --foo-bar 和 --baz-bot )。
然后你应该绑定(bind) loadConfig 中的标志。方法:

    cmd.Flags().VisitAll(func(flag *pflag.Flag) {
        if err := viper.BindPFlag(strings.ReplaceAll(flag.Name, "-", "_"), flag); err != nil {
            return
        }
    })

这对我有用。