登录
首页 >  Golang >  Go教程

读取Go项目中的配置文件的方法

来源:脚本之家

时间:2022-12-25 08:42:42 184浏览 收藏

知识点掌握了,还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战,手把手教大家学习《读取Go项目中的配置文件的方法》,在实现功能的过程中也带大家重新温习相关知识点,温故而知新,回头看看说不定又有不一样的感悟!

Go语言提供了很简便的读取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}}.yamlconf.yaml的总和,如果conf-{{profile}}.yaml中定义的属性和conf.yaml相同,那么会以conf-{{profile}}.yaml为准

约定

gonfigAPI的简洁性的原因是因为背后做了很多约束,只有符合约束的配置才能被成功读取,具体的约束条件如下:

  • 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)

  • 自定义设置profileProfileFunc(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

今天带大家了解了读取、配置文件的相关知识,希望对你有所帮助;关于Golang的技术知识我们会一点点深入介绍,欢迎大家关注golang学习网公众号,一起学习编程~

声明:本文转载于:脚本之家 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>
评论列表