feat:[REQ-3282] 编写协同关系查询

This commit is contained in:
liuyang 2024-12-23 13:45:56 +08:00
parent 8051c9eeee
commit 3da7df020d
6 changed files with 222 additions and 59 deletions

View File

@ -15,58 +15,54 @@ import java.util.Set;
@Data
@Builder
public class ListOrgCooperateShipReq {
public static final int FILTER_PERSON_TYPE_CURRENT_NODE = 1;
public static final int FILTER_PERSON_TYPE_CURRENT_AND_BELOW_NODE = 2;
/**
* 协同关系集合
*/
@CriteriaField(field = "id", operator = Operator.IN)
@Builder.Default
private Set<Long> ids = Collections.emptySet();
/**
* 工作台id
*/
@CriteriaField(field = "workspaceId", operator = Operator.IN)
@Builder.Default
private Set<Long> workspaceIds = Collections.emptySet();
/**
* 工作台类型
*/
@CriteriaField(field = "workspaceType", operator = Operator.IN)
@Builder.Default
private Set<Integer> workspaceTypes = Collections.emptySet();
/**
* 单位id集合(如果是班组则插入队伍id)
*/
@CriteriaField(field = "organizationalUnitId", operator = Operator.IN)
@Builder.Default
private Set<Long> ouIds = Collections.emptySet();
/**
* 指定的协同关系类型
*/
@CriteriaField(field = "cooperateType", operator = Operator.IN)
@Builder.Default
private Set<Integer> includeCooperateTypes = Collections.emptySet();
/**
* 待排除的协同关系类型
*/
@CriteriaField(field = "cooperateType", operator = Operator.NOT_IN)
@Builder.Default
private Set<Integer> excludeCooperateTypes = Collections.emptySet();
/**
* 节点id
*/
@CriteriaField(field = "organizationalNodeId", operator = Operator.IN)
@Builder.Default
private Set<Long> organizationNodeIds = Collections.emptySet();
/**
* 合作关系 1.合作 2.直属
*/
@CriteriaField(field = "partnerShip", operator = Operator.IN)
@Builder.Default
private Set<Integer> partnerShips = Collections.emptySet();
@ -74,62 +70,69 @@ public class ListOrgCooperateShipReq {
/**
* 在场状态集合
*/
@CriteriaField(field = "status", operator = Operator.IN)
@Builder.Default
private Set<Integer> statuses = Collections.emptySet();
@CriteriaField(field = "path", operator = Operator.SW)
/**
* 查询策略使用取模方式自由组合进行数据查询
* 参数说明1当前节点 2父节点 4祖父节点 8子节点 16孙节点
* 案例查询当前节点及子孙节点则传3, 当前节点及祖父节点则传5当前节点
*/
@Builder.Default
private Set<String> pathRight = Collections.emptySet();
private Integer selectStrategy = 1;
/**
* 需要过滤的personId
* personId
*/
@CriteriaField(ignore = true)
private Long filterPersonId ;
private Long filterByPersonId ;
/**
* 当personId有值时为必填项
* 1只过滤当前节点数据 2过滤当前节点及以下数据
*/
@CriteriaField(ignore = true)
private Integer filterPersonType;
/**
* 是否过滤工人
*/
@CriteriaField(ignore = true)
@Builder.Default
private Boolean isFilterWorker = false;
/**
* 人员过滤是否包含已退场
*/
@CriteriaField(ignore = true)
@Builder.Default
private Boolean includeExitPerson = false;
/**
* 是否查询祖先节点
* 是否包含祖先节点
*/
@CriteriaField(ignore = true)
@Builder.Default
private Boolean isSelectAncestors = false;
private Boolean includeAncestors = false;
/**
* 是否查询子孙节点
* 是否包含子孙节点
*/
@CriteriaField(ignore = true)
@Builder.Default
private Boolean isSelectDescendants = false;
private Boolean includeDescendants = false;
/**
* 是否包含父节点
*/
@Builder.Default
private Boolean includeParent = false;
/**
* 是否包含子节点
*/
@Builder.Default
private Boolean includeChildren = false;
/**
* 条件当需要查询上下级数据时启用
* 是否只查一级
*/
@CriteriaField(ignore = true)
@Builder.Default
private Boolean isSelectLevelOne = false;

View File

@ -1,21 +1,22 @@
package cn.axzo.orgmanax.infra.dao.cooperateship.repository;
import cn.axzo.foundation.dao.support.wrapper.CriteriaField;
import cn.axzo.foundation.dao.support.wrapper.Operator;
import cn.axzo.orgmanax.api.cooperateship.req.ListOrgCooperateShipReq;
import cn.axzo.orgmanax.infra.dao.cooperateship.entity.SaasCooperateShip;
import cn.hutool.core.bean.BeanUtil;
import com.google.common.base.Preconditions;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.NoArgsConstructor;
import lombok.*;
import lombok.experimental.SuperBuilder;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.Set;
public interface CooperateShipQueryRepository {
List<SaasCooperateShip> list(ListOrgCooperateShipReq req);
List<SaasCooperateShip> list(ListReq req);
default SaasCooperateShip one(OneReq req) {
return oneOpt(req).orElse(null);
@ -23,7 +24,7 @@ public interface CooperateShipQueryRepository {
default Optional<SaasCooperateShip> oneOpt(OneReq req) {
req.check();
ListOrgCooperateShipReq listReq = BeanUtil.toBean(req, ListOrgCooperateShipReq.class);
ListReq listReq = BeanUtil.toBean(req, ListReq.class);
return list(listReq).stream().findFirst();
}
@ -50,4 +51,87 @@ public interface CooperateShipQueryRepository {
private String todo;
}
@NoArgsConstructor
@AllArgsConstructor
@Data
@Builder
class ListReq {
/**
* 协同关系集合
*/
@CriteriaField(field = "id", operator = Operator.IN)
@Builder.Default
private Set<Long> ids = Collections.emptySet();
/**
* 工作台id
*/
@CriteriaField(field = "workspaceId", operator = Operator.IN)
@Builder.Default
private Set<Long> workspaceIds = Collections.emptySet();
/**
* 工作台类型
*/
@CriteriaField(field = "workspaceType", operator = Operator.IN)
@Builder.Default
private Set<Integer> workspaceTypes = Collections.emptySet();
/**
* 单位id集合(如果是班组则插入队伍id)
*/
@CriteriaField(field = "organizationalUnitId", operator = Operator.IN)
@Builder.Default
private Set<Long> ouIds = Collections.emptySet();
/**
* 指定的协同关系类型
*/
@CriteriaField(field = "cooperateType", operator = Operator.IN)
@Builder.Default
private Set<Integer> includeCooperateTypes = Collections.emptySet();
/**
* 待排除的协同关系类型
*/
@CriteriaField(field = "cooperateType", operator = Operator.NOT_IN)
@Builder.Default
private Set<Integer> excludeCooperateTypes = Collections.emptySet();
/**
* 节点id
*/
@CriteriaField(field = "organizationalNodeId", operator = Operator.IN)
@Builder.Default
private Set<Long> organizationNodeIds = Collections.emptySet();
/**
* 合作关系 1.合作 2.直属
*/
@CriteriaField(field = "partnerShip", operator = Operator.IN)
@Builder.Default
private Set<Integer> partnerShips = Collections.emptySet();
/**
* 在场状态集合
*/
@CriteriaField(field = "status", operator = Operator.IN)
@Builder.Default
private Set<Integer> statuses = Collections.emptySet();
/**
* path右值
*/
@CriteriaField(field = "path", operator = Operator.IN)
@Builder.Default
private Set<String> pathsRight = Collections.emptySet();
/**
* path右值
*/
@CriteriaField(field = "parentId", operator = Operator.IN)
@Builder.Default
private Set<Long> parentIds = Collections.emptySet();
}
}

View File

@ -8,6 +8,8 @@ import cn.axzo.orgmanax.infra.dao.cooperateship.entity.SaasCooperateShip;
import cn.axzo.orgmanax.infra.dao.cooperateship.repository.CooperateShipQueryRepository;
import cn.axzo.orgmanax.infra.dao.nodeuser.repository.NodeUserQueryRepository;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
@ -27,26 +29,36 @@ public class CooperateShipQueryRepositoryImpl implements CooperateShipQueryRepos
private final NodeUserQueryRepository nodeUserQueryRepository;
@Override
public List<SaasCooperateShip> list(ListOrgCooperateShipReq req) {
public List<SaasCooperateShip> list(CooperateShipQueryRepository.ListReq req) {
QueryWrapper<SaasCooperateShip> wrapper = QueryWrapperHelper.fromBean(req, SaasCooperateShip.class);
if (wrapper.isEmptyOfWhere()) {
//
log.warn("CooperateShipResp empty of where query. return empty list. req = {}", req);
return Collections.emptyList();
}
// 查询子集
if (CollUtil.isNotEmpty(req.getPathsRight())) {
// 遍历pathsRight转换成wrapper
wrapper.and(w -> {
for (String pathRight : req.getPathsRight()) {
w.or().likeRight("path", pathRight);
}
});
}
return cooperateShipDao.list(wrapper)
.stream().map(e -> BeanUtil.toBean(e, SaasCooperateShip.class)).collect(Collectors.toList());
}
private List<SaasCooperateShip> filterByPersonIds(List<SaasCooperateShip> results, ListOrgCooperateShipReq req) {
if (req.getFilterPersonId() == null) {
if (req.getFilterByPersonId() == null) {
// filter by personIds
return results;
}
Set<Long> organizationalNodeIds = results.stream().map(SaasCooperateShip::getOrganizationalNodeId).collect(Collectors.toSet());
Set<Long> personInNodeIds = nodeUserQueryRepository.list(ListNodeUserReq.builder()
.personId(req.getFilterPersonId())
.personId(req.getFilterByPersonId())
.organizationalNodeIds(organizationalNodeIds)
.build())
.stream().map(NodeUserQueryRepository.NodeUserResp::getTopNodeId).collect(Collectors.toSet());

View File

@ -20,4 +20,19 @@ public interface CooperateShipFoundationService {
* @return
*/
Set<Long> extractAncestorIds(List<SaasCooperateShip> list);
/**
* 提取path数据
* @param list 当前节点
* @param suffix 后缀
* @return
*/
Set<String> extractPaths(List<SaasCooperateShip> list, String suffix);
/**
* 提取父级id
* @param currentNodeList
* @return
*/
Set<Long> extractParentIds(List<SaasCooperateShip> currentNodeList);
}

View File

@ -4,6 +4,7 @@ import cn.axzo.orgmanax.api.nodeuser.req.ListNodeUserReq;
import cn.axzo.orgmanax.infra.dao.cooperateship.entity.SaasCooperateShip;
import cn.axzo.orgmanax.infra.dao.nodeuser.repository.NodeUserQueryRepository;
import cn.axzo.orgmanax.server.cooperateship.foundation.CooperateShipFoundationService;
import cn.hutool.core.util.StrUtil;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
@ -41,4 +42,14 @@ public class CooperateShipFoundationServiceImpl implements CooperateShipFoundati
.collect(Collectors.toSet());
}
@Override
public Set<String> extractPaths(List<SaasCooperateShip> list, String suffix) {
return list.stream().map(t -> StrUtil.isNotBlank(suffix) ? StrUtil.concat(true, t.getPath(), suffix) : t.getPath()).collect(Collectors.toSet());
}
@Override
public Set<Long> extractParentIds(List<SaasCooperateShip> list) {
return list.stream().filter(t -> t.getParentId() != 0).map(SaasCooperateShip::getParentId).collect(Collectors.toSet());
}
}

View File

@ -126,44 +126,82 @@ public class CooperateShipServiceImpl implements CooperateShipService {
@Override
public List<OrgCooperateShipDTO> list(ListOrgCooperateShipReq req) {
// 是否查询当前节点的人员
boolean isFilterCurrentNodeByPerson = ObjectUtil.isNotNull(req.getFilterPersonId()) && Objects.equals(req.getFilterPersonType(), 1);
// 是否查询当前节点及以下节点的人员
boolean isFilterCurrentAndBelowByPerson = !isFilterCurrentNodeByPerson && ObjectUtil.isNotNull(req.getFilterPersonId()) && Objects.equals(req.getFilterPersonType(), 2);
// 全局变量定义是否查询当前节点的人员
boolean isFilterCurrentNodeByPerson = ObjectUtil.isNotNull(req.getFilterByPersonId()) &&
Objects.equals(req.getFilterPersonType(), ListOrgCooperateShipReq.FILTER_PERSON_TYPE_CURRENT_NODE);
// 全局变量定义是否查询当前节点及以下节点的人员
boolean isFilterCurrentAndBelowByPerson = !isFilterCurrentNodeByPerson && ObjectUtil.isNotNull(req.getFilterByPersonId())
&& Objects.equals(req.getFilterPersonType(), ListOrgCooperateShipReq.FILTER_PERSON_TYPE_CURRENT_AND_BELOW_NODE);
// 1. 根据入参查询数据
List<SaasCooperateShip> list = cooperateShipQueryRepository.list(req);
if (CollUtil.isEmpty(list)) {
// 根据入参查询数据
List<SaasCooperateShip> currentNodeList = cooperateShipQueryRepository.list(
CooperateShipQueryRepository.ListReq.builder()
.ids(req.getIds())
.workspaceIds(req.getWorkspaceIds())
.workspaceTypes(req.getWorkspaceTypes())
.ouIds(req.getOuIds())
.includeCooperateTypes(req.getIncludeCooperateTypes())
.excludeCooperateTypes(req.getExcludeCooperateTypes())
.organizationNodeIds(req.getOrganizationNodeIds())
.partnerShips(req.getPartnerShips())
.statuses(req.getStatuses())
.build());
if (CollUtil.isEmpty(currentNodeList)) {
return Collections.emptyList();
}
// 根据person过滤当前节点
if (isFilterCurrentNodeByPerson) {
cooperateShipFoundationService.filterByPerson(req.getFilterPersonId(), list);
cooperateShipFoundationService.filterByPerson(req.getFilterByPersonId(), currentNodeList);
if (CollUtil.isEmpty(currentNodeList)) {
return Collections.emptyList();
}
}
// 是否查询祖先
if (BooleanUtil.isTrue(req.getIsSelectAncestors())) {
Set<Long> ancestorIds = cooperateShipFoundationService.extractAncestorIds(list);
List<SaasCooperateShip> ancestorShipNodes = cooperateShipQueryRepository.list(ListOrgCooperateShipReq.builder().ids(ancestorIds).build());
list.addAll(ancestorShipNodes);
// 构建结果节点集合
List<SaasCooperateShip> resultNodeList = new ArrayList<>(currentNodeList);
// 是否查询祖先节点
if (BooleanUtil.isTrue(req.getIncludeParent())) {
Set<Long> parentIds = cooperateShipFoundationService.extractParentIds(currentNodeList);
List<SaasCooperateShip> ancestorShipNodes = cooperateShipQueryRepository.list(CooperateShipQueryRepository.ListReq.builder().ids(parentIds).build());
resultNodeList.addAll(ancestorShipNodes);
}
// 是否查询子孙
if (BooleanUtil.isTrue(req.getIsSelectDescendants())) {
List<SaasCooperateShip> descendantShipNodes = cooperateShipQueryRepository.list(ListOrgCooperateShipReq.builder().build());
// list.addAll(descendantShipNodes);
// 是否查询父级节点
if (BooleanUtil.isTrue(req.getIncludeAncestors())) {
Set<Long> ancestorIds = cooperateShipFoundationService.extractAncestorIds(currentNodeList);
List<SaasCooperateShip> ancestorShipNodes = cooperateShipQueryRepository.list(CooperateShipQueryRepository.ListReq.builder().ids(ancestorIds).build());
resultNodeList.addAll(ancestorShipNodes);
}
// 是否查询子节点
if (BooleanUtil.isTrue(req.getIncludeChildren())) {
Set<String> paths = cooperateShipFoundationService.extractPaths(currentNodeList, ",");
List<SaasCooperateShip> descendantShipNodes = cooperateShipQueryRepository.list(
CooperateShipQueryRepository.ListReq.builder()
.pathsRight(paths).build());
resultNodeList.addAll(descendantShipNodes);
// 是否查询当前节点及以下节点的人员
if (isFilterCurrentAndBelowByPerson) {
cooperateShipFoundationService.filterByPerson(req.getFilterByPersonId(), resultNodeList);
}
}
// 是否查询子孙节点
if (BooleanUtil.isTrue(req.getIncludeDescendants())) {
Set<Long> parentIds = cooperateShipFoundationService.extractParentIds(currentNodeList);
List<SaasCooperateShip> descendantShipNodes = cooperateShipQueryRepository.list(
CooperateShipQueryRepository.ListReq.builder()
.parentIds(parentIds).build());
resultNodeList.addAll(descendantShipNodes);
// 是否查询当前节点及以下节点的人员
if (isFilterCurrentAndBelowByPerson) {
cooperateShipFoundationService.filterByPerson(req.getFilterByPersonId(), resultNodeList);
}
}
// 根据person过滤当前节点及以下节点
if (isFilterCurrentAndBelowByPerson) {
cooperateShipFoundationService.filterByPerson(req.getFilterPersonId(), list);
}
return BeanUtil.copyToList(list, OrgCooperateShipDTO.class);
return BeanUtil.copyToList(resultNodeList, OrgCooperateShipDTO.class);
}
}