模板可切换,模板引擎模块化有问题
This commit is contained in:
@@ -7,7 +7,6 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"go_blog/config"
|
||||
"go_blog/pkg/util"
|
||||
"log/slog"
|
||||
@@ -15,6 +14,8 @@ import (
|
||||
"time"
|
||||
|
||||
"gorm.io/driver/mysql"
|
||||
"gorm.io/driver/postgres"
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/schema"
|
||||
)
|
||||
@@ -22,16 +23,20 @@ import (
|
||||
var DB *gorm.DB
|
||||
|
||||
func InitDatabase(conf *config.Config) {
|
||||
var dialector gorm.Dialector
|
||||
switch conf.DataBase.Driver {
|
||||
case "mysql":
|
||||
dialector = mysql.Open(conf.DataBase.DSN)
|
||||
case "postgres":
|
||||
dialector = postgres.Open(conf.DataBase.DSN)
|
||||
case "sqlite":
|
||||
dialector = sqlite.Open(conf.DataBase.DSN)
|
||||
default:
|
||||
slog.Error("不支持的数据库驱动", "driver", conf.DataBase.Driver)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
conUri := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s&parseTime=True&loc=Local",
|
||||
conf.DataBase.User,
|
||||
conf.DataBase.Password,
|
||||
conf.DataBase.Host,
|
||||
conf.DataBase.Port,
|
||||
conf.DataBase.DBName,
|
||||
conf.DataBase.Charset)
|
||||
// 2. 初始化数据库
|
||||
db, err := gorm.Open(mysql.Open(conUri), &gorm.Config{
|
||||
db, err := gorm.Open(dialector, &gorm.Config{
|
||||
NamingStrategy: schema.NamingStrategy{
|
||||
TablePrefix: conf.DataBase.Prefix,
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user