diff --git a/common/database/initialize.go b/common/database/initialize.go index f78fc0ca..f59ac3bb 100644 --- a/common/database/initialize.go +++ b/common/database/initialize.go @@ -2,6 +2,7 @@ package database +// Setup 配置数据库 func Setup(driver string) { dbType := driver if dbType == "mysql" { diff --git a/common/database/interface.go b/common/database/interface.go index 0fc32337..59c5e1ce 100644 --- a/common/database/interface.go +++ b/common/database/interface.go @@ -2,6 +2,7 @@ package database import "gorm.io/gorm" +// Database 数据库配置 type Database interface { Setup() Open(conn string, cfg *gorm.Config) (db *gorm.DB, err error) diff --git a/common/database/mysql_drive.go b/common/database/mysql_drive.go index c30d7a2b..faa4d977 100644 --- a/common/database/mysql_drive.go +++ b/common/database/mysql_drive.go @@ -5,12 +5,12 @@ import ( . "log" "time" + goAdminLogger "github.com/go-admin-team/go-admin-core/logger" "gorm.io/driver/mysql" "gorm.io/gorm" "gorm.io/gorm/logger" "gorm.io/gorm/schema" - goAdminLogger "github.com/go-admin-team/go-admin-core/logger" "go-admin/common/config" "go-admin/common/global" "go-admin/common/log" @@ -18,9 +18,11 @@ import ( toolsConfig "go-admin/tools/config" ) +// Mysql mysql配置结构体 type Mysql struct { } +// Setup 配置步骤 func (e *Mysql) Setup() { global.Source = e.GetConnect() log.Info(tools.Green(global.Source)) @@ -59,16 +61,17 @@ func (e *Mysql) Setup() { } } -// 打开数据库连接 +// Open 打开数据库连接 func (e *Mysql) Open(db *sql.DB, cfg *gorm.Config) (*gorm.DB, error) { return gorm.Open(mysql.New(mysql.Config{Conn: db}), cfg) } -// 获取数据库连接 +// GetConnect 获取数据库连接 func (e *Mysql) GetConnect() string { return toolsConfig.DatabaseConfig.Source } +// GetDriver 获取连接 func (e *Mysql) GetDriver() string { return toolsConfig.DatabaseConfig.Driver }