修复无法初始化数据库的bug

This commit is contained in:
jinnrry 2024-04-15 10:23:26 +08:00
parent 9371864379
commit da919571c6
3 changed files with 28 additions and 24 deletions

26
server/models/base.go Normal file
View File

@ -0,0 +1,26 @@
package models
import "pmail/db"
func SyncTables() {
err := db.Instance.Sync2(&User{})
if err != nil {
panic(err)
}
err = db.Instance.Sync2(&Email{})
if err != nil {
panic(err)
}
err = db.Instance.Sync2(&Group{})
if err != nil {
panic(err)
}
err = db.Instance.Sync2(&Rule{})
if err != nil {
panic(err)
}
err = db.Instance.Sync2(&UserAuth{})
if err != nil {
panic(err)
}
}

View File

@ -38,7 +38,7 @@ func Init(serverVersion string) {
if err != nil {
panic(err)
}
syncTables()
models.SyncTables()
session.Init()
hooks.Init(serverVersion)
// smtp server start
@ -86,26 +86,3 @@ func dirInit() {
}
}
}
func syncTables() {
err := db.Instance.Sync2(&models.User{})
if err != nil {
panic(err)
}
err = db.Instance.Sync2(&models.Email{})
if err != nil {
panic(err)
}
err = db.Instance.Sync2(&models.Group{})
if err != nil {
panic(err)
}
err = db.Instance.Sync2(&models.Rule{})
if err != nil {
panic(err)
}
err = db.Instance.Sync2(&models.UserAuth{})
if err != nil {
panic(err)
}
}

View File

@ -85,6 +85,7 @@ func SetDatabaseSettings(ctx *context.Context, dbType, dbDSN string) error {
if err != nil {
return errors.Wrap(err)
}
models.SyncTables()
return nil
}