update: php code

utils/global_scheduler.go

@@ -18,10 +18,10 @@ var (
 )
 
 // GetGlobalScheduler 获取全局调度器实例（单例模式）
-func GetGlobalScheduler(hotDramaRepo repo.HotDramaRepository) *GlobalScheduler {
+func GetGlobalScheduler(hotDramaRepo repo.HotDramaRepository, readyResourceRepo repo.ReadyResourceRepository, resourceRepo repo.ResourceRepository, systemConfigRepo repo.SystemConfigRepository) *GlobalScheduler {
 	once.Do(func() {
 		globalScheduler = &GlobalScheduler{
-			scheduler: NewScheduler(hotDramaRepo),
+			scheduler: NewScheduler(hotDramaRepo, readyResourceRepo, resourceRepo, systemConfigRepo),
 		}
 	})
 	return globalScheduler
@@ -67,6 +67,48 @@ func (gs *GlobalScheduler) GetHotDramaNames() ([]string, error) {
 	return gs.scheduler.GetHotDramaNames()
 }
 
+// StartReadyResourceScheduler 启动待处理资源自动处理任务
+func (gs *GlobalScheduler) StartReadyResourceScheduler() {
+	gs.mutex.Lock()
+	defer gs.mutex.Unlock()
+
+	if gs.scheduler.IsReadyResourceRunning() {
+		log.Println("待处理资源自动处理任务已在运行中")
+		return
+	}
+
+	gs.scheduler.StartReadyResourceScheduler()
+	log.Println("全局调度器已启动待处理资源自动处理任务")
+}
+
+// StopReadyResourceScheduler 停止待处理资源自动处理任务
+func (gs *GlobalScheduler) StopReadyResourceScheduler() {
+	gs.mutex.Lock()
+	defer gs.mutex.Unlock()
+
+	if !gs.scheduler.IsReadyResourceRunning() {
+		log.Println("待处理资源自动处理任务未在运行")
+		return
+	}
+
+	gs.scheduler.StopReadyResourceScheduler()
+	log.Println("全局调度器已停止待处理资源自动处理任务")
+}
+
+// IsReadyResourceRunning 检查待处理资源自动处理任务是否在运行
+func (gs *GlobalScheduler) IsReadyResourceRunning() bool {
+	gs.mutex.RLock()
+	defer gs.mutex.RUnlock()
+	return gs.scheduler.IsReadyResourceRunning()
+}
+
+// ProcessReadyResources 手动触发待处理资源处理
+func (gs *GlobalScheduler) ProcessReadyResources() {
+	gs.mutex.Lock()
+	defer gs.mutex.Unlock()
+	gs.scheduler.processReadyResources()
+}
+
 // UpdateSchedulerStatus 根据系统配置更新调度器状态
 func (gs *GlobalScheduler) UpdateSchedulerStatus(autoFetchHotDramaEnabled bool) {
 	gs.mutex.Lock()

utils/scheduler.go

@@ -1,28 +1,43 @@
 package utils
 
 import (
+	"fmt"
 	"log"
 	"res_db/db/entity"
 	"res_db/db/repo"
+	"res_db/utils"
 	"strings"
+	"sync"
 	"time"
 )
 
 // Scheduler 定时任务管理器
 type Scheduler struct {
-	doubanService *DoubanService
-	hotDramaRepo  repo.HotDramaRepository
-	stopChan      chan bool
-	isRunning     bool
+	doubanService        *DoubanService
+	hotDramaRepo         repo.HotDramaRepository
+	readyResourceRepo    repo.ReadyResourceRepository
+	resourceRepo         repo.ResourceRepository
+	systemConfigRepo     repo.SystemConfigRepository
+	stopChan             chan bool
+	isRunning            bool
+	readyResourceRunning bool
+	processingMutex      sync.Mutex // 防止ready_resource任务重叠执行
+	hotDramaMutex        sync.Mutex // 防止热播剧任务重叠执行
 }
 
 // NewScheduler 创建新的定时任务管理器
-func NewScheduler(hotDramaRepo repo.HotDramaRepository) *Scheduler {
+func NewScheduler(hotDramaRepo repo.HotDramaRepository, readyResourceRepo repo.ReadyResourceRepository, resourceRepo repo.ResourceRepository, systemConfigRepo repo.SystemConfigRepository) *Scheduler {
 	return &Scheduler{
-		doubanService: NewDoubanService(),
-		hotDramaRepo:  hotDramaRepo,
-		stopChan:      make(chan bool),
-		isRunning:     false,
+		doubanService:        NewDoubanService(),
+		hotDramaRepo:         hotDramaRepo,
+		readyResourceRepo:    readyResourceRepo,
+		resourceRepo:         resourceRepo,
+		systemConfigRepo:     systemConfigRepo,
+		stopChan:             make(chan bool),
+		isRunning:            false,
+		readyResourceRunning: false,
+		processingMutex:      sync.Mutex{},
+		hotDramaMutex:        sync.Mutex{},
 	}
 }
 
@@ -46,7 +61,15 @@ func (s *Scheduler) StartHotDramaScheduler() {
 		for {
 			select {
 			case <-ticker.C:
-				s.fetchHotDramaData()
+				// 使用TryLock防止任务重叠执行
+				if s.hotDramaMutex.TryLock() {
+					go func() {
+						defer s.hotDramaMutex.Unlock()
+						s.fetchHotDramaData()
+					}()
+				} else {
+					log.Println("上一次热播剧任务还在执行中，跳过本次执行")
+				}
 			case <-s.stopChan:
 				log.Println("停止热播剧定时任务")
 				return
@@ -173,3 +196,200 @@ func (s *Scheduler) IsRunning() bool {
 func (s *Scheduler) GetHotDramaNames() ([]string, error) {
 	return s.doubanService.FetchHotDramaNames()
 }
+
+// StartReadyResourceScheduler 启动待处理资源自动处理任务
+func (s *Scheduler) StartReadyResourceScheduler() {
+	if s.readyResourceRunning {
+		log.Println("待处理资源自动处理任务已在运行中")
+		return
+	}
+
+	s.readyResourceRunning = true
+	log.Println("启动待处理资源自动处理任务")
+
+	go func() {
+		ticker := time.NewTicker(5 * time.Minute) // 每5分钟检查一次
+		defer ticker.Stop()
+
+		// 立即执行一次
+		s.processReadyResources()
+
+		for {
+			select {
+			case <-ticker.C:
+				// 使用TryLock防止任务重叠执行
+				if s.processingMutex.TryLock() {
+					go func() {
+						defer s.processingMutex.Unlock()
+						s.processReadyResources()
+					}()
+				} else {
+					log.Println("上一次待处理资源任务还在执行中，跳过本次执行")
+				}
+			case <-s.stopChan:
+				log.Println("停止待处理资源自动处理任务")
+				return
+			}
+		}
+	}()
+}
+
+// StopReadyResourceScheduler 停止待处理资源自动处理任务
+func (s *Scheduler) StopReadyResourceScheduler() {
+	if !s.readyResourceRunning {
+		log.Println("待处理资源自动处理任务未在运行")
+		return
+	}
+
+	s.stopChan <- true
+	s.readyResourceRunning = false
+	log.Println("已发送停止信号给待处理资源自动处理任务")
+}
+
+// processReadyResources 处理待处理资源
+func (s *Scheduler) processReadyResources() {
+	log.Println("开始处理待处理资源...")
+
+	// 检查系统配置，确认是否启用自动处理
+	config, err := s.systemConfigRepo.GetOrCreateDefault()
+	if err != nil {
+		log.Printf("获取系统配置失败: %v", err)
+		return
+	}
+
+	if !config.AutoProcessReadyResources {
+		log.Println("自动处理待处理资源功能已禁用")
+		return
+	}
+
+	// 获取所有待处理资源
+	readyResources, err := s.readyResourceRepo.FindAll()
+	if err != nil {
+		log.Printf("获取待处理资源失败: %v", err)
+		return
+	}
+
+	if len(readyResources) == 0 {
+		log.Println("没有待处理的资源")
+		return
+	}
+
+	log.Printf("找到 %d 个待处理资源，开始处理...", len(readyResources))
+
+	processedCount := 0
+	for _, readyResource := range readyResources {
+
+		//readyResource.URL 是 查重
+		exits, err := s.resourceRepo.FindExists(readyResource.URL)
+		if err != nil {
+			log.Printf("查重失败: %v", err)
+			continue
+		}
+		if exits {
+			log.Printf("资源已存在: %s", readyResource.URL)
+			s.readyResourceRepo.Delete(readyResource.ID)
+			continue
+		}
+
+		if err := s.convertReadyResourceToResource(readyResource); err != nil {
+			log.Printf("处理资源失败 (ID: %d): %v", readyResource.ID, err)
+		}
+		s.readyResourceRepo.Delete(readyResource.ID)
+		processedCount++
+		log.Printf("成功处理资源: %s", readyResource.URL)
+	}
+
+	log.Printf("待处理资源处理完成，共处理 %d 个资源", processedCount)
+}
+
+// convertReadyResourceToResource 将待处理资源转换为正式资源
+func (s *Scheduler) convertReadyResourceToResource(readyResource entity.ReadyResource) error {
+	// 这里可以添加你的自定义逻辑
+	// 例如：URL解析、标题提取、分类判断等
+
+	fmt.Println("run task")
+	shareID, serviceType := utils.ExtractShareId(readyResource.URL)
+	fmt.Println(shareID, serviceType)
+	if serviceType == utils.NotFound {
+		// 不支持的链接地址
+		return nil
+	}
+
+	// 非夸克地址， 需要有标题， 否则直接放弃
+	if serviceType != utils.Quark {
+		// 阿里云盘
+		checkResult, _ := utils.CheckURL(readyResource.URL)
+		if !checkResult.Status {
+			return nil
+		}
+		// 可以添加
+	}
+
+	if serviceType == utils.Quark {
+		// 通过api 获取到夸克数据的详情
+	}
+
+	// 处理Title字段（可能为nil）
+	// var title string
+	// if readyResource.Title != nil {
+	// 	title = *readyResource.Title
+	// }
+
+	// // 基础转换逻辑
+	// resource := &entity.Resource{
+	// 	Title:       title,
+	// 	Description: readyResource.Description,
+	// 	URL:         readyResource.URL,
+	// 	// 根据URL判断平台
+	// 	PanID:    s.determinePanID(readyResource.URL),
+	// 	IsValid:  true,
+	// 	IsPublic: true,
+	// }
+
+	// // 如果有分类信息，尝试查找或创建分类
+	// if readyResource.Category != "" {
+	// 	categoryID, err := s.getOrCreateCategory(readyResource.Category)
+	// 	if err == nil {
+	// 		resource.CategoryID = &categoryID
+	// 	}
+	// }
+
+	// // 创建资源
+	// return s.resourceRepo.Create(resource)
+
+	// 临时返回nil，等待你添加自定义逻辑
+	return nil
+}
+
+// determinePanID 根据URL确定平台ID
+func (s *Scheduler) determinePanID(url string) *uint {
+	url = strings.ToLower(url)
+
+	// 这里可以根据你的平台配置来判断
+	// 示例逻辑，你需要根据实际情况调整
+	if strings.Contains(url, "pan.baidu.com") {
+		panID := uint(1) // 百度网盘
+		return &panID
+	} else if strings.Contains(url, "www.aliyundrive.com") {
+		panID := uint(2) // 阿里云盘
+		return &panID
+	} else if strings.Contains(url, "pan.quark.cn") {
+		panID := uint(3) // 夸克网盘
+		return &panID
+	}
+
+	return nil
+}
+
+// getOrCreateCategory 获取或创建分类
+func (s *Scheduler) getOrCreateCategory(categoryName string) (uint, error) {
+	// 这里需要实现分类的查找和创建逻辑
+	// 由于没有CategoryRepository的注入，这里先返回0
+	// 你可以根据需要添加CategoryRepository的依赖
+	return 0, nil
+}
+
+// IsReadyResourceRunning 检查待处理资源自动处理任务是否在运行
+func (s *Scheduler) IsReadyResourceRunning() bool {
+	return s.readyResourceRunning
+}