优化调整

@@ -910,10 +910,14 @@ func (srv TaskService) ReplyTaskStage(transactionContext application.Transaction
 		"transactionContext": transactionContext,
 	})
 	for _, val := range param {
-		taskData, err := taskRepo.FindOne(map[string]interface{}{"id": val.TaskId})
+		_, taskDataList, err := taskRepo.Find(map[string]interface{}{"id": val.TaskId})
 		if err != nil {
 			return application.ThrowError(application.INTERNAL_SERVER_ERROR, "找不到任务数据"+err.Error())
 		}
+		if len(taskDataList) == 0 {
+			continue
+		}
+		taskData := taskDataList[0]
 		_, stageList, err := taskStageRepo.Find(map[string]interface{}{"taskId": val.TaskId})
 		if err != nil {
 			return application.ThrowError(application.INTERNAL_SERVER_ERROR, "找不到任务里程碑数据"+err.Error())