gorm框架连接数据库

gorm框架连接数据库

go get 依赖

go get gorm.io/gorm

go get gorm.io/driver/mysql

连接数据库，返回gorm连接对象

package core

import (
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"log"
	"time"
)

// InitGorm 返回数据库连接对象
func InitGorm() *gorm.DB {

	// 获取数据库连接信息
	dsn := "root:1234@tcp(127.0.0.1:3306)/im_server?charset=utf8mb4&parseTime=True&loc=Local"

	// 日志变量
	var mysqlLogger logger.Interface

	// 连接数据库，db 是一个数据库连接对象
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
		Logger: mysqlLogger,
	})
	if err != nil {
		log.Fatalf("[%s] mysql 连接失败", dsn)
	}
	// 获取底层的数据库对象
	sqlDB, _ := db.DB()
	// 对该数据库进行限制
	sqlDB.SetMaxIdleConns(10)
	sqlDB.SetMaxOpenConns(100)
	sqlDB.SetConnMaxLifetime(time.Hour * 4)
	return db
}