Go语言读取各种配置文件
配置文件结构体
config.go
package config type System struct { Mode string `mapstructure:"mode" json:"mode" ini:"mode"` } type Log struct { Prefix string `mapstructure:"prefix" json:"prefix" ini:"prefix"` LogFile bool `mapstructure:"log-file" json:"log-file" ini:"log-file" yaml:"log-file" toml:"log-file"` Stdout string `mapstructure:"stdout" json:"stdout" ini:"stdout"` File string `mapstructure:"file" json:"file" ini:"file"` } type Config struct { System System `json:"system" ini:"system"` Log Log `json:"log" ini:"log"` }
全局配置
global.go
package global import "go_dev/go_read_config/config" var ( CONFIG = new(config.Config) )
一、读取json
config.json
{ "system":{ "mode": "development" }, "log": { "prefix": "[MY-LOG] ", "log-file": true, "stdout": "DEBUG", "file": "WARNING" } }
读取配置
package initialize import ( "encoding/json" "fmt" "go_dev/go_read_config/global" "os" ) func InitConfigFromJson() { // 打开文件 file, _ := os.Open("config.json") // 关闭文件 defer file.Close() //NewDecoder创建一个从file读取并解码json对象的*Decoder,解码器有自己的缓冲,并可能超前读取部分json数据。 decoder := json.NewDecoder(file) //Decode从输入流读取下一个json编码值并保存在v指向的值里 err := decoder.Decode(&global.CONFIG) if err != nil { panic(err) } fmt.Println(global.CONFIG) }
二、读取ini
config.ini
[system] mode='development' ;mode='production' [log] prefix='[MY-LOG] ' log-file=true stdout=DEBUG file=DEBUG
读取
package initialize import ( "fmt" "github.com/go-ini/ini" "go_dev/go_read_config/global" "log" ) func InitConfigFromIni() { err := ini.MapTo(global.CONFIG, "config.ini") if err != nil { log.Println(err) return } fmt.Println(global.CONFIG) }
三、读取yaml
config.yaml
system: mode: 'development' log: prefix: '[MY-LOG] ' log-file: true stdout: 'DEBUG' file: 'DEBUG' mylog: level: 'debug' prefix: '[MY-LOG]' file_path: 'log' file_name: 'mylog.log' max_file_size: 10485760 max_age: 24
读取
package initialize import ( "fmt" "go_dev/go_read_config/global" "gopkg.in/yaml.v2" "io/ioutil" ) func IniConfigFromYaml() { file, err := ioutil.ReadFile("config.yaml") if err != nil { panic(err) } err = yaml.Unmarshal(file, global.CONFIG) if err != nil { fmt.Println(err) return } fmt.Println(global.CONFIG) }
四、读取toml
config.toml
[system] mode = 'development' # mode = 'production' [log] prefix = "[MY-LOG] " log-file = true stdout = "DEBUG" file = "DEBUG"
读取
import ( "fmt" "github.com/BurntSushi/toml" "go_dev/go_read_config/global" ) func InitConfigFromToml() { _, err := toml.DecodeFile("config.toml", global.CONFIG) if err != nil { panic(err) } fmt.Println(global.CONFIG) }
五、万能的viper
Viper是一个方便Go语言应用程序处理配置信息的库。它可以处理多种格式的配置。它支持的特性:
- 设置默认值
- 从JSON、TOML、YAML、HCL和Java properties文件中读取配置数据
- 可以监视配置文件的变动、重新读取配置文件
- 从环境变量中读取配置数据
- 从远端配置系统中读取数据,并监视它们(比如etcd、Consul)
- 从命令参数中读物配置
- 从buffer中读取
- 调用函数设置配置信息
config.json
#json文件 { "appId": "123456789", "secret": "maple123456", "host": { "address": "localhost", "port": 5799 } }
读取
package main import ( "fmt" "github.com/spf13/viper" ) //定义config结构体 type Config struct { AppId string Secret string Host Host } //json中的嵌套对应结构体的嵌套 type Host struct { Address string Port int } func main() { config := viper.New() config.AddConfigPath("./kafka_demo") config.SetConfigName("config") config.SetConfigType("json") if err := config.ReadInConfig(); err != nil { panic(err) } v.WatchConfig() v.OnConfigChange(func(e fsnotify.Event) { fmt.Println("config file changed:", e.Name) if err := config.ReadInConfig(); err != nil { panic(err) } }) fmt.Println(config.GetString("appId")) fmt.Println(config.GetString("secret")) fmt.Println(config.GetString("host.address")) fmt.Println(config.GetString("host.port")) //直接反序列化为Struct var configjson Config if err :=config.Unmarshal(&configjson);err !=nil{ fmt.Println(err) } fmt.Println(configjson.Host) fmt.Println(configjson.AppId) fmt.Println(configjson.Secret)
config.yaml
log: prefix: '[MY-LOG] ' log-file: true stdout: 'DEBUG' file: 'DEBUG'
config.go
type Log struct { Prefix string `mapstructure:"prefix" json:"prefix"` LogFile bool `mapstructure:"log-file" json:"logFile"` Stdout string `mapstructure:"stdout" json:"stdout"` File string `mapstructure:"file" json:"file"` } type Config struct { Log Log `json:"log"` }
global
package global import ( oplogging "github.com/op/go-logging" "github.com/spf13/viper" "go_Logger/config" ) var ( CONFIG config.Config VP *viper.Viper LOG *oplogging.Logger )
读取
const defaultConfigFile = "config/config.yaml" func init() { v := viper.New() v.SetConfigFile(defaultConfigFile) err := v.ReadInConfig() if err != nil { panic(fmt.Errorf("Fatal error config file: %s \n", err)) } v.WatchConfig() v.OnConfigChange(func(e fsnotify.Event) { fmt.Println("config file changed:", e.Name) if err := v.Unmarshal(&global.CONFIG); err != nil { fmt.Println(err) } }) if err := v.Unmarshal(&global.CONFIG); err != nil { fmt.Println(err) } global.VP = v }