feat: add database.
This commit is contained in:
57
database/gorm/client.go
Normal file
57
database/gorm/client.go
Normal file
@@ -0,0 +1,57 @@
|
||||
package gorm
|
||||
|
||||
import (
|
||||
"gorm.io/driver/clickhouse"
|
||||
"gorm.io/driver/mysql"
|
||||
"gorm.io/driver/postgres"
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/driver/sqlserver"
|
||||
|
||||
"gorm.io/gorm"
|
||||
|
||||
"github.com/go-kratos/kratos/v2/log"
|
||||
|
||||
conf "github.com/tx7do/kratos-bootstrap/api/gen/go/conf/v1"
|
||||
)
|
||||
|
||||
// NewGormClient 创建GORM数据库客户端
|
||||
func NewGormClient(cfg *conf.Bootstrap, l *log.Helper, migrates []interface{}) *gorm.DB {
|
||||
|
||||
var driver gorm.Dialector
|
||||
switch cfg.Data.Database.Driver {
|
||||
default:
|
||||
fallthrough
|
||||
case "mysql":
|
||||
driver = mysql.Open(cfg.Data.Database.Source)
|
||||
break
|
||||
case "postgres":
|
||||
driver = postgres.Open(cfg.Data.Database.Source)
|
||||
break
|
||||
case "clickhouse":
|
||||
driver = clickhouse.Open(cfg.Data.Database.Source)
|
||||
break
|
||||
case "sqlite":
|
||||
driver = sqlite.Open(cfg.Data.Database.Source)
|
||||
break
|
||||
case "sqlserver":
|
||||
driver = sqlserver.Open(cfg.Data.Database.Source)
|
||||
break
|
||||
}
|
||||
|
||||
client, err := gorm.Open(driver, &gorm.Config{})
|
||||
if err != nil {
|
||||
l.Fatalf("failed opening connection to db: %v", err)
|
||||
return nil
|
||||
}
|
||||
|
||||
// 运行数据库迁移工具
|
||||
if cfg.Data.Database.Migrate {
|
||||
if err = client.AutoMigrate(
|
||||
migrates...,
|
||||
); err != nil {
|
||||
l.Fatalf("failed creating schema resources: %v", err)
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return client
|
||||
}
|
||||
Reference in New Issue
Block a user