🐛 cookie更新

internal/framework/schedule/cpdaily_handler.go

@@ -90,16 +90,17 @@ func (c *CpdailyTaskHandler) ProcessTask(ctx context.Context, task *asynq.Task)
 
 	// 查询学校详情
 	schoolDetail := school.GetSchoolDetail(ctx, targetTask.Edges.CpUser.SchoolID)
-	restyClient := login.GetClient(ctx, targetTask.Edges.CpUser.Cookie)
+	cookie := targetTask.Edges.CpUser.Cookie
+	restyClient := login.GetClient(ctx, cookie)
 	if !login.CookieEnabled(ctx, schoolDetail.WecloudUrl, restyClient) {
 		loginExecutor := login.GetCpdailyLoginBySchoolInfo(ctx, &schoolDetail)
 		loginExecutor.Login(ctx, &schoolDetail, targetTask.Edges.CpUser.Username, targetTask.Edges.CpUser.Password)
-		cookie := loginExecutor.GetCookieString()
+		cookie = loginExecutor.GetCookieString()
 		db.Global.CpUser.UpdateOne(targetTask.Edges.CpUser).SetCookie(cookie).ExecX(ctx)
 		restyClient = login.GetClient(ctx, cookie)
 	}
 	// 获取任务执行handler
-	taskExecutor := task2.GetCpdailyTaskService(ctx, targetTask.TaskType, targetTask.Edges.CpUser.Cookie, targetTask.Edges.CpUser.SchoolID)
+	taskExecutor := task2.GetCpdailyTaskService(ctx, targetTask.TaskType, cookie, targetTask.Edges.CpUser.SchoolID)
 
 	cpdailyRealTask := taskExecutor.QueryFormByTitle(ctx, targetTask.Title)
 

internal/framework/schedule/scanner.go

@@ -87,7 +87,7 @@ func (ts *TaskScanner) scanCpdailyTask() {
 	// 只查询30分钟以内的任务
 	minExecuteTime := currentTime.Add(-30 * time.Minute).Unix()
 
-	allTasks := db.Global.Debug().CpTask.Query().
+	allTasks := db.Global.CpTask.Query().
 		Where(
 			cptask.NextExecuteTimeLTE(currentTime.Unix()),
 			cptask.NextExecuteTimeNotNil(),

internal/service/cpdaily/cptask.go

@@ -178,9 +178,6 @@ func (v CpTaskServiceV1) PostExecute(ctx context.Context, req *v1.PostCpTaskExec
 		utils.OkOrPanic(errors2.NewBusinessError(400, "执行次数应不低于原有次数"))
 	}
 
-	// 记录原始任务状态，用于判断是否需要立即扫描
-	originalStatus := cpTask.TaskStatus
-
 	//扣积分
 	basePoint := decimal.NewFromInt(cst2.BasePointCpdaily)
 	if user.Edges.Agent != nil {
@@ -190,35 +187,20 @@ func (v CpTaskServiceV1) PostExecute(ctx context.Context, req *v1.PostCpTaskExec
 	balance.BalanceService.ChangeUserBalance(ctx, user.ID, needPoint.Neg())
 	// 更新任务
 	e := db2.WithTx(ctx, func(ctx context.Context, tx *ent.Tx) {
-		cpTask := db2.Global.CpTask.UpdateOne(cpTask).
+		update := db2.Global.CpTask.UpdateOne(cpTask).
 			SetStep(cptask.StepComplete).
 			SetFrequency(req.Frequency).
 			SetTaskStatus(cptask.TaskStatusNotStart).
 			SetNotificationType(req.NotificationType).
 			SetExecuteHour(req.ExecuteHour).
-			SetExecuteMinute(req.ExecuteMinute).
-			SaveX(ctx)
-
-		// 计算下次执行时间
-		var nextExecuteAt int64
-
-		// 只有在原始状态为未开始时，才考虑5分钟窗口立即扫描
-		if originalStatus == cptask.TaskStatusNotStart {
-			currentTime := time.Now().In(cst2.TimeZoneShangHai)
-			executeTime := time.Date(currentTime.Year(), currentTime.Month(), currentTime.Day(), req.ExecuteHour, req.ExecuteMinute, 0, 0, cst2.TimeZoneShangHai)
-			timeDiff := executeTime.Sub(currentTime)
-
-			if timeDiff >= -5*time.Minute && timeDiff <= 5*time.Minute {
-				// 执行时间在当前时间前后5分钟内，设置为当前时间，让扫描器立即扫描到
-				nextExecuteAt = currentTime.Unix()
-			} else {
-				// 否则使用正常的计算逻辑
-				nextExecuteAt = utils.GetCpdailyNextExecuteTime(cpTask)
-			}
-		} else {
-			// 其他状态（暂停、失败等），使用正常的计算逻辑
-			nextExecuteAt = utils.GetCpdailyNextExecuteTime(cpTask)
+			SetExecuteMinute(req.ExecuteMinute)
+		if cpTask.Step == cptask.StepExecute {
+			update.SetTaskStatus(cptask.TaskStatusNotStart)
 		}
+		cpTask := update.SaveX(ctx)
+
+		// 其他状态（暂停、失败等），使用正常的计算逻辑
+		nextExecuteAt := utils.GetCpdailyNextExecuteTime(cpTask)
 
 		db2.Global.CpTask.UpdateOne(cpTask).SetNextExecuteTime(nextExecuteAt).ExecX(ctx)
 	}, &sql.TxOptions{})
@@ -283,6 +265,7 @@ func (v CpTaskServiceV1) ExecuteNow(ctx context.Context, req *v1.ExecuteNowReq)
 			asynq.Unique(5*time.Second),
 			asynq.MaxRetry(0),
 			asynq.Timeout(10*time.Minute),
+			asynq.Deadline(time.Now().Add(10*time.Minute)),
 			asynq.Queue(cst2.TaskQueueCpdaily),
 		)
 		// 更新未开始的任务状态为队列中