diff --git a/dxhy-admin/src/main/java/com/dxhy/admin/sys/service/impl/SysUserServiceImpl.java b/dxhy-admin/src/main/java/com/dxhy/admin/sys/service/impl/SysUserServiceImpl.java index 119dc70e..10bb0abd 100644 --- a/dxhy-admin/src/main/java/com/dxhy/admin/sys/service/impl/SysUserServiceImpl.java +++ b/dxhy-admin/src/main/java/com/dxhy/admin/sys/service/impl/SysUserServiceImpl.java @@ -8,6 +8,7 @@ import javax.annotation.Resource; import com.dxhy.common.util.StringHelper; import org.apache.commons.lang.RandomStringUtils; import org.apache.commons.lang.StringUtils; +import org.jetbrains.annotations.NotNull; import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; @@ -351,36 +352,7 @@ public class SysUserServiceImpl extends ServiceImpl i user.setUserOrg(userOrgEntity); user.setUserBusiness(businessList); user.setCompanyCodes(companyEntitys); -// user.setUserModels(this.getUserCompanyList(userEntity.getCompany())); - List deptEntities = sysUserOrgService.queryOrgCodes(userId); - List userModels = new ArrayList<>(); - for (SysDeptEntity entity : deptEntities){ - List models = this.getUserAuthList(String.valueOf(entity.getDeptId())); - for(UserModel model : models) { - if(userModels.stream().noneMatch(e -> model.getUserId().equals(e.getUserId()) && model.getDeptId().equals(e.getDeptId()))) { - userModels.add(model); - - if(!model.getDeptId().equals(model.getDId())){ - UserModel userModel = new UserModel(); - userModel.setUserId(model.getUserId()); - userModel.setLoginname(model.getLoginname()); - userModel.setUsername(model.getUsername()); - userModel.setDeptId(model.getDId()); - userModels.add(userModel); - } - } - } - - //每个deptId手动增加一个admin - UserModel userModel = new UserModel(); - userModel.setUserId("10086"); - userModel.setLoginname("admin"); - userModel.setUsername("admin"); - userModel.setDeptId(String.valueOf(entity.getDeptId())); - userModels.add(userModel); - } - - log.info("userModels={}",userModels); + List userModels = setUserModels(userEntity, userId); user.setUserModels(userModels); // 刷新菜单缓存处理 this.setMetnuRedis(userEntity.getUserId()); @@ -392,6 +364,50 @@ public class SysUserServiceImpl extends ServiceImpl i return user; } + @NotNull + private List setUserModels(SysUserEntity userEntity, long userId) { + // user.setUserModels(this.getUserCompanyList(userEntity.getCompany())); + List deptEntities = sysUserOrgService.queryOrgCodes(userId); + List userModels = new ArrayList<>(); + for (SysDeptEntity entity : deptEntities){ + List models = this.getUserAuthList(String.valueOf(entity.getDeptId())); + for(UserModel model : models) { + if(userModels.stream().noneMatch(e -> model.getUserId().equals(e.getUserId()) && model.getDeptId().equals(e.getDeptId()))) { + userModels.add(model); + + if(!model.getDeptId().equals(model.getDId())){ + UserModel userModel = new UserModel(); + userModel.setUserId(model.getUserId()); + userModel.setLoginname(model.getLoginname()); + userModel.setUsername(model.getUsername()); + userModel.setDeptId(model.getDId()); + userModels.add(userModel); + } + } + } + + //每个deptId手动增加一个admin + UserModel userModel = new UserModel(); + userModel.setUserId("10086"); + userModel.setLoginname("admin"); + userModel.setUsername("admin"); + userModel.setDeptId(String.valueOf(entity.getDeptId())); + userModels.add(userModel); + } + if(userModels.stream().noneMatch(e -> userEntity.getUserId().equals(e.getUserId()) && userEntity.getDeptId().equals(e.getDeptId()))) { + //当前登录用户手动增加一个admin + UserModel userModel = new UserModel(); + userModel.setUserId("10086"); + userModel.setLoginname("admin"); + userModel.setUsername("admin"); + userModel.setDeptId(String.valueOf(userEntity.getDeptId())); + userModels.add(userModel); + } + + log.info("userModels={}",userModels); + return userModels; + } + private List getUserCompanyList(String company) { DynamicContextHolder.push(DbConstant.BASICS_READ); return baseMapper.getUserCompanyList(company);