欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

golang 使用 viper 读取自定义配置文件

程序员文章站 2023-09-04 21:52:20
viper 支持 yaml、json、 toml、hcl 等格式,读取非常的方便。 viper 官网有案例: go get github.com/spf13/viper...

viper 支持 yaml、json、 toml、hcl 等格式,读取非常的方便。

viper 官网有案例:

go get github.com/spf13/viper

创建 config.yaml 文件

database:
 driver: mysql
 host: 127.0.0.1
 port: 3306
 username: blog
 dbname: blog
 password: 123456

建一个 config.go 用于初始化配置文件

func initconfig() {
  path, err := os.getwd()
  if err != nil {
    panic(err)
  }
  viper.addconfigpath(path + "/config/dev")
  viper.setconfigname("config")
  viper.setconfigtype("yaml")
  if err := viper.readinconfig(); err != nil {
    panic(err)
  }
}

简单使用:

  username := viper.getstring("database.username")
  password := viper.getstring("database.password")
  host := viper.getstring("database.host")
  port := viper.getint("database.port")
  dbname := viper.getstring("database.dbname")
  dsn := fmt.sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8&parsetime=true&loc=local",username,password,host, port, dbname)
  gormpool, err = gorm.open("mysql", dsn)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。