Commit c6226f36 by CaIon

feat(database): implement database migration logic for PostgreSQL and add fast migration fallback

parent eaf3b6ac
...@@ -232,6 +232,30 @@ func InitLogDB() (err error) { ...@@ -232,6 +232,30 @@ func InitLogDB() (err error) {
} }
func migrateDB() error { func migrateDB() error {
if !common.UsingPostgreSQL {
return migrateDBFast()
}
err := DB.AutoMigrate(
&Channel{},
&Token{},
&User{},
&Option{},
&Redemption{},
&Ability{},
&Log{},
&Midjourney{},
&TopUp{},
&QuotaData{},
&Task{},
&Setup{},
)
if err != nil {
return err
}
return nil
}
func migrateDBFast() error {
var wg sync.WaitGroup var wg sync.WaitGroup
errChan := make(chan error, 12) // Buffer size matches number of migrations errChan := make(chan error, 12) // Buffer size matches number of migrations
...@@ -273,7 +297,6 @@ func migrateDB() error { ...@@ -273,7 +297,6 @@ func migrateDB() error {
return err return err
} }
} }
common.SysLog("database migrated") common.SysLog("database migrated")
return nil return nil
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment