修改用户接口

src/main/java/com/isu/gaswellwatch/dto/UserDTO.java

@@ -29,5 +29,7 @@ public class UserDTO {
     private String isEnable;
     @NotNull(message = "角色不能为空")
     private List<RoleVO> roles;
+    //部门ID
+    private Long departmentId;
 
 }

src/main/java/com/isu/gaswellwatch/service/impl/UserServiceImpl.java

@@ -132,6 +132,11 @@ public class UserServiceImpl extends ServiceImpl<UserDao, User> implements UserS
         save(user);
         //关联新增
         addRelation(userDTO.getRoles(), user);
+
+        if(userDTO.getDepartmentId()!=null){
+            userDepartmentService.remove(new LambdaQueryWrapper<UserDepartment>().eq(UserDepartment::getUserId, user.getId()));
+            addDepartment(userDTO.getDepartmentId(), user);
+        }
     }
 
     @Override