feat: add gormv2 and support sqlite

This commit is contained in:
engigu
2024-04-29 17:31:13 +08:00
parent 44054ae157
commit 33e44ce596
22 changed files with 236 additions and 279 deletions
+16 -13
View File
@@ -3,10 +3,9 @@ package migrate
import (
"errors"
"fmt"
"github.com/jinzhu/gorm"
"github.com/sirupsen/logrus"
"gorm.io/gorm"
"message-nest/models"
"message-nest/pkg/setting"
"message-nest/service/settings_service"
)
@@ -43,16 +42,16 @@ func InitAuthTableData() {
func Setup() {
db := models.Setup()
defer func(db *gorm.DB) {
err := db.Close()
if err != nil {
//defer func(db *gorm.DB) {
// err := db.Close()
// if err != nil {
//
// }
//}(db)
}
}(db)
if setting.AppSetting.InitData != "enable" {
return
}
//if setting.AppSetting.InitData != "enable" {
// return
//}
entry := logrus.WithFields(logrus.Fields{
"prefix": "[Init Data]",
@@ -69,9 +68,13 @@ func Setup() {
}
for _, table := range tables {
tableName := db.NewScope(table).TableName()
//tableName := db.NewScope(table).TableName()
tableName := models.GetSchema(table)
entry.Infof("Migrate table: %s", tableName)
db.AutoMigrate(table)
err := db.AutoMigrate(table)
if err != nil {
entry.Infof("Migrate table erorr: %s", err.Error())
}
}
entry.Infof("Init Account data...")