com.didiglobal.logi.security.service.impl.DeptServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of logi-security-spring-boot-starter Show documentation
logi-common/logi-security
The newest version!
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);
    }
}