mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-29 03:26:37 +08:00
truewhile
80 lines
3.2 KiB
Go
80 lines
3.2 KiB
Go
package database
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/ShukeBta/MMTL/internal/model"
|
|
)
|
|
|
|
// AutoMigrate creates tables for every model registered in the model package.
|
|
func AutoMigrate(db *gorm.DB) error {
|
|
if err := db.AutoMigrate(model.AllModels()...); err != nil {
|
|
return err
|
|
}
|
|
if err := ensurePostgresColumnCompatibility(db); err != nil {
|
|
return err
|
|
}
|
|
if err := ensurePerformanceIndexes(db); err != nil {
|
|
return err
|
|
}
|
|
if err := ensureLibraryRootsCompatibility(db); err != nil {
|
|
return err
|
|
}
|
|
if isSQLite(db) {
|
|
return ensureMediaSearchIndex(db)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ensurePostgresColumnCompatibility(db *gorm.DB) error {
|
|
if !isPostgres(db) {
|
|
return nil
|
|
}
|
|
statements := []string{
|
|
`ALTER TABLE media ALTER COLUMN container TYPE varchar(128)`,
|
|
`ALTER TABLE media ALTER COLUMN genres TYPE text`,
|
|
`ALTER TABLE media ALTER COLUMN series_id TYPE varchar(128)`,
|
|
`ALTER TABLE media ALTER COLUMN duplicate_of TYPE varchar(128)`,
|
|
`ALTER TABLE playback_histories ALTER COLUMN media_id TYPE varchar(128)`,
|
|
`ALTER TABLE favorites ALTER COLUMN media_id TYPE varchar(128)`,
|
|
`ALTER TABLE playlist_items ALTER COLUMN media_id TYPE varchar(128)`,
|
|
}
|
|
for _, stmt := range statements {
|
|
if err := db.Exec(stmt).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ensurePerformanceIndexes(db *gorm.DB) error {
|
|
statements := []string{
|
|
`CREATE INDEX IF NOT EXISTS idx_media_library_created_active ON media(library_id, created_at DESC) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_media_library_release_active ON media(library_id, release_date DESC, year DESC) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_media_library_episode_active ON media(library_id, season_num, episode_num, created_at DESC) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_media_library_root_active ON media(library_id, library_root_id) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_media_series_active ON media(series_id, season_num, episode_num) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_favorites_user_media_active ON favorites(user_id, media_id) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_playback_histories_user_media_active ON playback_histories(user_id, media_id, watched_at DESC) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_playback_histories_resume_active ON playback_histories(user_id, completed, watched_at DESC) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_play_profiles_user_created_active ON play_profiles(user_id, created_at DESC) WHERE deleted_at IS NULL`,
|
|
}
|
|
if isSQLite(db) {
|
|
statements = append(statements,
|
|
`CREATE INDEX IF NOT EXISTS idx_media_title_active ON media(title COLLATE NOCASE) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_media_original_name_active ON media(original_name COLLATE NOCASE) WHERE deleted_at IS NULL`,
|
|
)
|
|
} else {
|
|
statements = append(statements,
|
|
`CREATE INDEX IF NOT EXISTS idx_media_title_active ON media(title) WHERE deleted_at IS NULL`,
|
|
`CREATE INDEX IF NOT EXISTS idx_media_original_name_active ON media(original_name) WHERE deleted_at IS NULL`,
|
|
)
|
|
}
|
|
for _, stmt := range statements {
|
|
if err := db.Exec(stmt).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|