国家地区接口优化

internal/service/fetcher/fetcher.go

@@ -61,15 +61,8 @@ func (f *Fetcher) Fetch(ctx context.Context, n int) error {
 	}
 
 	for _, mkt := range regions {
-		util.Logger.Info("Fetching images for region", zap.String("mkt", mkt))
-		// 调用两次 API 获取最多两周的数据
-		// 第一次 idx=0&n=8 (今天起往回数 8 张)
-		if err := f.fetchByMkt(ctx, mkt, 0, 8); err != nil {
-			util.Logger.Error("Failed to fetch images", zap.String("mkt", mkt), zap.Int("idx", 0), zap.Error(err))
-		}
-		// 第二次 idx=7&n=8 (7天前起往回数 8 张，与第一次有重叠，确保不漏)
-		if err := f.fetchByMkt(ctx, mkt, 7, 8); err != nil {
-			util.Logger.Error("Failed to fetch images", zap.String("mkt", mkt), zap.Int("idx", 7), zap.Error(err))
+		if err := f.FetchRegion(ctx, mkt); err != nil {
+			util.Logger.Error("Failed to fetch region images", zap.String("mkt", mkt), zap.Error(err))
 		}
 	}
 
@@ -77,6 +70,27 @@ func (f *Fetcher) Fetch(ctx context.Context, n int) error {
 	return nil
 }
 
+// FetchRegion 抓取指定地区的图片
+func (f *Fetcher) FetchRegion(ctx context.Context, mkt string) error {
+	if !util.IsValidRegion(mkt) {
+		util.Logger.Warn("Skipping fetch for invalid region", zap.String("mkt", mkt))
+		return fmt.Errorf("invalid region code: %s", mkt)
+	}
+	util.Logger.Info("Fetching images for region", zap.String("mkt", mkt))
+	// 调用两次 API 获取最多两周的数据
+	// 第一次 idx=0&n=8 (今天起往回数 8 张)
+	if err := f.fetchByMkt(ctx, mkt, 0, 8); err != nil {
+		util.Logger.Error("Failed to fetch images", zap.String("mkt", mkt), zap.Int("idx", 0), zap.Error(err))
+		return err
+	}
+	// 第二次 idx=7&n=8 (7天前起往回数 8 张，与第一次有重叠，确保不漏)
+	if err := f.fetchByMkt(ctx, mkt, 7, 8); err != nil {
+		util.Logger.Error("Failed to fetch images", zap.String("mkt", mkt), zap.Int("idx", 7), zap.Error(err))
+		// 第二次失败不一定返回错误，因为可能第一次已经拿到了
+	}
+	return nil
+}
+
 func (f *Fetcher) fetchByMkt(ctx context.Context, mkt string, idx int, n int) error {
 	url := fmt.Sprintf("%s?format=js&idx=%d&n=%d&uhd=1&mkt=%s", config.BingAPIBase, idx, n, mkt)
 	util.Logger.Debug("Requesting Bing API", zap.String("url", url))

internal/service/image/image_service.go

@@ -8,6 +8,7 @@ import (
 	"BingPaper/internal/config"
 	"BingPaper/internal/model"
 	"BingPaper/internal/repo"
+	"BingPaper/internal/service/fetcher"
 	"BingPaper/internal/storage"
 	"BingPaper/internal/util"
 
@@ -58,8 +59,19 @@ func GetTodayImage(mkt string) (*model.Image, error) {
 		tx = tx.Where("mkt = ?", mkt)
 	}
 	err := tx.Preload("Variants").First(&img).Error
+	if err != nil && mkt != "" && config.GetConfig().API.EnableOnDemandFetch && util.IsValidRegion(mkt) {
+		// 如果没找到，尝试按需抓取该地区
+		util.Logger.Info("Image not found in DB, attempting on-demand fetch", zap.String("mkt", mkt))
+		f := fetcher.NewFetcher()
+		_ = f.FetchRegion(context.Background(), mkt)
+
+		// 抓取后重新查询
+		tx = repo.DB.Where("date = ?", today).Where("mkt = ?", mkt)
+		err = tx.Preload("Variants").First(&img).Error
+	}
+
 	if err != nil {
-		// 如果今天没有，尝试获取最近的一张
+		// 如果今天还是没有，尝试获取最近的一张
 		tx = repo.DB.Order("date desc")
 		if mkt != "" {
 			tx = tx.Where("mkt = ?", mkt)
@@ -79,6 +91,22 @@ func GetTodayImage(mkt string) (*model.Image, error) {
 	return &img, err
 }
 
+func GetAllRegionsTodayImages() ([]model.Image, error) {
+	regions := config.GetConfig().Fetcher.Regions
+	if len(regions) == 0 {
+		regions = []string{config.GetConfig().GetDefaultMkt()}
+	}
+
+	var images []model.Image
+	for _, mkt := range regions {
+		img, err := GetTodayImage(mkt)
+		if err == nil {
+			images = append(images, *img)
+		}
+	}
+	return images, nil
+}
+
 func GetRandomImage(mkt string) (*model.Image, error) {
 	var img model.Image
 	// SQLite 使用 RANDOM(), MySQL/Postgres 使用 RANDOM() 或 RAND()
@@ -89,6 +117,17 @@ func GetRandomImage(mkt string) (*model.Image, error) {
 		tx = tx.Where("mkt = ?", mkt)
 	}
 	tx.Count(&count)
+	if count == 0 && mkt != "" && config.GetConfig().API.EnableOnDemandFetch && util.IsValidRegion(mkt) {
+		// 如果没找到，尝试按需抓取该地区
+		util.Logger.Info("No images found in DB for region, attempting on-demand fetch", zap.String("mkt", mkt))
+		f := fetcher.NewFetcher()
+		_ = f.FetchRegion(context.Background(), mkt)
+
+		// 抓取后重新计数
+		tx = repo.DB.Model(&model.Image{}).Where("mkt = ?", mkt)
+		tx.Count(&count)
+	}
+
 	if count == 0 {
 		return nil, fmt.Errorf("no images found")
 	}
@@ -127,6 +166,16 @@ func GetImageByDate(date string, mkt string) (*model.Image, error) {
 		tx = tx.Where("mkt = ?", mkt)
 	}
 	err := tx.Preload("Variants").First(&img).Error
+	if err != nil && mkt != "" && config.GetConfig().API.EnableOnDemandFetch && util.IsValidRegion(mkt) {
+		// 如果没找到，尝试按需抓取该地区
+		util.Logger.Info("Image not found in DB for date, attempting on-demand fetch", zap.String("mkt", mkt), zap.String("date", date))
+		f := fetcher.NewFetcher()
+		_ = f.FetchRegion(context.Background(), mkt)
+
+		// 抓取后重新查询
+		tx = repo.DB.Where("date = ?", date).Where("mkt = ?", mkt)
+		err = tx.Preload("Variants").First(&img).Error
+	}
 
 	// 兜底逻辑
 	if err != nil && mkt != "" && config.GetConfig().API.EnableMktFallback {