Golang viper读取配置

go get github.com/spf13/viper

config.toml

title = "toml exaples"
redis = "127.0.0.1:3300"
[mysql]
host = "192.168.1.1"
ports = 3306
username = "root"
password = "root123456"
package main
import(
	"fmt"
	"github.com/spf13/viper"
)
// 读取配置文件config
type Config struct {
	Redis string
	MySQL MySQLConfig
}
type MySQLConfig struct {
	Port int
	Host string
	Username string
	Password string
}
func main() {
	// 把配置文件读取到结构体上
	var config Config
	viper.SetConfigName("config")
	viper.AddConfigPath(".")
	err := viper.ReadInConfig()
	if err != nil {
		fmt.Println(err)
		return
	}
	//将配置文件绑定到config上
	viper.Unmarshal(&config)
	fmt.Println("config: ", config, "redis: ", config.Redis)
}
config:  {127.0.0.1:3300 {0 192.168.1.1 root root123456}} redis:  127.0.0.1:3300

读取多个配置,新增一个config1.json

{
  "redis": "127.0.0.1:33000",
  "mysql": {
    "port": 3306,
    "host": "127.0.0.1",
    "username": "root",
    "password": "123456"
  }
}
package main
import (
	"fmt"
	"github.com/spf13/viper"
)
type Config struct {
	Redis string
	MySQL MySQLConfig
}
type MySQLConfig struct {
	Port     int
	Host     string
	Username string
	Password string
}
func main() {
	// 读取 toml 配置文件
	var config1 Config
	vtoml := viper.New()
	vtoml.SetConfigName("config")
	vtoml.SetConfigType("toml")
	vtoml.AddConfigPath(".")
	if err := vtoml.ReadInConfig(); err != nil {
		fmt.Println(err)
		return
	}
	vtoml.Unmarshal(&config1)
	fmt.Println("read config.toml")
	fmt.Println("config: ", config1, "redis: ", config1.Redis)
	// 读取 json 配置文件
	var config2 Config
	vjson := viper.New()
	vjson.SetConfigName("config1")
	vjson.SetConfigType("json")
	vjson.AddConfigPath(".")
	if err := vjson.ReadInConfig(); err != nil {
		fmt.Println("bb",err)
		return
	}
	vjson.Unmarshal(&config2)
	fmt.Println("read config1.json")
	fmt.Println("config: ", config2, "redis: ", config2.Redis)
}
# 输出
read config.toml
config:  {127.0.0.1:3300 {0 192.168.1.1 root root123456}} redis:  127.0.0.1:3300

read config1.json
config:  {127.0.0.1:33000 {3306 127.0.0.1 root 123456}} redis:  127.0.0.1:33000
posted @ 2023-10-21 14:27  朝阳1  阅读(55)  评论(0编辑  收藏  举报