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 
		}
	}
 
}