You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
301 lines
12 KiB
Java
301 lines
12 KiB
Java
3 years ago
|
package com.engine.organization.service.impl;
|
||
|
|
|
||
|
|
import com.api.hrm.bean.TreeNode;
|
||
|
|
import com.engine.core.impl.Service;
|
||
|
|
import com.engine.organization.entity.comp.bo.CompBO;
|
||
|
|
import com.engine.organization.entity.comp.po.CompPO;
|
||
|
|
import com.engine.organization.entity.department.bo.DepartmentBO;
|
||
|
|
import com.engine.organization.entity.department.po.DepartmentPO;
|
||
|
|
import com.engine.organization.entity.searchtree.SearchTree;
|
||
|
|
import com.engine.organization.entity.searchtree.SearchTreeParams;
|
||
|
|
import com.engine.organization.mapper.comp.CompMapper;
|
||
|
|
import com.engine.organization.mapper.department.DepartmentMapper;
|
||
|
|
import com.engine.organization.service.JobService;
|
||
|
|
import com.engine.organization.util.db.MapperProxyFactory;
|
||
|
|
import org.apache.commons.collections4.CollectionUtils;
|
||
|
|
import org.apache.commons.lang3.StringUtils;
|
||
|
|
import weaver.conn.RecordSet;
|
||
|
|
import weaver.general.StringUtil;
|
||
|
|
import weaver.general.Util;
|
||
|
|
|
||
|
|
import java.util.*;
|
||
|
|
import java.util.stream.Collectors;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @description: TODO
|
||
|
|
* @author:dxfeng
|
||
|
|
* @createTime: 2022/05/26
|
||
|
|
* @version: 1.0
|
||
|
|
*/
|
||
|
|
public class JobServiceImpl extends Service implements JobService {
|
||
|
|
|
||
|
|
private static final String TYPE_GROUP = "0";
|
||
|
|
private static final String TYPE_COMP = "1";
|
||
|
|
private static final String TYPE_DEPT = "2";
|
||
|
|
|
||
|
|
private CompMapper getCompMapper() {
|
||
|
|
return MapperProxyFactory.getProxy(CompMapper.class);
|
||
|
|
}
|
||
|
|
|
||
|
|
private DepartmentMapper getDepartmentMapper() {
|
||
|
|
return MapperProxyFactory.getProxy(DepartmentMapper.class);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public Map<String, Object> getSearchTree(SearchTreeParams params) {
|
||
|
|
Map<String, Object> dataMap = new HashMap<>();
|
||
|
|
SearchTree topGroup = getTopGroup();
|
||
|
|
// 集团
|
||
|
|
List<SearchTree> companyList = new ArrayList<>();
|
||
|
|
companyList.add(topGroup);
|
||
|
|
String keyword = params.getKeyword();
|
||
|
|
String type = Util.null2String(params.getType());
|
||
|
|
String id = params.getId();
|
||
|
|
// 没有搜索条件查询
|
||
|
|
if (StringUtil.isEmpty(keyword)) {
|
||
|
|
// 无条件，初始化树结构
|
||
|
|
if (StringUtil.isEmpty(type)) {
|
||
|
|
dataMap.put("companys", companyList);
|
||
|
|
SearchTree rootCompany = getTopGroup();
|
||
|
|
rootCompany.setSubs(getCompany("", ""));
|
||
|
|
rootCompany.setIsParent(CollectionUtils.isNotEmpty(rootCompany.getSubs()));
|
||
|
|
Map<String, Object> rootCompanyMap = new HashMap<>();
|
||
|
|
rootCompanyMap.put("rootCompany", rootCompany);
|
||
|
|
dataMap.put("datas", rootCompanyMap);
|
||
|
|
} else {
|
||
|
|
// 根据ID查询
|
||
|
|
dataMap.put("datas", getCompany(id, type));
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if (StringUtil.isEmpty(type)) {
|
||
|
|
List<TreeNode> collect = getFilterCompany(id, type, keyword);
|
||
|
|
|
||
|
|
SearchTree rootCompany = getTopGroup();
|
||
|
|
rootCompany.setSubs(collect);
|
||
|
|
rootCompany.setIsParent(CollectionUtils.isNotEmpty(rootCompany.getSubs()));
|
||
|
|
Map<String, Object> rootCompanyMap = new HashMap<>();
|
||
|
|
rootCompanyMap.put("rootCompany", rootCompany);
|
||
|
|
dataMap.put("datas", rootCompanyMap);
|
||
|
|
|
||
|
|
dataMap.put("companys", companyList);
|
||
|
|
} else {
|
||
|
|
dataMap.put("datas", getFilterCompany(id, type, keyword));
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return dataMap;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 添加查询元素的父级元素
|
||
|
|
*
|
||
|
|
* @param departmentPO
|
||
|
|
* @param builderDeparts
|
||
|
|
*/
|
||
|
|
private void buildParentDepts(DepartmentPO departmentPO, Set<DepartmentPO> builderDeparts) {
|
||
|
|
builderDeparts.add(departmentPO);
|
||
|
|
if (isTop(departmentPO.getParentDept())) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
DepartmentPO parentDept = getDepartmentMapper().getDeptById(departmentPO.getParentDept());
|
||
|
|
if (null != parentDept) {
|
||
|
|
buildParentDepts(parentDept, builderDeparts);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 添加查询元素的父级元素
|
||
|
|
*
|
||
|
|
* @param compPO
|
||
|
|
* @param builderComps
|
||
|
|
*/
|
||
|
|
private void buildParentComps(CompPO compPO, Set<CompPO> builderComps) {
|
||
|
|
builderComps.add(compPO);
|
||
|
|
if (isTop(compPO.getParentCompany())) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
CompPO parentComp = getCompMapper().listById(compPO.getParentCompany());
|
||
|
|
if (null != parentComp) {
|
||
|
|
buildParentComps(parentComp, builderComps);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 获取集团
|
||
|
|
*
|
||
|
|
* @return
|
||
|
|
*/
|
||
|
|
private SearchTree getTopGroup() {
|
||
|
|
RecordSet rs = new RecordSet();
|
||
|
|
String sql = "select * from HrmCompany ";
|
||
|
|
rs.executeQuery(sql);
|
||
|
|
SearchTree groupTree = new SearchTree();
|
||
|
|
groupTree.setCanClick(false);
|
||
|
|
groupTree.setCanceled(false);
|
||
|
|
groupTree.setCompanyid("1");
|
||
|
|
groupTree.setIcon("icon-coms-LargeArea");
|
||
|
|
groupTree.setId("0");
|
||
|
|
groupTree.setIsVirtual("0");
|
||
|
|
while (rs.next()) {
|
||
|
|
groupTree.setName(rs.getString("COMPANYNAME"));
|
||
|
|
}
|
||
|
|
groupTree.setSelected(false);
|
||
|
|
groupTree.setType(TYPE_GROUP);
|
||
|
|
return groupTree;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 获取分部数据
|
||
|
|
*
|
||
|
|
* @param id
|
||
|
|
* @return
|
||
|
|
*/
|
||
|
|
private List<TreeNode> getCompany(String id, String type) {
|
||
|
|
List<TreeNode> compSearchTree = new ArrayList<>();
|
||
|
|
List<CompPO> compPOS;
|
||
|
|
if (StringUtils.isAllEmpty(id, type)) {
|
||
|
|
compPOS = getCompMapper().listParent();
|
||
|
|
compSearchTree.addAll(CompBO.buildListToSearchTree(compPOS));
|
||
|
|
} else {
|
||
|
|
if (TYPE_COMP.equals(type)) {
|
||
|
|
compPOS = getCompMapper().listChildByPID(id);
|
||
|
|
// 获取该分部下的部门
|
||
|
|
List<DepartmentPO> deptListByCompId = getDepartmentMapper().getDeptListByCompId(Integer.parseInt(id)).stream().filter(item -> null == item.getParentDept()).collect(Collectors.toList());
|
||
|
|
compSearchTree.addAll(CompBO.buildListToSearchTree(compPOS));
|
||
|
|
compSearchTree.addAll(DepartmentBO.buildListToSearchTree(deptListByCompId));
|
||
|
|
} else if (TYPE_DEPT.equals(type)) {
|
||
|
|
List<DepartmentPO> deptListByCompId = getDepartmentMapper().getDeptListByPId(Long.parseLong(id));
|
||
|
|
compSearchTree.addAll(DepartmentBO.buildListToSearchTree(deptListByCompId));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return compSearchTree;
|
||
|
|
}
|
||
|
|
|
||
|
|
private List<TreeNode> getFilterCompany(String id, String type, String keyword) {
|
||
|
|
List<TreeNode> compSearchTree = new ArrayList<>();
|
||
|
|
if (StringUtils.isAllEmpty(id, type)) {
|
||
|
|
// 查询部门信息
|
||
|
|
List<DepartmentPO> filterDeparts = getDepartmentMapper().listByFilter(DepartmentPO.builder().deptName(keyword).build());
|
||
|
|
Set<DepartmentPO> builderDeparts = new HashSet<>();
|
||
|
|
for (DepartmentPO departmentPO : filterDeparts) {
|
||
|
|
buildParentDepts(departmentPO, builderDeparts);
|
||
|
|
}
|
||
|
|
List<SearchTree> deptTrees = builderTreeMode(DepartmentBO.buildSetToSearchTree(builderDeparts));
|
||
|
|
|
||
|
|
// 查询分部信息
|
||
|
|
List<CompPO> filterComps = getCompMapper().listByFilter(CompPO.builder().compName(keyword).build());
|
||
|
|
Set<CompPO> builderComps = new HashSet<>();
|
||
|
|
for (CompPO compPO : filterComps) {
|
||
|
|
buildParentComps(compPO, builderComps);
|
||
|
|
}
|
||
|
|
List<TreeNode> compTrees = builderTreeMode(CompBO.buildSetToSearchTree(builderComps), deptTrees);
|
||
|
|
|
||
|
|
// 排序，设置是否为叶子节点
|
||
|
|
List<TreeNode> collect = compTrees.stream().peek(item ->
|
||
|
|
item.setIsParent(CollectionUtils.isNotEmpty(item.getSubs()))
|
||
|
|
).sorted(Comparator.comparing(item -> Integer.parseInt(item.getId()))).collect(Collectors.toList());
|
||
|
|
if (CollectionUtils.isNotEmpty(collect)) {
|
||
|
|
compSearchTree.addAll(collect);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if (TYPE_COMP.equals(type)) { // 查询部门信息
|
||
|
|
List<DepartmentPO> filterDeparts = getDepartmentMapper().listByFilter(DepartmentPO.builder().deptName(keyword).parentComp(Long.parseLong(id)).build());
|
||
|
|
Set<DepartmentPO> builderDeparts = new HashSet<>();
|
||
|
|
for (DepartmentPO departmentPO : filterDeparts) {
|
||
|
|
buildParentDepts(departmentPO, builderDeparts);
|
||
|
|
}
|
||
|
|
List<SearchTree> deptTrees = builderTreeMode(DepartmentBO.buildSetToSearchTree(builderDeparts));
|
||
|
|
|
||
|
|
// 查询分部信息
|
||
|
|
List<CompPO> filterComps = getCompMapper().listByFilter(CompPO.builder().compName(keyword).parentCompany(Long.parseLong(id)).build());
|
||
|
|
Set<CompPO> builderComps = new HashSet<>();
|
||
|
|
for (CompPO compPO : filterComps) {
|
||
|
|
buildParentComps(compPO, builderComps);
|
||
|
|
}
|
||
|
|
List<TreeNode> compTrees = builderTreeMode(CompBO.buildSetToSearchTree(builderComps), deptTrees);
|
||
|
|
|
||
|
|
// 排序，设置是否为叶子节点
|
||
|
|
List<TreeNode> collect = compTrees.stream().peek(item ->
|
||
|
|
item.setIsParent(CollectionUtils.isNotEmpty(item.getSubs()))
|
||
|
|
).sorted(Comparator.comparing(item -> Integer.parseInt(item.getId()))).collect(Collectors.toList());
|
||
|
|
if (CollectionUtils.isNotEmpty(collect)) {
|
||
|
|
compSearchTree.addAll(collect);
|
||
|
|
}
|
||
|
|
} else if (TYPE_DEPT.equals(type)) {
|
||
|
|
List<DepartmentPO> filterDeparts = getDepartmentMapper().listByFilter(DepartmentPO.builder().deptName(keyword).parentDept(Long.parseLong(id)).build());
|
||
|
|
Set<DepartmentPO> builderDeparts = new HashSet<>();
|
||
|
|
for (DepartmentPO departmentPO : filterDeparts) {
|
||
|
|
buildParentDepts(departmentPO, builderDeparts);
|
||
|
|
}
|
||
|
|
List<SearchTree> deptTrees = builderTreeMode(DepartmentBO.buildSetToSearchTree(builderDeparts));
|
||
|
|
// 排序，设置是否为叶子节点
|
||
|
|
List<TreeNode> collect = deptTrees.stream().peek(item ->
|
||
|
|
item.setIsParent(CollectionUtils.isNotEmpty(item.getSubs()))
|
||
|
|
).sorted(Comparator.comparing(item -> Integer.parseInt(item.getId()))).collect(Collectors.toList());
|
||
|
|
if (CollectionUtils.isNotEmpty(collect)) {
|
||
|
|
compSearchTree.addAll(collect);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return compSearchTree;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 判断是为顶层数据
|
||
|
|
*
|
||
|
|
* @param pid
|
||
|
|
* @return
|
||
|
|
*/
|
||
|
|
private boolean isTop(Long pid) {
|
||
|
|
return null == pid;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 判断是为顶层数据
|
||
|
|
*
|
||
|
|
* @param pid
|
||
|
|
* @return
|
||
|
|
*/
|
||
|
|
private boolean isTop(String pid) {
|
||
|
|
return StringUtil.isEmpty(pid) || "0".equals(pid);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 处理树层级
|
||
|
|
*
|
||
|
|
* @param treeList
|
||
|
|
* @return
|
||
|
|
*/
|
||
|
|
private List<SearchTree> builderTreeMode(List<SearchTree> treeList) {
|
||
|
|
Map<String, List<TreeNode>> collects = treeList.stream().collect(Collectors.groupingBy(TreeNode::getPid));
|
||
|
|
return treeList.stream().peek(e -> e.setSubs(collects.get(e.getId()))).filter(item -> isTop(item.getPid())).collect(Collectors.toList());
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 组合分部、部门层级关系
|
||
|
|
*
|
||
|
|
* @param treeList
|
||
|
|
* @param deptTrees
|
||
|
|
* @return
|
||
|
|
*/
|
||
|
|
private List<TreeNode> builderTreeMode(List<TreeNode> treeList, List<SearchTree> deptTrees) {
|
||
|
|
Map<String, List<TreeNode>> parentMap = treeList.stream().collect(Collectors.groupingBy(TreeNode::getPid));
|
||
|
|
Map<String, List<SearchTree>> childMap = deptTrees.stream().collect(Collectors.groupingBy(SearchTree::getParentComp));
|
||
|
|
boolean isAdd = !childMap.isEmpty();
|
||
|
|
return treeList.stream().peek(e -> {
|
||
|
|
List<TreeNode> treeNodes = new ArrayList<>();
|
||
|
|
List<TreeNode> nodes = parentMap.get(e.getId());
|
||
|
|
if (CollectionUtils.isNotEmpty(nodes)) {
|
||
|
|
treeNodes.addAll(nodes);
|
||
|
|
}
|
||
|
|
if (isAdd && CollectionUtils.isNotEmpty(childMap.get(e.getId()))) {
|
||
|
|
treeNodes.addAll(childMap.get(e.getId()));
|
||
|
|
}
|
||
|
|
e.setSubs(treeNodes);
|
||
|
|
}).filter(item -> isTop(item.getPid())).collect(Collectors.toList());
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|