ginCms/utils/db.go

package utils

import (
	"github.com/gin-gonic/gin"
	"github.com/spf13/viper"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	gormLogger "gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
	"log"
	"time"
)

// InitDB 初始化数据库连接
func InitDB() {
	// 1. 配置
	logLevel := gormLogger.Warn
	// 根据应用的mod，控制级别
	switch gin.Mode() {
	case gin.ReleaseMode:
		logLevel = gormLogger.Warn
	case gin.TestMode, gin.DebugMode:
		fallthrough
	default:
		// 最多的日志
		logLevel = gormLogger.Info
	}
	// db 日志
	unionLogger := gormLogger.New(
		log.New(LogWriter(), "\n", log.LstdFlags),
		gormLogger.Config{
			SlowThreshold:             time.Second,
			Colorful:                  false,
			IgnoreRecordNotFoundError: false,
			ParameterizedQueries:      false,
			LogLevel:                  logLevel,
		},
	)
	// gorm 连接配置
	conf := &gorm.Config{
		SkipDefaultTransaction: false,
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true, // 单数表名
		}, // 数据表命名策略
		FullSaveAssociations:                     false,
		Logger:                                   unionLogger,
		NowFunc:                                  nil,
		DryRun:                                   false,
		PrepareStmt:                              false,
		DisableAutomaticPing:                     false,
		DisableForeignKeyConstraintWhenMigrating: true, // 数据表迁移时禁用外键约束
		IgnoreRelationshipsWhenMigrating:         false,
		DisableNestedTransaction:                 false,
		AllowGlobalUpdate:                        false,
		QueryFields:                              false,
		CreateBatchSize:                          0,
		TranslateError:                           false,
		PropagateUnscoped:                        false,
		ClauseBuilders:                           nil,
		ConnPool:                                 nil,
		Dialector:                                nil,
		Plugins:                                  nil,
	}

	// 2. 创建db对象
	dsn := viper.GetString("db.dsn")
	if dbNew, err := gorm.Open(mysql.Open(dsn), conf); err != nil {
		log.Fatalln(err)
	} else {
		db = dbNew
	}
}

// 全局的db对象
var db *gorm.DB

// DB 全局访问db对象的方法
func DB() *gorm.DB {
	return db
}