|
@@ -53,25 +53,27 @@ public class MtCompanyUserService extends ServiceImpl<MtCompanyUserMapper, MtCom
|
|
|
mtCompanyUser.setJob(mtCompanyUserRequest.getJob());
|
|
|
this.updateById(mtCompanyUser);
|
|
|
//保存用户角色关联信息
|
|
|
- //1.删除原有的关联关系
|
|
|
Long roleId = mtCompanyUserRequest.getRoleId();
|
|
|
Long userId = mtCompanyUser.getUserId();
|
|
|
- UserRole userRole = userRoleService.getOne(new QueryWrapper<UserRole>()
|
|
|
- .eq("role_id", roleId)
|
|
|
- .eq("user_id", userId)
|
|
|
- );
|
|
|
- if (userRole != null) {
|
|
|
- boolean deleteUserRoleFlag = userRoleService.removeById(userRole.getId());
|
|
|
- if (!deleteUserRoleFlag) {
|
|
|
- return RestResponse.fail(ApiConstants.RESULT_ERROR, "删除原有关联关系失败");
|
|
|
+ if(roleId != null){
|
|
|
+ UserRole userRole = userRoleService.getOne(new QueryWrapper<UserRole>()
|
|
|
+ .eq("role_id", roleId)
|
|
|
+ .eq("user_id", userId)
|
|
|
+ );
|
|
|
+ if (userRole != null) {
|
|
|
+ //1.删除原有的关联关系
|
|
|
+ boolean deleteUserRoleFlag = userRoleService.removeById(userRole.getId());
|
|
|
+ if (!deleteUserRoleFlag) {
|
|
|
+ return RestResponse.fail(ApiConstants.RESULT_ERROR, "删除原有关联关系失败");
|
|
|
+ }
|
|
|
}
|
|
|
+ //保存新的关联关系
|
|
|
+ userRole = new UserRole();
|
|
|
+ userRole.setCompanyId(mtCompanyUser.getMtCompanyId());
|
|
|
+ userRole.setRoleId(roleId);
|
|
|
+ userRole.setUserId(userId);
|
|
|
+ userRoleService.save(userRole);
|
|
|
}
|
|
|
- //2.保存新的关联关系
|
|
|
- userRole = new UserRole();
|
|
|
- userRole.setCompanyId(mtCompanyUser.getMtCompanyId());
|
|
|
- userRole.setRoleId(roleId);
|
|
|
- userRole.setUserId(userId);
|
|
|
- userRoleService.save(userRole);
|
|
|
return RestResponse.success(null, ApiConstants.RESULT_SUCCESS, "分配角色成功");
|
|
|
}
|
|
|
|
|
@@ -175,6 +177,9 @@ public class MtCompanyUserService extends ServiceImpl<MtCompanyUserMapper, MtCom
|
|
|
//组装公司用户信息
|
|
|
for (MtCompanyUser mtCompanyUser : mtCompanyUserList) {
|
|
|
MtCompanyUserResponse mtCompanyUserResponse = new MtCompanyUserResponse();
|
|
|
+ mtCompanyUserResponse.setId(mtCompanyUser.getId());
|
|
|
+ //基础信息
|
|
|
+ mtCompanyUserResponse.setRepairFlag(mtCompanyUser.getRepairFlag());
|
|
|
UserResponse userResponse = userIdToUserResponse.get(mtCompanyUser.getUserId());
|
|
|
//设置用户角色信息
|
|
|
mtCompanyUserResponse.setJob(mtCompanyUser.getJob());
|