add film recover crontab

server/plugin/SystemInit/SpiderInit.go

@@ -65,16 +65,28 @@ func CollectCrontabInit() {
 				}
 				// 将定时任务Id记录到Task中
 				task.Cid = cid
+			case 2:
+				cid, err := spider.AddFilmRecoverCron(task.Spec)
+				// 如果任务添加失败则直接返回错误信息
+				if err != nil {
+					log.Println("自动清理失败采集记录定时任务添加失败: ", err.Error())
+					continue
+				}
+				// 将定时任务Id记录到Task中
+				task.Cid = cid
 			}
 			system.UpdateFilmTask(task)
 		}
 	} else {
-		// 如果系统中不存在任何定时任务信息, 则添加默认的定时任务
-		// 1. 添加一条默认任务, 定时更新所有已启用站点的影片信息
-		// 生成任务信息
+		/*
+			如果系统中不存在任何定时任务信息, 则添加默认的定时任务
+			1. 添加一条默认任务, 定时更新所有已启用站点的影片信息
+			2.  添加一条默认任务, 定时处理采集失败的记录
+			3.生成任务信息
+		*/
 		task := system.FilmCollectTask{Id: util.GenerateSalt(), Time: config.DefaultUpdateTime, Spec: config.DefaultUpdateSpec,
 			Model: 0, State: false, Remark: "每20分钟执行一次已启用站点数据的自动更新"}
-		// 添加一条定时任务
+		// 添加一条定时任务-影片定时更新
 		cid, err := spider.AddAutoUpdateCron(task.Id, task.Spec)
 		// 如果任务添加失败则直接返回错误信息
 		if err != nil {
@@ -85,6 +97,21 @@ func CollectCrontabInit() {
 		task.Cid = cid
 		// 如果没有异常则将当前定时任务信息记录到redis中
 		system.SaveFilmTask(task)
+
+		// 添加一条定时任务-定期处理失败请求
+		recoverTask := system.FilmCollectTask{Id: util.GenerateSalt(), Time: 0, Spec: config.EveryWeekSpec,
+			Model: 2, State: false, Remark: "每周日凌晨4点清理一次采集失败的采集记录"}
+		// 添加一条定时任务-影片定时更新
+		cid, err = spider.AddFilmRecoverCron(recoverTask.Spec)
+		// 如果任务添加失败则直接返回错误信息
+		if err != nil {
+			log.Println("失败采集恢复定时任务添加失败: ", err.Error())
+			return
+		}
+		// 将定时任务Id记录到Task中
+		recoverTask.Cid = cid
+		// 如果没有异常则将当前定时任务信息记录到redis中
+		system.SaveFilmTask(recoverTask)
 	}
 
 	// 完成初始化后启动 Cron

server/plugin/db/mysql.go

@@ -3,6 +3,7 @@ package db
 import (
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
 	"gorm.io/gorm/schema"
 	"server/config"
 )
@@ -24,7 +25,7 @@ func InitMysql() (err error) {
 			SingularTable: true, //是否使用 结构体名称作为表名 (关闭自动变复数)
 			//NameReplacer:  strings.NewReplacer("spider_", ""), // 替表名和字段中的 Me 为 空
 		},
-		//Logger: logger.Default.LogMode(logger.Info), //设置日志级别为Info
+		Logger: logger.Default.LogMode(logger.Info), //设置日志级别为Info
 	})
 	return
 }

server/plugin/spider/Spider.go

@@ -319,6 +319,39 @@ func SingleRecoverSpider(fr *system.FailureRecord) {
 		// 其余范围,暂不处理
 		break
 	}
+}
+
+// FullRecoverSpider 扫描记录表中的失败记录, 并进行处理 (用于定时任务定期处理失败采集)
+func FullRecoverSpider() {
+	/*
+		获取待处理的记录数据
+		1. 采集时长 > 168h (一周,7天)  状态-1 待处理, | 只获取满足条件的最早的待处理记录
+		2. 采集时长 > 4320h (半年,180天)  状态-1 待处理, | 获取满足条件的所有数据
+	*/
+	list := system.PendingRecord()
+
+	// 遍历记录信息切片, 针对不同时长进行不同处理
+	for _, fr := range list {
+		switch {
+		case fr.Hour > 0 && fr.Hour < 4320:
+			// 将此记录之后的所有同类采集记录变更为已重试
+			system.ChangeRecord(&fr, 0)
+			// 如果采集的内容是 7~15 天之内更新的内容,则采集此记录之后的所有更新内容
+			// 获取采集参数h, 采集时长变更为 原采集时长 + 采集记录距现在的时长
+			h := fr.Hour + int(math.Ceil(time.Since(fr.CreatedAt).Hours()))
+			// 对当前所有已启用的站点 更新最新 h 小时的内容
+			AutoCollect(h)
+		case fr.Hour < 0, fr.Hour > 4320:
+			// 将此记录状态修改为已重试
+			system.ChangeRecord(&fr, 0)
+			// 如果采集的是 最近180天内更新的内容 或全部内容, 则只对当前一条记录进行二次采集
+			s := system.FindCollectSourceById(fr.OriginId)
+			collectFilm(s, fr.Hour, fr.PageNumber)
+		default:
+			// 其余范围,暂不处理
+			break
+		}
+	}
 
 }
 

server/plugin/spider/SpiderCron.go

@@ -60,6 +60,19 @@ func AddAutoUpdateCron(id, spec string) (cron.EntryID, error) {
 	})
 }
 
+// AddFilmRecoverCron 失败采集记录处理
+func AddFilmRecoverCron(spec string) (cron.EntryID, error) {
+	// 校验 spec 表达式的有效性
+	if err := ValidSpec(spec); err != nil {
+		return -99, errors.New(fmt.Sprint("定时任务添加失败,Cron表达式校验失败: ", err.Error()))
+	}
+	return CronCollect.AddFunc(spec, func() {
+		// 执行失败采集记录恢复
+		FullRecoverSpider()
+		log.Println("执行一次失败采集恢复任务")
+	})
+}
+
 // RemoveCron 删除定时任务
 func RemoveCron(id cron.EntryID) {
 	// 通过定时任务EntryID移出对应的定时任务