Go读取配置文件
范闲 人气:0Go语言提供了很简便的读取json和yaml文件的api,我们可以很轻松将一个json或者yaml文件转换成Go的结构体,但是如果直接在项目中读取配置文件,这种方式并不够好。缺点如下:
实际开发中配置项在不同环境下,配置的值是不同的
上面的问题可以通过不同的配置文件去决定在那个环境读取那个配置文件,但是还有一个问题就是实际开发中,配置项有些是相同的,有些是不同的,如果配置文件有一个主配置文件,里面存放的是不同环境相同的配置项,还有一个跟随环境的配置文件,里面存放的是不同环境的配置项,然后读取两个配置文件后,做一个merge,这样得到的结果就是总共的配置项信息。
有些配置项是必填的,有些配置项的值是一些特殊的值,比如,邮箱,手机号,IP信息等
来看看gonfig是怎么解决这个问题的
- 安装gonfig
go get github.com/xiao-ren-wu/gonfig
- 项目中新建配置目录,并编写对应配置文件,在配置目录同级目录添加读取配置文件的go文件
conf.yaml
文件中存放的是通用的配置,conf-{{active}}.yaml
中存放不同环境不同的配置信息。
conf |-conf.yaml |-conf-dev.yaml |-conf-prod.yaml |config.go
- 利用
go:embed
将配置文件载入到内存,调用gonfig.Unmarshal
读取配置文件
package config import ( "model" "github.com/xiao-ren-wu/gonfig" "embed" ) //go:embed *.yaml var confDir embed.FS // 我们配置文件的配置struct type AppConf struct { AppName string `yaml:"app-name" json:"app-name"` DB DBConf `yaml:"db" json:"db"` } type DBConf struct { Username string `yaml:"username" json:"username"` Password string `yaml:"password" json:"password"` } var conf Conf func Init() { if err := gonfig.Unmarshal(confDir, &conf); err != nil { panic(err) } } func GetConf() *Conf { return &conf }
这样就完成了项目中配置文件的读取,是不是很简单? 此时读到的配置形式是conf-{{profile}}.yaml
和conf.yaml
的总和,如果conf-{{profile}}.yaml
中定义的属性和conf.yaml
相同,那么会以conf-{{profile}}.yaml
为准
约定
gonfig
API的简洁性的原因是因为背后做了很多约束,只有符合约束的配置才能被成功读取,具体的约束条件如下:
gonfig.Unmarshal
会默认读取文件名称有前缀conf
的文件通过环境变量
profile
作为环境名称,如果没有配置,默认dev。程序会寻找
conf.yaml
作为主配置文件,conf-{{profile}}.yaml
作为环境特有配置文件,然后对文件内容进行合并如果
conf-{{profile}}.yaml
中的属性和conf.yaml
中属性都有定义,那么会以conf-{{profile}}.yaml
为准。
根据项目定制化配置文件
gonfig.Unmarshal
的函数签名func Unmarshal(confDir ReadFileDir, v interface{}, ops ...Option) error
提供很多配置项,供用户定制化需求,具体的配置信息如下:
更改配置文件名称前缀
FilePrefix(prefix string)
更改读取配置文件类型
UnmarshalWith(uType UnmarshalType)
更改读取的环境变量名称
ProfileUseEnv(envName, defaultProfile string)
自定义设置profile
ProfileFunc(f func() string)
原理篇
gonfig
的实现也很简单,核心的源码如下:
func Unmarshal(confDir ReadFileDir, v interface{}, ops ...Option) error { if v != nil && reflect.ValueOf(v).Kind() != reflect.Ptr { return gonfig_error.ErrNonPointerArgument } var cs = &confStruct{ confPrefix: "conf", envName: "profile", defaultEnvValue: "dev", unmarshalType: Yaml, } cs.activeProfileFunc = func() string { return getActiveProfile(cs.envName, cs.defaultEnvValue) } for _, op := range ops { op(cs) } cs.profileActive = cs.activeProfileFunc() if err := loadConf(confDir, cs); err != nil { return err } // copy val v1 := reflect.New(reflect.TypeOf(v).Elem()).Interface() if err := fileUnmarshal(cs.activeConfRaw, v1, cs.unmarshalType); err != nil { return err } if len(cs.masterConfRaw) == 0 { return gonfig_error.MasterProfileConfNotSetError } if err := fileUnmarshal(cs.masterConfRaw, v, cs.unmarshalType); err != nil { return err } return mergo.Merge(v, v1, mergo.WithOverride) }
大概的原理就是复制了一份用户传给函数的结构体v1,结构体v1和v分别用于接收conf-{{profile}}.yaml
中的属性和conf.yaml
的配置信息,然后通过调用三方开源库mergo
对两个结构体的属性做一个merge。
这就是关于gonfig
的全部内容啦~~~
github地址是:https://github.com/xiao-ren-wu/gonfig
加载全部内容