yanweidong
21716c4340
feat(database): 自动迁移表结构并优化数据库初始化函数 将 `AutoMigrate` 逻辑从各数据库初始化方法中提取至统一的 `NewDatabase` 方法内， 避免重复代码。同时修改 `Databases`、`Etcd`、`Memory` 和 `RedisCache` 函数签名， 使其返回实例而非通过参数传递，提高代码可读性和一致性。 ```
125 lines
3.2 KiB
Go
125 lines
3.2 KiB
Go
// Package database provides database connection and management functionality
|
|
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"git.apinb.com/bsm-sdk/core/database/sql"
|
|
"git.apinb.com/bsm-sdk/core/types"
|
|
"git.apinb.com/bsm-sdk/core/vars"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var (
|
|
// MigrateTables holds the tables that need to be auto-migrated on database initialization
|
|
MigrateTables []any
|
|
)
|
|
|
|
// NewDatabase creates a new database connection based on the provided driver type
|
|
// It supports both MySQL and PostgreSQL databases
|
|
func NewDatabase(driver string, dsn []string, options *types.SqlOptions) (db *gorm.DB, err error) {
|
|
driver = strings.ToLower(driver)
|
|
switch driver {
|
|
case "mysql":
|
|
db, err = NewMysql(dsn, options)
|
|
case "postgres":
|
|
db, err = NewPostgres(dsn, options)
|
|
default:
|
|
return nil, fmt.Errorf("unsupported database driver: %s", driver)
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// auto migrate table.
|
|
if len(MigrateTables) > 0 {
|
|
err = db.AutoMigrate(MigrateTables...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return db, nil
|
|
}
|
|
|
|
// NewMysql creates a MySQL database service
|
|
func NewMysql(dsn []string, options *types.SqlOptions) (gormDb *gorm.DB, err error) {
|
|
//set connection default val.
|
|
if options == nil {
|
|
options = &types.SqlOptions{
|
|
MaxIdleConns: vars.SqlOptionMaxIdleConns,
|
|
MaxOpenConns: vars.SqlOptionMaxOpenConns,
|
|
ConnMaxLifetime: vars.SqlOptionConnMaxLifetime,
|
|
LogStdout: false,
|
|
Debug: true,
|
|
}
|
|
}
|
|
|
|
gormDb, err = gorm.Open(mysql.Open(dsn[0]), &gorm.Config{
|
|
SkipDefaultTransaction: true,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if options.Debug {
|
|
gormDb = gormDb.Debug()
|
|
}
|
|
|
|
// 获取通用数据库对象 sql.DB ，然后使用其提供的功能
|
|
sqlDB, err := gormDb.DB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
|
|
sqlDB.SetMaxIdleConns(options.MaxIdleConns)
|
|
// SetMaxOpenConns 设置打开数据库连接的最大数量。
|
|
sqlDB.SetMaxOpenConns(options.MaxOpenConns)
|
|
// SetConnMaxLifetime 设置了连接可复用的最大时间。
|
|
sqlDB.SetConnMaxLifetime(options.ConnMaxLifetime)
|
|
|
|
return gormDb, nil
|
|
}
|
|
|
|
// NewPostgres creates a PostgreSQL database service
|
|
func NewPostgres(dsn []string, options *types.SqlOptions) (gormDb *gorm.DB, err error) {
|
|
//set connection default val.
|
|
if options == nil {
|
|
options = &types.SqlOptions{
|
|
MaxIdleConns: vars.SqlOptionMaxIdleConns,
|
|
MaxOpenConns: vars.SqlOptionMaxOpenConns,
|
|
ConnMaxLifetime: vars.SqlOptionConnMaxLifetime,
|
|
LogStdout: false,
|
|
Debug: true,
|
|
}
|
|
}
|
|
|
|
gormDb, err = sql.NewPostgreSql(dsn[0], options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if options.Debug {
|
|
gormDb = gormDb.Debug()
|
|
}
|
|
|
|
// 获取通用数据库对象 sql.DB ，然后使用其提供的功能
|
|
sqlDB, err := gormDb.DB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
|
|
sqlDB.SetMaxIdleConns(options.MaxIdleConns)
|
|
// SetMaxOpenConns 设置打开数据库连接的最大数量。
|
|
sqlDB.SetMaxOpenConns(options.MaxOpenConns)
|
|
// SetConnMaxLifetime 设置了连接可复用的最大时间。
|
|
sqlDB.SetConnMaxLifetime(options.ConnMaxLifetime)
|
|
|
|
return
|
|
}
|