forked from github/dataease
217 lines
8.2 KiB
Java
217 lines
8.2 KiB
Java
package io.dataease.service.sys;
|
||
|
||
import io.dataease.base.domain.SysDept;
|
||
import io.dataease.base.domain.SysDeptExample;
|
||
import io.dataease.base.mapper.SysDeptMapper;
|
||
import io.dataease.base.mapper.ext.ExtDeptMapper;
|
||
import io.dataease.base.mapper.ext.query.GridExample;
|
||
import io.dataease.commons.utils.BeanUtils;
|
||
import io.dataease.commons.utils.CommonBeanFactory;
|
||
import io.dataease.controller.sys.base.BaseGridRequest;
|
||
import io.dataease.controller.sys.base.ConditionEntity;
|
||
import io.dataease.controller.sys.request.DeptCreateRequest;
|
||
import io.dataease.controller.sys.request.DeptDeleteRequest;
|
||
import io.dataease.controller.sys.request.DeptStatusRequest;
|
||
import io.dataease.controller.sys.request.SimpleTreeNode;
|
||
import io.dataease.controller.sys.response.DeptTreeNode;
|
||
import org.apache.commons.collections.CollectionUtils;
|
||
import org.apache.commons.lang3.ObjectUtils;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.stereotype.Service;
|
||
import org.springframework.transaction.annotation.Transactional;
|
||
import javax.annotation.Resource;
|
||
import java.util.*;
|
||
import java.util.stream.Collectors;
|
||
|
||
@Service
|
||
public class DeptService {
|
||
|
||
private final static Integer DEPT_ROOT_LEVEL = 0;
|
||
private final static Integer DEFAULT_SUBCOUNT = 0;
|
||
public final static Long DEPT_ROOT_PID = 0L;
|
||
|
||
@Autowired(required = false)
|
||
private SysDeptMapper sysDeptMapper;
|
||
|
||
|
||
@Autowired(required = false)
|
||
private ExtDeptMapper extDeptMapper;
|
||
|
||
public List<SysDept> nodesByPid(Long pid){
|
||
SysDeptExample example = new SysDeptExample();
|
||
SysDeptExample.Criteria criteria = example.createCriteria();
|
||
if (ObjectUtils.isEmpty(pid)){
|
||
criteria.andPidEqualTo(0L);
|
||
}else {
|
||
criteria.andPidEqualTo(pid);
|
||
}
|
||
example.setOrderByClause("dept_sort");
|
||
List<SysDept> sysDepts = sysDeptMapper.selectByExample(example);
|
||
return sysDepts;
|
||
}
|
||
|
||
@Transactional
|
||
public boolean add(DeptCreateRequest deptCreateRequest){
|
||
SysDept sysDept = BeanUtils.copyBean(new SysDept(), deptCreateRequest);
|
||
|
||
if (deptCreateRequest.isTop()){
|
||
sysDept.setPid(DEPT_ROOT_PID);
|
||
}
|
||
long now = System.currentTimeMillis();
|
||
sysDept.setCreateTime(now);
|
||
sysDept.setUpdateTime(now);
|
||
sysDept.setCreateBy(null);
|
||
sysDept.setUpdateBy(null);
|
||
sysDept.setSubCount(DEFAULT_SUBCOUNT);
|
||
try {
|
||
int insert = sysDeptMapper.insert(sysDept);
|
||
Long pid = null;
|
||
if ((pid = sysDept.getPid()) != DEPT_ROOT_PID ){
|
||
//这里需要更新上级节点SubCount
|
||
extDeptMapper.incrementalSubcount(pid);
|
||
}
|
||
if (insert == 1){
|
||
return true;
|
||
}
|
||
}catch (Exception e){
|
||
e.printStackTrace();
|
||
}
|
||
return false;
|
||
}
|
||
|
||
@Transactional
|
||
public int batchDelete(List<DeptDeleteRequest> requests){
|
||
/* Integer index = ids.stream().map(sysDeptMapper::deleteByPrimaryKey).reduce(Integer::sum).orElse(-1);
|
||
return index;*/
|
||
List<Long> ids = requests.stream().map(request -> {
|
||
Long pid = request.getPid();
|
||
if (pid != DEPT_ROOT_PID){
|
||
extDeptMapper.decreasingSubcount(pid);
|
||
}
|
||
return request.getDeptId();
|
||
}).collect(Collectors.toList());
|
||
return extDeptMapper.batchDelete(ids);
|
||
}
|
||
|
||
@Transactional
|
||
public int update(DeptCreateRequest deptCreateRequest){
|
||
SysDept sysDept = BeanUtils.copyBean(new SysDept(), deptCreateRequest);
|
||
if (deptCreateRequest.isTop()){
|
||
sysDept.setPid(DEPT_ROOT_PID);
|
||
}
|
||
sysDept.setUpdateTime(System.currentTimeMillis());
|
||
sysDept.setUpdateBy(null);
|
||
Long deptId = sysDept.getDeptId();
|
||
SysDept dept_old = sysDeptMapper.selectByPrimaryKey(deptId);
|
||
//如果PID发生了改变
|
||
//判断oldPid是否是跟节点PID ？ nothing : parent.subcount-1
|
||
//判断newPid是否是跟节点PID ？ nothing : parent.subcount+1
|
||
if (sysDept.getPid() != dept_old.getPid()){
|
||
Long oldPid = dept_old.getPid();
|
||
if (oldPid != DEPT_ROOT_PID){
|
||
extDeptMapper.decreasingSubcount(oldPid);
|
||
}
|
||
if (sysDept.getPid() != DEPT_ROOT_PID){
|
||
extDeptMapper.incrementalSubcount(sysDept.getPid());
|
||
}
|
||
}
|
||
return sysDeptMapper.updateByPrimaryKeySelective(sysDept);
|
||
}
|
||
|
||
public int updateStatus(DeptStatusRequest request){
|
||
Long deptId = request.getDeptId();
|
||
boolean status = request.isStatus();
|
||
SysDept sysDept = new SysDept();
|
||
sysDept.setDeptId(deptId);
|
||
return sysDeptMapper.updateByPrimaryKeySelective(sysDept);
|
||
}
|
||
|
||
|
||
|
||
public List<SysDept> nodesTreeByCondition(BaseGridRequest request){
|
||
//DeptService proxy = proxy();
|
||
List<SimpleTreeNode> allNodes = allNodes();
|
||
List<SimpleTreeNode> targetNodes = nodeByCondition(request);
|
||
if(CollectionUtils.isEmpty(targetNodes)){
|
||
return new ArrayList<>();
|
||
}
|
||
List<Long> ids = upTree(allNodes, targetNodes);
|
||
SysDeptExample example = new SysDeptExample();
|
||
if (CollectionUtils.isNotEmpty(ids)){
|
||
SysDeptExample.Criteria criteria = example.createCriteria();
|
||
criteria.andDeptIdIn(ids);
|
||
}
|
||
example.setOrderByClause("dept_sort");
|
||
List<SysDept> sysDepts = sysDeptMapper.selectByExample(example);
|
||
return sysDepts;
|
||
}
|
||
|
||
public List<DeptTreeNode> searchTree(Long deptId){
|
||
List<SysDept> roots = nodesByPid(0L);
|
||
if (deptId == DEPT_ROOT_PID) return roots.stream().map(this::format).collect(Collectors.toList());
|
||
SysDept sysDept = sysDeptMapper.selectByPrimaryKey(deptId);
|
||
if (roots.stream().anyMatch(node -> node.getDeptId() == deptId)) return roots.stream().map(this::format).collect(Collectors.toList());
|
||
SysDept current = sysDept;
|
||
DeptTreeNode currentNode = format(sysDept);
|
||
while (current.getPid() != DEPT_ROOT_PID){
|
||
SysDept parent = sysDeptMapper.selectByPrimaryKey(current.getPid()); //pid上有索引 所以效率不会太差
|
||
DeptTreeNode parentNode = format(parent);
|
||
parentNode.setChildren(currentNode.toList());
|
||
current = parent;
|
||
currentNode = parentNode;
|
||
}
|
||
|
||
DeptTreeNode targetRootNode = currentNode;
|
||
return roots.stream().map(node -> node.getDeptId() == targetRootNode.getId() ? targetRootNode : format(node)).collect(Collectors.toList());
|
||
}
|
||
|
||
private DeptTreeNode format(SysDept sysDept){
|
||
DeptTreeNode deptTreeNode = new DeptTreeNode();
|
||
deptTreeNode.setId(sysDept.getDeptId());
|
||
deptTreeNode.setLabel(sysDept.getName());
|
||
deptTreeNode.setHasChildren(sysDept.getSubCount() > 0);
|
||
return deptTreeNode;
|
||
}
|
||
|
||
private DeptService proxy(){
|
||
return CommonBeanFactory.getBean(DeptService.class);
|
||
}
|
||
|
||
|
||
private List<SimpleTreeNode> allNodes(){
|
||
List<SimpleTreeNode> simpleTreeNodes = extDeptMapper.allNodes();
|
||
return simpleTreeNodes;
|
||
}
|
||
|
||
private List<SimpleTreeNode> nodeByCondition(BaseGridRequest request){
|
||
GridExample gridExample = request.convertExample();
|
||
List<SimpleTreeNode> simpleTreeNodes = extDeptMapper.nodesByExample(gridExample);
|
||
return simpleTreeNodes;
|
||
}
|
||
|
||
/**
|
||
* 找出目标节点所在路径上的所有节点 向上找
|
||
* @param allNodes 所有节点
|
||
* @param targetNodes 目标节点
|
||
* @return
|
||
*/
|
||
private List<Long> upTree(List<SimpleTreeNode> allNodes, List<SimpleTreeNode> targetNodes){
|
||
final Map<Long, SimpleTreeNode> map = allNodes.stream().collect(Collectors.toMap(SimpleTreeNode::getId, node -> node));
|
||
List<Long> results = targetNodes.parallelStream().flatMap(targetNode -> {
|
||
//向上逐级找爹
|
||
List<Long> ids = new ArrayList<>();
|
||
SimpleTreeNode node = targetNode;
|
||
while (node != null) {
|
||
ids.add(node.getId());
|
||
Long pid = node.getPid();
|
||
node = map.get(pid);
|
||
}
|
||
return ids.stream();
|
||
}).distinct().collect(Collectors.toList());
|
||
return results;
|
||
}
|
||
|
||
|
||
|
||
}
|