优化

ruoyi-demo/src/main/java/com/ruoyi/demo/controller/PlanController.java

@@ -3,6 +3,7 @@ package com.ruoyi.demo.controller;
 import cn.dev33.satoken.annotation.SaCheckPermission;
 import cn.dev33.satoken.annotation.SaIgnore;
 import cn.dev33.satoken.stp.StpUtil;
+import cn.hutool.core.date.DateUtil;
 import com.ruoyi.common.core.domain.R;
 import com.ruoyi.common.core.domain.entity.SysUser;
 import com.ruoyi.common.core.validate.AddGroup;
@@ -17,6 +18,7 @@ import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 
 import javax.validation.constraints.NotBlank;
+import java.util.Date;
 
 @RestController
 @RequestMapping("/plan")
@@ -62,15 +64,19 @@ public class PlanController {
     }
     /**
      * 修改广告计划时间段
-     * @param bo
+     * @param planId 广告计划id
+     * @param planStartTime 广告开始时间
+     * @param planEndTime 广告结束时间
      * @return
      */
     @SaCheckPermission("szzs:planManage:update")
     @PutMapping("/update/time")
-    public R planUpdateTime(PlanBo bo){
-        if (bo.getPlanId() == null)
+    public R planUpdateTime(@NotBlank(message = "广告id不能为空") String planId,
+                            String planStartTime,
+                            String planEndTime){
+        if (planId == null)
             return R.fail("广告id不能为空");
-        int i = planService.updateTime(bo);
+        int i = planService.updateTime(planId, DateUtil.parse(planStartTime), DateUtil.parse(planEndTime));
         if (i == 1)
             return R.ok("修改成功");
         if (i == -1)

ruoyi-demo/src/main/java/com/ruoyi/demo/service/PlanService.java

@@ -11,6 +11,7 @@ import com.ruoyi.demo.entity.bo.PlanBo;
 import com.ruoyi.demo.entity.bo.UpDownBo;
 import com.ruoyi.demo.entity.vo.*;
 
+import java.util.Date;
 import java.util.List;
 
 public interface PlanService {
@@ -29,7 +30,7 @@ public interface PlanService {
 
     PlanDetailVo getPlanDetail(String planId);
 
-    int updateTime(PlanBo bo);
+    int updateTime(String planId, Date planStartTime, Date planEndTime);
 
     List<PlanPushVo> getPlanPush(String equipmentId);
 

ruoyi-demo/src/main/java/com/ruoyi/demo/service/impl/PlanServiceImpl.java

@@ -230,32 +230,32 @@ public class PlanServiceImpl implements PlanService {
     }
 
     @Override
-    public int updateTime(PlanBo bo) {
+    public int updateTime(String planId, Date planStartTime, Date planEndTime) {
         Plan plan = planMapper.selectOne(new QueryWrapper<Plan>()
-            .eq("plan_id", bo.getPlanId())
+            .eq("plan_id", planId)
             .eq("deleted", "0"));
         if (plan == null)
             return -1; // 无效广告id
         // 判断所修改的时间是否存在和合理
-        if (bo.getStartTime() == null && bo.getEndTime() != null) { // 只修改结束时间
-            if (plan.getPlanStartTime().compareTo(bo.getEndTime()) > 0) { // 原开始时间大于现修改结束时间
+        if (planStartTime == null && planEndTime != null) { // 只修改结束时间
+            if (plan.getPlanStartTime().compareTo(planEndTime) > 0) { // 原开始时间大于现修改结束时间
                 // 调换开始时间与结束时间位置
                 plan.setPlanEndTime(plan.getPlanStartTime());
-                plan.setPlanStartTime(bo.getEndTime());
+                plan.setPlanStartTime(planEndTime);
             } else {
-                plan.setPlanEndTime(bo.getEndTime());
+                plan.setPlanEndTime(planEndTime);
             }
-        } else if (bo.getStartTime() != null && bo.getEndTime() == null) { // 只修改开始时间
-            if (plan.getPlanEndTime().compareTo(bo.getStartTime()) < 0) { // 原结束时间小于现修改开始时间
+        } else if (planStartTime != null && planEndTime == null) { // 只修改开始时间
+            if (plan.getPlanEndTime().compareTo(planStartTime) < 0) { // 原结束时间小于现修改开始时间
                 // 调换开始时间与结束时间位置
                 plan.setPlanStartTime(plan.getPlanEndTime());
-                plan.setPlanEndTime(bo.getStartTime());
+                plan.setPlanEndTime(planStartTime);
             } else {
-                plan.setPlanStartTime(bo.getStartTime());
+                plan.setPlanStartTime(planStartTime);
             }
-        } else if (bo.getStartTime() != null && bo.getEndTime() != null) { // 两者皆修改
-            plan.setPlanStartTime(bo.getStartTime());
-            plan.setPlanEndTime(bo.getEndTime());
+        } else if (planStartTime != null && planEndTime != null) { // 两者皆修改
+            plan.setPlanStartTime(planStartTime);
+            plan.setPlanEndTime(planEndTime);
         } else
             return 1; // 两者皆无
         return planMapper.updateById(plan);