gin框架——使用viper读取配置

什么是viper

Viper是Go应用程序的完整配置解决方案，包括12-Factor（也称为“十二要素”，是一套流行的应用程序开发原则。 其实我也不是很清楚）应用程序。它被设计为在应用程序中工作，可以处理基本上所有类型的配置需求和格式。它支持:

设置默认值

读取JSON, TOML, YAML, HCL, envfile和Java属性配置文件

实时监视和重新读取配置文件(可选)

读取环境变量

从远程配置系统(etcd或Consul)读取数据，并观察变化

从命令行标志读取

读取缓冲区

设置明确的值

具体请指教看官网

https://pkg.go.dev/github.com/spf13/viper#section-readme

安装

go get github.com/spf13/viper
go get gorm.io/gorm
go get gorm.io/driver/mysql

使用

在项目文件下 config/application.yaml文件，内容如下：

server:
  port: 1016
datasource:
  driverName: mysql
  host: 127.0.0.1
  port: 3306
  database: ginInessential
  username: root
  password: admin*123
  charset: utf8
  loc: Asia/Shanghai

写完之后 在main函数中要初始化配置

func InitConfig() {
	workDir, _ := os.Getwd()
	viper.SetConfigName("application")
	viper.SetConfigType("yml")
	viper.AddConfigPath(workDir + "/config")

	err := viper.ReadInConfig()
	if err != nil {
		return
	}
}

在初始化DB中使用

package common

import (
	"GinDemo/model"
	"fmt"
	"github.com/spf13/viper"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"log"
	"net/url"
)

var DB *gorm.DB

func InitDB() *gorm.DB {
	//driverName := "mysql"
	host := viper.GetString("datasource.host")
	port := viper.GetString("datasource.port")
	username := viper.GetString("datasource.username")
	password := viper.GetString("datasource.password")
	database := viper.GetString("datasource.database")
	charset := viper.GetString("datasource.charset")
	loc:= viper.GetString("datasource.loc")
	args := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s&parseTime=true&loc=%s",
		username,
		password,
		host,
		port,
		database,
		charset,
		url.QueryEscape(loc),
	)

	var err error

	DB, err = gorm.Open(mysql.Open(args), &gorm.Config{})
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("连接成功：%v\n", DB)

	return DB
}

main函数调用

func main() {
	//初始化配置
	InitConfig()

	//初始化数据库
	common.InitDB()

	//使用gin
	r := gin.Default()
	r = CollectRoute(r)

	port := viper.GetString("server.port")

	if port != "" {
		_ = r.Run(":" + port)
	} else {
		err := r.Run()
		if err != nil {
			return
		}
	}

}