diff --git a/internal/service/service.go b/internal/service/service.go index acdad49..20cffee 100644 --- a/internal/service/service.go +++ b/internal/service/service.go @@ -5,14 +5,11 @@ package service import ( "context" - "strconv" - "strings" "time" "go.uber.org/zap" "github.com/ShukeBta/MediaStationGo/internal/config" - "github.com/ShukeBta/MediaStationGo/internal/model" "github.com/ShukeBta/MediaStationGo/internal/repository" ) @@ -124,132 +121,6 @@ func (c *Container) Boot() { } } -func (c *Container) warmMediaSearchIndex(ctx context.Context) { - if c == nil || c.Repo == nil || c.Repo.Media == nil { - return - } - if !mediaSearchWarmupEnabled(ctx, c.Repo) { - if c.Log != nil { - c.Log.Info("media search index warmup disabled") - } - return - } - // 错峰:FTS 正常由 media 表触发器实时维护,回填只是升级或异常后的 - // 兜底。先让登录、首页等关键路径跑起来,再开始后台补索引。 - select { - case <-ctx.Done(): - return - case <-time.After(mediaSearchWarmupDelay(ctx, c.Repo)): - } - batchSize := mediaSearchWarmupBatchSize(ctx, c.Repo) - pause := mediaSearchWarmupPause(ctx, c.Repo) - total := int64(0) - for { - select { - case <-ctx.Done(): - return - default: - } - n, err := c.Repo.Media.BackfillSearchIndex(ctx, batchSize) - if err != nil { - c.Log.Debug("media search index warmup stopped", zap.Error(err)) - return - } - if n == 0 { - if total > 0 { - c.Log.Info("media search index warmed", zap.Int64("indexed", total)) - } - return - } - total += n - select { - case <-ctx.Done(): - return - case <-time.After(pause): - } - } -} - -func mediaSearchWarmupEnabled(ctx context.Context, repo *repository.Container) bool { - if repo == nil || repo.Setting == nil { - return true - } - value, err := repo.Setting.Get(ctx, "search.index_warmup_enabled") - if err != nil || strings.TrimSpace(value) == "" { - return true - } - return parseBoolSetting(value, true) -} - -func mediaSearchWarmupDelay(ctx context.Context, repo *repository.Container) time.Duration { - seconds := mediaSearchWarmupIntSetting(ctx, repo, "search.index_warmup_delay_seconds", 120) - if seconds < 30 { - seconds = 30 - } - return time.Duration(seconds) * time.Second -} - -func mediaSearchWarmupBatchSize(ctx context.Context, repo *repository.Container) int { - size := mediaSearchWarmupIntSetting(ctx, repo, "search.index_warmup_batch_size", 100) - if size < 10 { - size = 10 - } - if size > 1000 { - size = 1000 - } - return size -} - -func mediaSearchWarmupPause(ctx context.Context, repo *repository.Container) time.Duration { - ms := mediaSearchWarmupIntSetting(ctx, repo, "search.index_warmup_pause_ms", 2000) - if ms < 250 { - ms = 250 - } - return time.Duration(ms) * time.Millisecond -} - -func mediaSearchWarmupIntSetting(ctx context.Context, repo *repository.Container, key string, fallback int) int { - if repo == nil || repo.Setting == nil { - return fallback - } - value, err := repo.Setting.Get(ctx, key) - if err != nil { - return fallback - } - n, err := strconv.Atoi(strings.TrimSpace(value)) - if err != nil || n <= 0 { - return fallback - } - return n -} - -func (c *Container) NormalizeCloudLibraryTypes(ctx context.Context) error { - if c == nil || c.Repo == nil || c.Repo.Library == nil || c.Repo.DB == nil { - return nil - } - libs, err := c.Repo.Library.List(ctx) - if err != nil { - return err - } - for _, lib := range libs { - info, ok := ParseCloudLibraryMount(lib.Path) - if !ok { - continue - } - want := InferCloudMountMediaType(info.DisplayDir, lib.Name) - if want == "" || want == lib.Type { - continue - } - if err := c.Repo.DB.WithContext(ctx). - Model(&model.Library{}). - Where("id = ?", lib.ID). - Update("type", want).Error; err != nil { - return err - } - } - return nil -} - // Context is canceled when the service container is closing. func (c *Container) Context() context.Context { if c == nil || c.stopCtx == nil { diff --git a/internal/service/service_library_normalize.go b/internal/service/service_library_normalize.go new file mode 100644 index 0000000..73d0aa0 --- /dev/null +++ b/internal/service/service_library_normalize.go @@ -0,0 +1,34 @@ +package service + +import ( + "context" + + "github.com/ShukeBta/MediaStationGo/internal/model" +) + +func (c *Container) NormalizeCloudLibraryTypes(ctx context.Context) error { + if c == nil || c.Repo == nil || c.Repo.Library == nil || c.Repo.DB == nil { + return nil + } + libs, err := c.Repo.Library.List(ctx) + if err != nil { + return err + } + for _, lib := range libs { + info, ok := ParseCloudLibraryMount(lib.Path) + if !ok { + continue + } + want := InferCloudMountMediaType(info.DisplayDir, lib.Name) + if want == "" || want == lib.Type { + continue + } + if err := c.Repo.DB.WithContext(ctx). + Model(&model.Library{}). + Where("id = ?", lib.ID). + Update("type", want).Error; err != nil { + return err + } + } + return nil +} diff --git a/internal/service/service_search_warmup.go b/internal/service/service_search_warmup.go new file mode 100644 index 0000000..6aa0250 --- /dev/null +++ b/internal/service/service_search_warmup.go @@ -0,0 +1,111 @@ +package service + +import ( + "context" + "strconv" + "strings" + "time" + + "go.uber.org/zap" + + "github.com/ShukeBta/MediaStationGo/internal/repository" +) + +func (c *Container) warmMediaSearchIndex(ctx context.Context) { + if c == nil || c.Repo == nil || c.Repo.Media == nil { + return + } + if !mediaSearchWarmupEnabled(ctx, c.Repo) { + if c.Log != nil { + c.Log.Info("media search index warmup disabled") + } + return + } + // 错峰:FTS 正常由 media 表触发器实时维护,回填只是升级或异常后的 + // 兜底。先让登录、首页等关键路径跑起来,再开始后台补索引。 + select { + case <-ctx.Done(): + return + case <-time.After(mediaSearchWarmupDelay(ctx, c.Repo)): + } + batchSize := mediaSearchWarmupBatchSize(ctx, c.Repo) + pause := mediaSearchWarmupPause(ctx, c.Repo) + total := int64(0) + for { + select { + case <-ctx.Done(): + return + default: + } + n, err := c.Repo.Media.BackfillSearchIndex(ctx, batchSize) + if err != nil { + c.Log.Debug("media search index warmup stopped", zap.Error(err)) + return + } + if n == 0 { + if total > 0 { + c.Log.Info("media search index warmed", zap.Int64("indexed", total)) + } + return + } + total += n + select { + case <-ctx.Done(): + return + case <-time.After(pause): + } + } +} + +func mediaSearchWarmupEnabled(ctx context.Context, repo *repository.Container) bool { + if repo == nil || repo.Setting == nil { + return true + } + value, err := repo.Setting.Get(ctx, "search.index_warmup_enabled") + if err != nil || strings.TrimSpace(value) == "" { + return true + } + return parseBoolSetting(value, true) +} + +func mediaSearchWarmupDelay(ctx context.Context, repo *repository.Container) time.Duration { + seconds := mediaSearchWarmupIntSetting(ctx, repo, "search.index_warmup_delay_seconds", 120) + if seconds < 30 { + seconds = 30 + } + return time.Duration(seconds) * time.Second +} + +func mediaSearchWarmupBatchSize(ctx context.Context, repo *repository.Container) int { + size := mediaSearchWarmupIntSetting(ctx, repo, "search.index_warmup_batch_size", 100) + if size < 10 { + size = 10 + } + if size > 1000 { + size = 1000 + } + return size +} + +func mediaSearchWarmupPause(ctx context.Context, repo *repository.Container) time.Duration { + ms := mediaSearchWarmupIntSetting(ctx, repo, "search.index_warmup_pause_ms", 2000) + if ms < 250 { + ms = 250 + } + return time.Duration(ms) * time.Millisecond +} + +func mediaSearchWarmupIntSetting(ctx context.Context, repo *repository.Container, key string, fallback int) int { + if repo == nil || repo.Setting == nil { + return fallback + } + value, err := repo.Setting.Get(ctx, key) + if err != nil { + return fallback + } + n, err := strconv.Atoi(strings.TrimSpace(value)) + if err != nil || n <= 0 { + return fallback + } + return n +}