com.didiglobal.logi.security.service.impl.DeptServiceImpl Maven / Gradle / Ivy
package com.didiglobal.logi.security.service.impl;
import com.didiglobal.logi.security.common.dto.dept.DeptDTO;
import com.didiglobal.logi.security.common.entity.dept.Dept;
import com.didiglobal.logi.security.common.vo.dept.DeptBriefVO;
import com.didiglobal.logi.security.common.enums.ResultCode;
import com.didiglobal.logi.security.common.vo.dept.DeptTreeVO;
import com.didiglobal.logi.security.dao.DeptDao;
import com.didiglobal.logi.security.exception.LogiSecurityException;
import com.didiglobal.logi.security.service.DeptService;
import com.didiglobal.logi.security.util.CopyBeanUtil;
import com.didiglobal.logi.security.util.MathUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
import java.util.*;
/**
* @author cjm
*/
@Service("logiSecurityDeptServiceImpl")
public class DeptServiceImpl implements DeptService {
@Autowired
private DeptDao deptDao;
@Override
public DeptTreeVO buildDeptTree() {
// 获取全部部门并根据level小到大排序
List deptList = deptDao.selectAllAndAscOrderByLevel();
// 创建一个虚拟根节点
DeptTreeVO root = DeptTreeVO
.builder().leaf(false).id(0).childList(new ArrayList<>()).build();
// 转成树
Map parentMap = new HashMap<>();
parentMap.put(0, root);
for(Dept dept : deptList) {
DeptTreeVO deptTreeVO = CopyBeanUtil.copy(dept, DeptTreeVO.class);
if(deptTreeVO.getLeaf() != null && !deptTreeVO.getLeaf()) {
deptTreeVO.setChildList(new ArrayList<>());
}
DeptTreeVO parent = parentMap.get(dept.getParentId());
if (parent == null) {
// 如果parent为null，则需要查看下数据库部门表的数据是否有误
// 1.可能出现了本来该是父节点的节点（有其他子节点的parent为它），但该节点parent为其他子节点的情况（数据异常）
// 2.也可能是level填写错了（因为前面根据level大小排序）
throw new LogiSecurityException(ResultCode.DEPT_DATA_ERROR);
}
parent.getChildList().add(deptTreeVO);
parentMap.put(deptTreeVO.getId(), deptTreeVO);
}
return root;
}
@Override
public List getDeptBriefListByChildId(Integer deptId) {
// 提前获取所有部门
Map deptMap = getAllDeptMap();
return getDeptBriefListFromDeptMapByChildId(deptMap, deptId);
}
@Override
public List getDeptIdListByParentId(Integer deptId) {
if(deptId == null) {
// 如果为null，则获取全部部门的id
return deptDao.selectAllDeptIdList();
}
List deptList = deptDao.selectAllAndAscOrderByLevel();
HashSet deptIdSet = new HashSet<>();
deptIdSet.add(deptId);
// 遍历所有dept
for(Dept dept : deptList) {
if(deptIdSet.contains(dept.getParentId())) {
deptIdSet.add(dept.getId());
}
}
return new ArrayList<>(deptIdSet);
}
@Override
public List getDeptIdListByParentIdAndDeptName(Integer deptId, String deptName) {
// 获取所有子部门的id
List deptIdList = getDeptIdListByParentId(deptId);
if(deptIdList == null) {
return new ArrayList<>();
}
if(deptIdList.isEmpty() || StringUtils.isEmpty(deptName)) {
return deptIdList;
}
Set deptIdSet = new HashSet<>(deptIdList);
List result = new ArrayList<>();
// 获取和deptName相似的部门id
deptIdList = deptDao.selectIdListByLikeDeptName(deptName);
for(Integer id : deptIdList) {
if(deptIdSet.contains(id)) {
result.add(id);
}
}
return result;
}
@Override
public Map getAllDeptMap() {
List deptList = deptDao.selectAllAndAscOrderByLevel();
Map map = new HashMap<>();
for(Dept dept : deptList) {
map.put(dept.getId(), dept);
}
return map;
}
@Override
public List getDeptBriefListFromDeptMapByChildId(Map deptMap, Integer deptId) {
if(deptId == null || deptId == 0 || CollectionUtils.isEmpty(deptMap)) {
return new ArrayList<>();
}
LinkedList deptBriefVOList = new LinkedList<>();
while (deptId != null && deptId != 0) {
Dept dept = deptMap.get(deptId);
if(dept == null) {
break;
}
deptBriefVOList.addFirst(CopyBeanUtil.copy(dept, DeptBriefVO.class));
deptId = dept.getParentId();
}
return deptBriefVOList;
}
@Override
public void saveDept(List deptDTOList) {
if(CollectionUtils.isEmpty(deptDTOList)) {
return;
}
List deptList = new ArrayList<>();
Map deptDTOMap = new HashMap<>();
Queue queue = new LinkedList<>();
// 创建虚拟根节点
DeptDTO deptDTO = new DeptDTO();
deptDTO.setChildDeptDTOList(deptDTOList);
queue.offer(deptDTO);
int level = 0;
while(!queue.isEmpty()) {
int size = queue.size();
while(size-- > 0) {
DeptDTO dto = queue.poll();
if(dto == null) {
continue;
}
Dept dept = CopyBeanUtil.copy(dto, Dept.class);
// 设置层级
dept.setLevel(level);
// 设置父节点id
dept.setParentId(deptDTOMap.get(dto));
// 没有子节点就是叶子节点
dept.setLeaf(CollectionUtils.isEmpty(dto.getChildDeptDTOList()));
dept.setId(0);
if(level > 0) {
// 设置id
dept.setId((int) getDeptId());
deptList.add(dept);
}
for(DeptDTO child : dto.getChildDeptDTOList()) {
deptDTOMap.put(child, dept.getId());
queue.offer(child);
}
}
level++;
}
deptDao.insertBatch(deptList);
}
private long getDeptId() {
return System.currentTimeMillis() % 1000 * (long) Math.pow(10, 5) + MathUtil.getRandomNumber(5);
}
}