Database Cleaner

This commit is contained in:
2024-04-06 11:47:58 +08:00
parent f8377e7029
commit f36592dd37
5 changed files with 61 additions and 31 deletions

View File

@ -5,19 +5,24 @@ import (
"gorm.io/gorm"
)
var DatabaseAutoActionRange = []any{
&models.Account{},
&models.Realm{},
&models.RealmMember{},
&models.Category{},
&models.Tag{},
&models.Moment{},
&models.Article{},
&models.Comment{},
&models.Reaction{},
&models.Attachment{},
}
func RunMigration(source *gorm.DB) error {
if err := source.AutoMigrate(
&models.Account{},
&models.AccountMembership{},
&models.Realm{},
&models.RealmMember{},
&models.Category{},
&models.Tag{},
&models.Moment{},
&models.Article{},
&models.Comment{},
&models.Reaction{},
&models.Attachment{},
append([]any{
&models.AccountMembership{},
}, DatabaseAutoActionRange...)...,
); err != nil {
return err
}