diff --git a/src/main/java/com/gunshi/project/ss/controller/ByPlanController.java b/src/main/java/com/gunshi/project/ss/controller/ByPlanController.java index a0bc7f1..b245a98 100644 --- a/src/main/java/com/gunshi/project/ss/controller/ByPlanController.java +++ b/src/main/java/com/gunshi/project/ss/controller/ByPlanController.java @@ -56,10 +56,13 @@ public class ByPlanController extends AbstractCommonFileController{ throw new IllegalArgumentException("该计划编号已存在"); } dto.setId(IdWorker.getId()); - SysUser sysUser = sysUserMapper.selectUserById(Long.valueOf(dto.getUserId())); - if(sysUser != null){ - dto.setUserName(sysUser.getNickName()); + if(dto.getUserId() != null){ + SysUser sysUser = sysUserMapper.selectUserById(Long.valueOf(dto.getUserId())); + if(sysUser != null){ + dto.setUserName(sysUser.getNickName()); + } } + List byPlanDetail = dto.getByPlanDetail(); if(byPlanDetail == null || byPlanDetail.isEmpty()){ throw new IllegalArgumentException("防止部位至少需要一条数据"); @@ -78,9 +81,11 @@ public class ByPlanController extends AbstractCommonFileController{ @Operation(summary = "修改") @PostMapping("/update") public R update(@Validated(Update.class) @RequestBody ByPlan dto) { - SysUser sysUser = sysUserMapper.selectUserById(Long.valueOf(dto.getUserId())); - if(sysUser != null){ - dto.setUserName(sysUser.getNickName()); + if(dto.getUserId() != null){ + SysUser sysUser = sysUserMapper.selectUserById(Long.valueOf(dto.getUserId())); + if(sysUser != null){ + dto.setUserName(sysUser.getNickName()); + } } boolean flag = byPlanService.update(dto);