修复无法初始化数据库的bug
This commit is contained in:
parent
9371864379
commit
da919571c6
26
server/models/base.go
Normal file
26
server/models/base.go
Normal 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)
|
||||
}
|
||||
}
|
||||
@ -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)
|
||||
}
|
||||
}
|
||||
|
||||
@ -85,6 +85,7 @@ func SetDatabaseSettings(ctx *context.Context, dbType, dbDSN string) error {
|
||||
if err != nil {
|
||||
return errors.Wrap(err)
|
||||
}
|
||||
models.SyncTables()
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user