fix: unable to reset pending task status

crawlab-team · Jul 16, 2023 · f271a2f · f271a2f
1 parent 54f4a52
commit f271a2f
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/task/scheduler/service.go b/task/scheduler/service.go
@@ -158,7 +158,12 @@ func (svc *Service) SetInterval(interval time.Duration) {
 func (svc *Service) initTaskStatus() {
 	// set status of running tasks as TaskStatusAbnormal
 	runningTasks, err := svc.modelSvc.GetTaskList(bson.M{
-		"status": constants.TaskStatusRunning,
+		"status": bson.M{
+			"$in": []string{
+				constants.TaskStatusPending,
+				constants.TaskStatusRunning,
+			},
+		},
 	}, nil)
 	if err != nil {
 		if err == mongo2.ErrNoDocuments {
@@ -173,6 +178,9 @@ func (svc *Service) initTaskStatus() {
 			}
 		}(&t)
 	}
+	if err := svc.modelSvc.GetBaseService(interfaces.ModelIdTaskQueue).DeleteList(nil); err != nil {
+		return
+	}
 }
 
 func (svc *Service) isMasterNode(t *models.Task) (ok bool, err error) {