split service boot helpers

2026-09-28 11:16:37 +08:00 · 2026-06-27 12:46:55 +08:00
parent 3b74a68c83
commit 05826e2fc7
3 changed files with 145 additions and 129 deletions
@@ -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 {
@@ -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
+}
@@ -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
+}