在Golang的Gin框架中使用Viper配置中心管理Gorm

在Gin框架中使用Viper配置中心来管理Gorm，你可以按照以下步骤进行操作：

安装依赖：
安装Gin，Gorm和Viper：

go get -u github.com/gin-gonic/gin
go get -u gorm.io/gorm
go get -u github.com/spf13/viper

目录结构：

project_root
|-- config
|   |-- config.yaml
|-- database
|   |-- db.go
|-- main.go

创建配置文件（config/config.yaml）：

database:
  driver: mysql
  username: root
  password: your_password
  database: your_database
  host: localhost
  port: 3306

在 database/db.go 中创建Gorm连接：

package database

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"github.com/spf13/viper"
)

func Connect() (*gorm.DB, error) {
	host := viper.GetString("database.host")
	port := viper.GetString("database.port")
	user := viper.GetString("database.username")
	pass := viper.GetString("database.password")
	dbName := viper.GetString("database.database")
	driver := viper.GetString("database.driver")

	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", user, pass, host, port, dbName)

	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		return nil, err
	}

	return db, nil
}

在 main.go 中初始化Viper配置和Gin：

package main

import (
	"github.com/gin-gonic/gin"
	"github.com/spf13/viper"
	"your_project_path/database"
)

func initConfig() {
	viper.SetConfigName("config")
	viper.AddConfigPath("./config")
	err := viper.ReadInConfig()
	if err != nil {
		panic("Failed to read configuration file: " + err.Error())
	}
}

func main() {
	initConfig()

	db, err := database.Connect()
	if err != nil {
		panic("Failed to connect to the database: " + err.Error())
	}

	r := gin.Default()

	// 使用db实例处理HTTP请求
	// ...
	r.Run()
}

现在，你已经成功地使用Viper配置中心来管理Gorm连接。你可以在Gin框架中使用这个Gorm连接来处理HTTP请求和操作数据库。