模板可切换,模板引擎模块化有问题

This commit is contained in:
张超
2025-06-19 22:27:06 +08:00
parent 4c9f6c140f
commit 8241b8c61f
16 changed files with 401 additions and 122 deletions

View File

@@ -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,
},