| package com.doumee.service.system.impl; | 
|   | 
| import com.baomidou.mybatisplus.core.conditions.Wrapper; | 
| import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; | 
| import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; | 
| import com.baomidou.mybatisplus.core.metadata.IPage; | 
| import com.baomidou.mybatisplus.extension.plugins.pagination.Page; | 
| import com.doumee.biz.system.SystemDictDataBiz; | 
| import com.doumee.core.constants.ResponseStatus; | 
| import com.doumee.core.exception.BusinessException; | 
| import com.doumee.core.model.LoginUserInfo; | 
| import com.doumee.core.model.PageData; | 
| import com.doumee.core.model.PageWrap; | 
| import com.doumee.core.utils.Constants; | 
| import com.doumee.core.utils.Utils; | 
| import com.doumee.dao.business.model.Company; | 
| import com.doumee.dao.business.model.CompanyDepartment; | 
| import com.doumee.dao.business.CompanyDepartmentMapper; | 
| import com.doumee.dao.business.model.Company; | 
| import com.doumee.dao.business.model.CompanyDepartment; | 
| import com.doumee.dao.business.model.Member; | 
| import com.doumee.dao.system.SystemUserJoinMapper; | 
| import com.doumee.dao.system.SystemUserMapper; | 
| import com.doumee.dao.system.dto.QuerySystemUserDTO; | 
| import com.doumee.dao.system.model.SystemDepartment; | 
| import com.doumee.dao.system.model.SystemUser; | 
| import com.doumee.dao.system.vo.SystemDepartmentListVO; | 
| import com.doumee.dao.system.vo.SystemUserListVO; | 
| import com.doumee.service.aware.DepartmentDataPermissionAware; | 
| import com.doumee.service.system.SystemDepartmentService; | 
| import com.doumee.service.system.SystemPositionService; | 
| import com.doumee.service.system.SystemRoleService; | 
| import com.doumee.service.system.SystemUserService; | 
| import com.github.pagehelper.PageHelper; | 
| import com.github.pagehelper.PageInfo; | 
| import com.github.yulichang.wrapper.MPJLambdaWrapper; | 
| import org.apache.commons.lang3.StringUtils; | 
| import org.apache.shiro.SecurityUtils; | 
| import org.springframework.beans.factory.annotation.Autowired; | 
| import org.springframework.stereotype.Service; | 
| import org.springframework.transaction.annotation.Transactional; | 
| import org.springframework.util.CollectionUtils; | 
|   | 
| import java.util.ArrayList; | 
| import java.util.List; | 
| import java.util.stream.Collectors; | 
|   | 
| /** | 
|  * 系统用户Service实现 | 
|  * @author Eva.Caesar Liu | 
|  * @date 2023/03/21 14:49 | 
|  */ | 
| @Service | 
| public class SystemUserServiceImpl implements SystemUserService { | 
|   | 
|     @Autowired | 
|     private SystemDictDataBiz systemDictDataBiz; | 
|   | 
|     @Autowired | 
|     private SystemUserMapper systemUserMapper; | 
|   | 
|     @Autowired | 
|     private SystemUserJoinMapper systemUserJoinMapper; | 
|   | 
|     @Autowired | 
|     private SystemRoleService systemRoleService; | 
|   | 
|     @Autowired | 
|     private SystemDepartmentService systemDepartmentService; | 
|   | 
|     @Autowired | 
|     private SystemPositionService systemPositionService; | 
|   | 
|     @Autowired | 
|     private DepartmentDataPermissionAware departmentDataPermissionAware; | 
|   | 
|     @Autowired | 
|     private CompanyDepartmentMapper companyDepartmentMapper; | 
|   | 
|     @Override | 
|     public Integer create(SystemUser systemUser) { | 
|         systemUserMapper.insert(systemUser); | 
|         return systemUser.getId(); | 
|     } | 
|   | 
|     @Override | 
|     public void deleteById(Integer id) { | 
|         SystemUser systemUser = new SystemUser(); | 
|         systemUser.setId(id); | 
|         systemUser.setDeleted(Boolean.TRUE); | 
|         this.updateById(systemUser); | 
|     } | 
|   | 
|     @Override | 
|     @Transactional | 
|     public void deleteByIdInBatch(List<Integer> ids) { | 
|         if (CollectionUtils.isEmpty(ids)) return; | 
|         for (Integer id : ids) { | 
|             this.deleteById(id); | 
|         } | 
|     } | 
|   | 
|     @Override | 
|     public void updateById(SystemUser systemUser) { | 
|         systemUserMapper.updateById(systemUser); | 
|     } | 
|   | 
|     @Override | 
|     @Transactional | 
|     public void updateByIdInBatch(List<SystemUser> systemUsers) { | 
|         if (CollectionUtils.isEmpty(systemUsers)) return; | 
|         for (SystemUser systemUser: systemUsers) { | 
|             this.updateById(systemUser); | 
|         } | 
|     } | 
|   | 
|     @Override | 
|     public SystemUser findById(Integer id) { | 
|         LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); | 
|   | 
|         MPJLambdaWrapper<SystemUser> queryWrapper = new MPJLambdaWrapper<>(); | 
|         queryWrapper.selectAll(SystemUser.class); | 
|         queryWrapper.eq(SystemUser::getId,id ); | 
|         queryWrapper.eq(SystemUser::getType,loginUserInfo.getType() ); | 
|         if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){ | 
|             //如果是企业账号 | 
|             queryWrapper.eq(SystemUser::getCompanyId,loginUserInfo.getCompanyId() ); | 
|         } | 
|         queryWrapper.orderByDesc(SystemUser::getCreateTime); | 
|         SystemUser  result = systemUserJoinMapper.selectJoinOne( SystemUser.class, queryWrapper); | 
|         return result; | 
|     } | 
|   | 
|     @Override | 
|     public SystemUser findOne(SystemUser systemUser) { | 
|         Wrapper<SystemUser> wrapper = new QueryWrapper<SystemUser>(systemUser).last("limit 1"); | 
|         return systemUserMapper.selectOne(wrapper); | 
|     } | 
|   | 
|     @Override | 
|     public List<SystemUser> findList(SystemUser systemUser) { | 
|         LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); | 
|         systemUser.setType(loginUserInfo.getType()); | 
|         if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){ | 
|             //如果是企业账号 | 
|             systemUser.setCompanyId(loginUserInfo.getCompanyId()); | 
|         } | 
|         Wrapper<SystemUser> wrapper = new QueryWrapper<>(systemUser); | 
|         return systemUserMapper.selectList(wrapper); | 
|     } | 
|   | 
|     @Override | 
|     public PageData<SystemUserListVO> findPage(PageWrap<QuerySystemUserDTO> pageWrap) { | 
|         // 根部门条件处理(需查询根部门下所有部门的用户) | 
|         if (pageWrap.getModel().getRootDeptId() != null) { | 
| //            pageWrap.getModel().setDepartmentIds(getDeptIds(pageWrap.getModel().getRootDeptId())); | 
|         } else { | 
| //            List<SystemDepartment> list = systemDepartmentService.findList(new QueryWrapper<>()); | 
| //            List<Integer> collect = list.stream().map(s -> s.getId()).collect(Collectors.toList()); | 
| //            pageWrap.getModel().setDepartmentIds(collect); | 
|         } | 
|         LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); | 
| //        pageWrap.getModel().setType(loginUserInfo.getType()); | 
|         if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){ | 
|             //如果是企业账号 | 
|             pageWrap.getModel().setCompanyId(loginUserInfo.getCompanyId()); | 
|         } | 
|         // 执行查询 | 
|         PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity()); | 
|         List<SystemUserListVO> userList = systemUserMapper.selectManageList(pageWrap.getModel(), pageWrap.getOrderByClause()); | 
|         String zhuboRoomUrl = systemDictDataBiz.queryByCode(Constants.SYSTEM, Constants.ZHUBO_ROOM_URL).getCode(); | 
|   | 
|         for (SystemUserListVO user : userList) { | 
|             // 查询用户角色列表 | 
|             user.setRoles(systemRoleService.findByUserId(user.getId())); | 
|             // 查询用户岗位列表 | 
| //            user.setPositions(systemPositionService.findByUserId(user.getId())); | 
|             user.setZhuboRoomUrl(zhuboRoomUrl); | 
|         } | 
|         return PageData.from(new PageInfo<>(userList)); | 
|     } | 
|     @Override | 
|     public PageData<SystemUserListVO> findPageCompany(PageWrap<QuerySystemUserDTO> pageWrap) { | 
|         // 根部门条件处理(需查询根部门下所有部门的用户) | 
|         IPage<Company> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity()); | 
|         Utils.MP.blankToNull(pageWrap.getModel()); | 
|         pageWrap.getModel().setType(Constants.UserType.COMPANY.getKey());//查询企业类型 | 
|         // 执行查询 | 
|         MPJLambdaWrapper<SystemUser> wrapper = new MPJLambdaWrapper<>(); | 
|         wrapper .selectAll(SystemUser.class) | 
|                 .select("(select count(distinct(a.company_id) from company_permisson a where a.user_id=t.id and a.ideleted=0 and a.type=1)",SystemUser::getAuthNum) | 
|                 .selectAs(Company::getName,SystemUser::getCompanyName) | 
|                 .selectAs(CompanyDepartment::getName,SystemUser::getDepartmentName) | 
|                 .leftJoin(Company.class,Company::getId,SystemUser::getCompanyId) | 
|                 .leftJoin(CompanyDepartment.class,CompanyDepartment::getId,SystemUser::getDepartmentId) | 
|                 .eq( pageWrap.getModel().getDepartmentId() !=null,SystemUser::getDepartmentId, pageWrap.getModel().getDepartmentId()) | 
|                 .eq( pageWrap.getModel().getStatus() !=null,SystemUser::getStatus, pageWrap.getModel().getStatus()) | 
|                 .like(StringUtils.isNotBlank(pageWrap.getModel().getMobile()),SystemUser::getMobile, pageWrap.getModel().getMobile()) | 
|                 .like(StringUtils.isNotBlank(pageWrap.getModel().getRealname()),SystemUser::getRealname, pageWrap.getModel().getRealname()) | 
|                 .like(StringUtils.isNotBlank(pageWrap.getModel().getUsername()),SystemUser::getUsername, pageWrap.getModel().getUsername()) | 
|                 .like(StringUtils.isNotBlank(pageWrap.getModel().getCompanyName()),SystemUser::getCompanyName, pageWrap.getModel().getCompanyName()); | 
|         IPage<SystemUserListVO> result =  systemUserMapper.selectJoinPage(page,SystemUserListVO.class,wrapper); | 
|         return  PageData.from(result); | 
|     } | 
|   | 
|     @Override | 
|     public long count(SystemUser systemUser) { | 
|         Wrapper<SystemUser> wrapper = new QueryWrapper<>(systemUser); | 
|         return systemUserMapper.selectCount(wrapper); | 
|     } | 
|   | 
|   | 
|     /** | 
|      * 获取用户权限内允许查询的部门ID | 
|      */ | 
|     private List<Integer> getDeptIds(Integer rootDeptId) { | 
|         List<Integer> departmentIds = systemDepartmentService.findChildren(rootDeptId); | 
|         departmentIds.add(rootDeptId); | 
|         return departmentIds; | 
|     } | 
|   | 
|     /** | 
|      * 获取用户权限内允许查询的部门ID | 
|      */ | 
|     private List<Integer> getAllowedDeptIds(Integer rootDeptId) { | 
|         List<SystemDepartmentListVO> allowedDepartments = departmentDataPermissionAware.execute(); | 
|         List<Integer> allowedDeptIds = new ArrayList<>(); | 
|         for (SystemDepartmentListVO listVO : allowedDepartments) { | 
|             injectIds(allowedDeptIds, listVO); | 
|         } | 
|         // 没有允许的部门 | 
|         if (allowedDeptIds.size() == 0) { | 
|             allowedDeptIds.add(-1); | 
|             return allowedDeptIds; | 
|         } | 
|         if (rootDeptId == null) { | 
|             return allowedDeptIds; | 
|         } | 
|         List<Integer> departmentIds = systemDepartmentService.findChildren(rootDeptId); | 
|         departmentIds.add(rootDeptId); | 
|         departmentIds.removeIf(deptId -> !allowedDeptIds.contains(deptId)); | 
|         return departmentIds; | 
|     } | 
|   | 
|     /** | 
|      * 递归注入用户权限内的部门ID | 
|      */ | 
|     private void injectIds (List<Integer> pool, SystemDepartmentListVO listVO) { | 
|         pool.add(listVO.getId()); | 
|         if (CollectionUtils.isEmpty(listVO.getChildren())) { | 
|             return; | 
|         } | 
|         for (SystemDepartmentListVO child : listVO.getChildren()) { | 
|             injectIds(pool, child); | 
|         } | 
|     } | 
|   | 
|   | 
|     @Override | 
|     public PageData<SystemUser> findAllList(PageWrap<SystemUser> pageWrap) { | 
|         LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); | 
|         pageWrap.getModel().setType(loginUserInfo.getType()); | 
|         if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){ | 
|             //如果是企业账号 | 
|             pageWrap.getModel().setCompanyId(loginUserInfo.getCompanyId()); | 
|         } | 
|         IPage<SystemUser> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity()); | 
|         MPJLambdaWrapper<SystemUser> queryWrapper = new MPJLambdaWrapper<>(); | 
|         Utils.MP.blankToNull(pageWrap.getModel()); | 
|         queryWrapper.selectAll(SystemUser.class); | 
|         queryWrapper.eq(SystemUser::getDeleted, Constants.ZERO); | 
|         queryWrapper.and(StringUtils.isNotBlank(pageWrap.getModel().getUsername()), | 
|                 ms->ms.like(SystemUser::getUsername,pageWrap.getModel().getUsername()) | 
|                     .or().like(SystemUser::getMobile,pageWrap.getModel().getUsername()) | 
|                 .or().like(SystemUser::getRealname,pageWrap.getModel().getUsername()) | 
|         ); | 
|         queryWrapper.orderByDesc(SystemUser::getCreateTime); | 
|         IPage<SystemUser> result = systemUserJoinMapper.selectJoinPage(page, SystemUser.class, queryWrapper); | 
|   | 
|         return PageData.from(result); | 
|     } | 
|   | 
|   | 
|     @Override | 
|     public List<SystemUser> findAllList(SystemUser pageWrap){ | 
|         LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); | 
|         pageWrap.setType(loginUserInfo.getType()); | 
|         if(!Constants.equalsInteger(loginUserInfo.getType(),Constants.ZERO)){ | 
|             //如果是企业账号 | 
|             pageWrap.setCompanyId(loginUserInfo.getCompanyId()); | 
|         } | 
|         MPJLambdaWrapper<SystemUser> queryWrapper = new MPJLambdaWrapper<>(); | 
|   | 
|         queryWrapper.selectAll(SystemUser.class); | 
|         queryWrapper.eq(SystemUser::getDeleted, Constants.ZERO) | 
|                     .eq(SystemUser::getType,pageWrap.getType()) | 
|                     .eq(SystemUser::getCompanyId,loginUserInfo.getCompanyId()) | 
|                ; | 
|         queryWrapper.orderByDesc(SystemUser::getCreateTime); | 
|         List<SystemUser> result = systemUserJoinMapper.selectJoinList(SystemUser.class, queryWrapper); | 
|   | 
|         return result; | 
|     } | 
|   | 
|     @Override | 
|     public void updateHead(SystemUser systemUser) { | 
|         SystemUser model  =systemUserJoinMapper.selectById(systemUser.getId()); | 
|         if(model ==null) { | 
|             throw  new BusinessException(ResponseStatus.DATA_EMPTY.getCode(),"对不起,人员信息不存在!"); | 
|         } | 
|         if(model.getDepartmentId()!=null) { | 
|             if(Constants.equalsInteger(systemUser.getHeadStatus(),Constants.ONE)){ | 
|                 //如果是取消主管 | 
|                 companyDepartmentMapper.update(null,new UpdateWrapper<CompanyDepartment>().lambda() | 
|                         .eq(CompanyDepartment::getId,model.getDepartmentId()) | 
|                         .set(CompanyDepartment::getHeadId,model.getId())); | 
|             }else{ | 
|                 //如果是设为主管 | 
|                 companyDepartmentMapper.update(null,new UpdateWrapper<CompanyDepartment>().lambda() | 
|                         .eq(CompanyDepartment::getId,model.getDepartmentId()) | 
|                         .set(CompanyDepartment::getHeadId,null) | 
|                 ); | 
|             } | 
|         } | 
|     } | 
|   | 
|   | 
|   | 
|   | 
| } |