viper的用法

Viper作为配置管理库

Viper可以用来管理各种类型的配置文件，包括JSON、TOML、YAML、HCL、INI等。它提供了一个简单的接口来读取和写入配置文件，可以方便地实现配置的读取、修改和保存。

下面是一个示例：

import "github.com/spf13/viper"

viper.SetConfigType("yaml")
viper.SetConfigFile("config.yaml")
viper.ReadInConfig()

fmt.Println(viper.GetString("database.host"))

这段代码会读取一个名为config.yaml的yaml配置文件，并输出其中的database.host属性值。

Viper作为命令行参数解析库

Viper还可以用来解析命令行参数。它提供了一个简单的接口来定义命令行参数，并可以方便地将解析后的参数转换为结构体。

下面是一个示例：

import "github.com/spf13/viper"

viper.SetDefault("verbose", false)

viper.BindPFlag("verbose", rootCmd.PersistentFlags().Lookup("verbose"))
viper.BindEnv("verbose", "APP_VERBOSE")

type Config struct {
    Verbose bool `mapstructure:"verbose"`
}

var cfg Config

if err := viper.Unmarshal(&cfg); err != nil {
    log.Fatal(err)
}

if cfg.Verbose {
    log.SetLevel(log.DebugLevel)
}

这段代码定义了一个名为verbose的命令行参数，并将其绑定到环境变量APP_VERBOSE和一个结构体的字段上。如果解析出来的cfg.Verbose为true，则将日志级别设置为debug。

Viper作为远程配置库

Viper还可以用来从远程配置库中获取配置信息。它支持从Consul、Etcd、Zookeeper等远程配置库中读取配置信息，并可以根据需要自定义配置读取器。

下面是一个示例：

import "github.com/spf13/viper"

viper.SetConfigType("yaml")
viper.AddRemoteProvider("consul", "http://consul.example.com", "/config/app.yaml")
viper.Set("consul.token", "abc123")

viper.ReadRemoteConfig()

fmt.Println(viper.GetString("database.host"))

这段代码会从Consul的http://consul.example.com/config/app.yaml地址中读取一个名为config.yaml的yaml配置文件，并输出其中的database.host属性值。在读取之前，还需要设置Consul的访问token。