25 lines
937 B
Go
25 lines
937 B
Go
package live
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"kra/pkg/database/migration"
|
|
)
|
|
|
|
func Migrations() []migration.Step {
|
|
return []migration.Step{{ID: "202609190001_live_schema", Migrate: func(db *gorm.DB) error {
|
|
if err := migration.CreateMissingTables(db, &categoryPO{}, &roomPO{}); err != nil {
|
|
return err
|
|
}
|
|
var count int64
|
|
if err := db.Model(&categoryPO{}).Count(&count).Error; err != nil {
|
|
return err
|
|
}
|
|
if count > 0 {
|
|
return nil
|
|
}
|
|
return db.Create(&[]categoryPO{{Name: "游戏竞技", Icon: "🎮", Sort: 1, Enabled: true}, {Name: "音乐区", Icon: "🎵", Sort: 2, Enabled: true}, {Name: "闲聊电台", Icon: "🌙", Sort: 3, Enabled: true}, {Name: "舞蹈才艺", Icon: "💃", Sort: 4, Enabled: true}, {Name: "娱乐直播", Icon: "🎪", Sort: 5, Enabled: true}}).Error
|
|
}}, {ID: "202609220002_live_follow_schema", Migrate: func(db *gorm.DB) error {
|
|
return migration.CreateMissingTables(db, &followPO{})
|
|
}}}
|
|
}
|