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.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) { 
 | 
        // 根部门条件处理(需查询根部门下所有部门的用户) 
 | 
        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); 
 | 
        } 
 | 
        if (pageWrap.getModel().getCompanyId() == null) { 
 | 
            pageWrap.getModel().setCompanyId(-1); 
 | 
        } 
 | 
        // 执行查询 
 | 
        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 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) 
 | 
                ); 
 | 
            } 
 | 
        } 
 | 
    } 
 | 
  
 | 
  
 | 
  
 | 
  
 | 
} 
 |