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.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.doumee.biz.system.SystemDictDataBiz; import com.doumee.core.constants.Constants; import com.doumee.core.constants.ResponseStatus; import com.doumee.core.exception.BusinessException; import com.doumee.core.model.PageData; import com.doumee.core.model.PageWrap; import com.doumee.core.utils.Utils; import com.doumee.dao.system.SystemUserMapper; import com.doumee.dao.system.dto.QuerySystemUserDTO; 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.*; import com.github.pagehelper.PageHelper; import com.github.pagehelper.PageInfo; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.Lazy; 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.Objects; /** * 系统用户Service实现 * @author Eva.Caesar Liu * @date 2022/03/15 09:54 */ @Service public class SystemUserServiceImpl implements SystemUserService { @Autowired private SystemUserMapper systemUserMapper; @Autowired private SystemRoleService systemRoleService; @Autowired private SystemDepartmentService systemDepartmentService; @Autowired private SystemPositionService systemPositionService; @Autowired private DepartmentDataPermissionAware departmentDataPermissionAware; @Lazy @Autowired private SystemDictDataBiz systemDictDataBiz; @Autowired private SystemDictDataService systemDictDataService; @Override public void create(SystemUser systemUser) { systemUserMapper.insert(systemUser); } @Override public void deleteById(String id) { SystemUser systemUser = new SystemUser(); systemUser.setId(id); systemUser.setDeleted(Constants.ONE); this.updateById(systemUser); } @Override @Transactional public void deleteByIdInBatch(List ids) { if (CollectionUtils.isEmpty(ids)) return; for (String id : ids) { this.deleteById(id); } } @Override public void updateById(SystemUser systemUser) { systemUserMapper.updateById(systemUser); } @Override @Transactional public void updateByIdInBatch(List systemUsers) { if (CollectionUtils.isEmpty(systemUsers)) return; for (SystemUser systemUser: systemUsers) { this.updateById(systemUser); } } @Override public SystemUser findById(String id) { return systemUserMapper.selectById(id); } @Override public SystemUser findOne(SystemUser systemUser) { Wrapper wrapper = new QueryWrapper<>(systemUser).last("limit 1"); return systemUserMapper.selectOne(wrapper); } @Override public List findList(SystemUser systemUser) { Wrapper wrapper = new QueryWrapper<>(systemUser); return systemUserMapper.selectList(wrapper); } @Override public PageData findPage(PageWrap pageWrap) { // 根部门条件处理（需查询根部门下所有部门的用户） if (pageWrap.getModel().getRootDeptId() != null) { pageWrap.getModel().setDepartmentIds(getAllowedDeptIds(pageWrap.getModel().getRootDeptId())); } else { pageWrap.getModel().setDepartmentIds(getAllowedDeptIds(null)); } // 执行查询 PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity()); List userList = systemUserMapper.selectManageList(pageWrap.getModel(), pageWrap.getOrderByClause()); for (SystemUserListVO user : userList) { // 查询用户角色列表 user.setRoles(systemRoleService.findByUserId(user.getId())); // 查询用户岗位列表 user.setPositions(systemPositionService.findByUserId(user.getId())); } return PageData.from(new PageInfo<>(userList)); } @Override public PageData findPageNew(PageWrap pageWrap) { // 执行查询 PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity()); List userList = systemUserMapper.selectManageListNew(pageWrap.getModel(), pageWrap.getOrderByClause()); for (SystemUserListVO user : userList) { // 查询用户角色列表 user.setRoles(systemRoleService.findByUserId(user.getId())); // 查询用户岗位列表 user.setPositions(systemPositionService.findByUserId(user.getId())); String imgPath = systemDictDataBiz.queryByCode(Constants.OSS,Constants.RESOURCE_PATH).getCode()+systemDictDataBiz.queryByCode(Constants.FILES_DIR,Constants.AVATAR_FILE).getCode(); user.setAvatar(Constants.formatImgPath(user.getAvatar(),imgPath)); } return PageData.from(new PageInfo<>(userList)); } private PageData getUserListByParam(PageWrap pageWrap) { PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity()); List userList = systemUserMapper.selectManageListNew(pageWrap.getModel(), pageWrap.getOrderByClause()); for (SystemUserListVO user : userList) { // 查询用户角色列表 user.setRoles(systemRoleService.findByUserId(user.getId())); // 查询用户岗位列表 user.setPositions(systemPositionService.findByUserId(user.getId())); String imgPath = systemDictDataBiz.queryByCode(Constants.OSS,Constants.RESOURCE_PATH).getCode()+systemDictDataBiz.queryByCode(Constants.FILES_DIR,Constants.AVATAR_FILE).getCode(); user.setAvatar(Constants.formatImgPath(user.getAvatar(),imgPath)); } return PageData.from(new PageInfo<>(userList)); } @Override public PageData findUserPage(PageWrap pageWrap) { // 执行查询 PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity()); List userList = systemUserMapper.selectManageList(pageWrap.getModel(), pageWrap.getOrderByClause()); return PageData.from(new PageInfo<>(userList)); } @Override public PageData selectPage(PageWrap pageWrap) { IPage page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity()); QueryWrapper queryWrapper = new QueryWrapper<>(); Utils.MP.blankToNull(pageWrap.getModel()); if (pageWrap.getModel().getUsername() != null) { queryWrapper.lambda().like(SystemUser::getUsername, pageWrap.getModel().getUsername()); } for(PageWrap.SortData sortData: pageWrap.getSorts()) { if (sortData.getDirection().equalsIgnoreCase(PageWrap.DESC)) { queryWrapper.orderByDesc(sortData.getProperty()); } else { queryWrapper.orderByAsc(sortData.getProperty()); } } return PageData.from(systemUserMapper.selectPage(page, queryWrapper)); } @Override public long count(SystemUser systemUser) { Wrapper wrapper = new QueryWrapper<>(systemUser); return systemUserMapper.selectCount(wrapper); } @Override public void updateLoginDate(SystemUser u){ systemUserMapper.updateLoginDate(u); } @Override public void updateNumByModel(SystemUser u){ systemUserMapper.updateNumByModel(u); } @Override public void updateBySelective(SystemUser u){ systemUserMapper.updateBySelective(u); } /** * 获取用户权限内允许查询的部门ID */ private List getAllowedDeptIds(Integer rootDeptId) { List allowedDepartments = departmentDataPermissionAware.execute(); List 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 departmentIds = systemDepartmentService.findChildren(rootDeptId); departmentIds.add(rootDeptId); departmentIds.removeIf(deptId -> !allowedDeptIds.contains(deptId)); return departmentIds; } /** * 递归注入用户权限内的部门ID */ private void injectIds (List pool, SystemDepartmentListVO listVO) { pool.add(listVO.getId()); if (CollectionUtils.isEmpty(listVO.getChildren())) { return; } for (SystemDepartmentListVO child : listVO.getChildren()) { injectIds(pool, child); } } }