gohub/pkg/database/database.go

// Project: go-orm
// File: pkg/database/database.go
// Package database 数据库连接
package database

import (
	"database/sql"
	"fmt"

	"gorm.io/gorm"
	gormlgger "gorm.io/gorm/logger"
)

// DB 数据库连接实例
var DB *gorm.DB
var SQLDB *sql.DB

func Connect(dbConfig gorm.Dialector, _logger gormlgger.Interface) {
	// 使用 gorm.Open 连接数据库
	var err error
	DB, err = gorm.Open(dbConfig, &gorm.Config{
		Logger: _logger,
	})
	// 处理错误
	if err != nil {
		fmt.Println("数据库连接失败", err.Error())
	}
	// 获取底层的 sql.DB 实例
	SQLDB, err = DB.DB()
	if err != nil {
		fmt.Println("数据库连接失败", err.Error())
	}
}