From 68d7678a1fb2f2ce97d537e70ed85ef92eda480e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=AD=99=E5=B0=8F=E4=BA=91?= <sunpengchina85@gmail.com>
Date: Mon, 9 Mar 2026 13:30:52 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=8F=82=E6=95=B0=E9=BB=98?=
 =?UTF-8?q?=E8=AE=A4=E5=80=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../task/controller/TaskPlanController.java   | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/src/main/java/com/ruoyi/task/controller/TaskPlanController.java b/src/main/java/com/ruoyi/task/controller/TaskPlanController.java
index 848052a..a3c59e0 100644
--- a/src/main/java/com/ruoyi/task/controller/TaskPlanController.java
+++ b/src/main/java/com/ruoyi/task/controller/TaskPlanController.java
@@ -7,6 +7,10 @@ import com.ruoyi.common.core.web.page.TableDataInfo;
 import com.ruoyi.common.security.annotation.InnerAuth;
 import com.ruoyi.task.api.domain.TaskPlanDTO;
 import com.ruoyi.task.api.domain.TaskPlanQueryDTO;
+import com.ruoyi.task.api.enums.ExecuteTypeEnum;
+import com.ruoyi.task.api.enums.PlanTypeEnum;
+import com.ruoyi.task.api.enums.StatusEnum;
+import com.ruoyi.task.domain.model.TaskPlan;
 import com.ruoyi.task.service.api.ITaskPlanService;
 import com.ruoyi.task.controller.convert.TaskPlanControllerConvert;
 import com.ruoyi.task.controller.convert.TaskPlanQueryControllerConvert;
@@ -14,6 +18,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
+import java.util.Objects;
 
 /**
  * 任务计划Controller
@@ -28,12 +33,26 @@ public class TaskPlanController extends BaseController
     @Autowired
     private ITaskPlanService taskPlanService;
 
+    private void setDefaultPlan(TaskPlanDTO plan) {
+        if(Objects.isNull(plan)) {
+            return;
+        }
+        if(Objects.isNull(plan.getExecuteType())) {
+            plan.setExecuteType(ExecuteTypeEnum.ONCE);
+        }
+        if(Objects.isNull(plan.getStatus())) {
+            plan.setStatus(StatusEnum.PENDING);
+        }
+    }
+
     /**
      * 创建定时任务计划
      */
     @PostMapping("/timed")
     public R<Long> createTimedTaskPlan(@RequestBody TaskPlanDTO taskPlanDTO)
     {
+        taskPlanDTO.setPlanType(PlanTypeEnum.TIMED);
+        setDefaultPlan(taskPlanDTO);
         Long planId = taskPlanService.createTimedTaskPlan(TaskPlanControllerConvert.to(taskPlanDTO));
         return R.ok(planId);
     }
@@ -44,6 +63,8 @@ public class TaskPlanController extends BaseController
     @PostMapping("/cycle")
     public R<Long> createCycleTaskPlan(@RequestBody TaskPlanDTO taskPlanDTO)
     {
+        taskPlanDTO.setPlanType(PlanTypeEnum.CYCLE);
+        setDefaultPlan(taskPlanDTO);
         Long planId = taskPlanService.createCycleTaskPlan(TaskPlanControllerConvert.to(taskPlanDTO));
         return R.ok(planId);
     }