diff --git a/src/main/java/com/isu/gaswellwatch/dto/UserDTO.java b/src/main/java/com/isu/gaswellwatch/dto/UserDTO.java index 1d7db85..3cb23f0 100644 --- a/src/main/java/com/isu/gaswellwatch/dto/UserDTO.java +++ b/src/main/java/com/isu/gaswellwatch/dto/UserDTO.java @@ -29,5 +29,7 @@ public class UserDTO { private String isEnable; @NotNull(message = "角色不能为空") private List roles; + //部门ID + private Long departmentId; } diff --git a/src/main/java/com/isu/gaswellwatch/service/impl/UserServiceImpl.java b/src/main/java/com/isu/gaswellwatch/service/impl/UserServiceImpl.java index e54003b..8c29fff 100644 --- a/src/main/java/com/isu/gaswellwatch/service/impl/UserServiceImpl.java +++ b/src/main/java/com/isu/gaswellwatch/service/impl/UserServiceImpl.java @@ -132,6 +132,11 @@ public class UserServiceImpl extends ServiceImpl implements UserS save(user); //关联新增 addRelation(userDTO.getRoles(), user); + + if(userDTO.getDepartmentId()!=null){ + userDepartmentService.remove(new LambdaQueryWrapper().eq(UserDepartment::getUserId, user.getId())); + addDepartment(userDTO.getDepartmentId(), user); + } } @Override