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

}

巴特西

Go语言读取各种配置文件

配置文件结构体

全局配置

一、读取json

二、读取ini

config.ini

三、读取yaml

四、读取toml

五、万能的viper

最新文章

热门文章