表头
工具类
public class ThreeUtils {
/**
* 树结构转换处理-每次递归查询全部下级以及下级的子集
*
* @param menuList 需要处理的数据集
* @param threeResult 返回对象
* @param parentId 父级ID
* @param dataTreating 逻辑处理
* @param <T>
*/
public static <T extends Three, E extends Three> void treeListAncestral(List<T> menuList, E threeResult, Long parentId, ThreeFunctional<T, E> dataTreating) {
if (ObjectUtil.isEmpty(menuList)) {
return;
}
Map<Long, List<T>> collect = menuList.stream().filter(menu -> menu != null && menu.getParentId() != null).collect(Collectors.groupingBy(Three::getParentId));
//根据父级ID查询下级
List<T> threes = collect.get(parentId);
if (ObjectUtil.isEmpty(threes)) {
return;
}
ArrayList<T> groupT = threes.stream().collect(Collectors.collectingAndThen(Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(T::getId))), ArrayList::new));
for (T three : groupT) {
/*根据本次循环ID查询所有下级以及子集*/
List<T> ancestralThreeList = menuList.stream().filter(m -> StringUtil.leftPad(String.valueOf(m.getId()), String.valueOf(m.getId()).length()).equals(String.valueOf(three.getId()))).collect(Collectors.toList());
if (ObjectUtil.isEmpty(ancestralThreeList)) {
continue;
}
//树处理逻辑
E e = dataTreating.dataTreating(ancestralThreeList, threeResult);
if (null == e) {
return;
}
/*创建下级返回对象类，并且存入当前上级的下级集合中*/
List<E> itemList = threeResult.getItemList();
if (null == itemList) {
itemList = new ArrayList<>();
}
itemList.add(e);
threeResult.setItemList(itemList);
//执行递归处理
treeListAncestral(menuList, e, three.getId(), dataTreating);
}
}
/**
* 树结构转换处理-每次递归查询下级
*
* @param menuList 树处理集合
* @param threeResult 返回对象
* @param parentId 父级ID
* @param dataTreating 逻辑处理
* @param <T>
*/
public static <T extends Three, E extends Three> void treeList(List<T> menuList, E threeResult, Long parentId, ThreeFunctional<T, E> dataTreating) {
if (ObjectUtil.isEmpty(menuList)) {
return;
}
Map<Long, List<T>> collect = menuList.stream().filter(menu -> menu != null && menu.getParentId() != null).collect(Collectors.groupingBy(Three::getParentId));
//根据父级ID查询下级
List<T> threes = collect.get(parentId);
if (ObjectUtil.isEmpty(threes)) {
return;
}
ArrayList<T> groupT = threes.stream().collect(Collectors.collectingAndThen(Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(T::getId))), ArrayList::new));
for (T three : groupT) {
/*根据本次循环ID查询所有下级*/
List<T> ancestralThreeList = menuList.stream().filter(menu -> menu != null && menu.getParentId() != null && menu.getParentId().equals(three.getId())).collect(Collectors.toList());
if (ObjectUtil.isEmpty(ancestralThreeList)) {
continue;
}
//树处理逻辑
E e = dataTreating.dataTreating(ancestralThreeList, threeResult);
if (null == e) {
return;
}
/*创建下级返回对象类，并且存入当前上级的下级集合中*/
List<E> itemList = threeResult.getItemList();
if (null == itemList) {
itemList = new ArrayList<>();
}
itemList.add(e);
threeResult.setItemList(itemList);
//执行递归处理
treeList(menuList, e, three.getId(), dataTreating);
}
}
}
树处理函数接口
@FunctionalInterface
public interface ThreeFunctional<T, E> {
/**
* 树处理逻辑
*
* @param threeList 处理后的树结构
* @param data 处理后的树类结果对象
* 这里返回创建一个新的空对象即可
*/
E dataTreating(List<T> threeList, E data);
}
树实体类
@Data
public class Three<E> implements Serializable {
private static final long serialVersionUID = 2883288479291688378L;
private Long id;
private String name;
private Long parentId;
//数据结果集
private List<E> ItemList;
}
请求响应实体类
@Data
@ApiModel("返回VO")
public class CoveredCountyEnteringVO extends Three<CoveredCountyEnteringVO> implements Serializable {
private static final long serialVersionUID = -1571960669989120842L;
private Integer monitorBabyNumber;
private Integer standardBabyNumber;
private BigDecimal percentOfPass;
}
@Data
@ApiModel("响应DTO")
public class CoveredCountyEnteringDTO extends Three<CoveredCountyEnteringDTO> implements Serializable {
private static final long serialVersionUID = -1571960669989120842L;
private String medicalQuestion1;
private String medicalQuestion3;
private String medicalQuestion7;
}
调用
//需要处理的数据集合-需要继承 Three类
List<CoveredCountyEnteringDTO> questionnaireList = cniQuestionnaireMapper.selectByAreaCode(reportFormsDTO);
//创建返回对象-继承 Three类
CoveredCountyEnteringVO countyEnteringVO = new CoveredCountyEnteringVO();
ThreeUtils.treeListAncestral(questionnaireList, countyEnteringVO, reportFormsDTO.getAreaCode(), (threeList, data) -> {
if (ObjectUtil.isEmpty(threeList)) {
return null;
}
Long monitorBabyNumber = threeList.stream().count();
Stream<CoveredCountyEnteringDTO> standardBabyNumber = threeList.stream().filter(coveredCountyEnteringDTO -> null != coveredCountyEnteringDTO
&& StringUtils.isNotBlank(coveredCountyEnteringDTO.getMedicalQuestion1())
&& StringUtils.isNotBlank(coveredCountyEnteringDTO.getMedicalQuestion3())
&& StringUtils.isNotBlank(coveredCountyEnteringDTO.getMedicalQuestion7()));
CoveredCountyEnteringDTO coveredCountyEnteringDTO = threeList.get(0);
/*将处理后的数据放入 data集合中*/
CoveredCountyEnteringVO resultCoveredCountVO = new CoveredCountyEnteringVO();
resultCoveredCountVO.setId(coveredCountyEnteringDTO.getId());
resultCoveredCountVO.setName(coveredCountyEnteringDTO.getName());
resultCoveredCountVO.setParentId(coveredCountyEnteringDTO.getParentId());
resultCoveredCountVO.setMonitorBabyNumber(null != monitorBabyNumber ? Integer.valueOf(monitorBabyNumber + "") : 0);
resultCoveredCountVO.setStandardBabyNumber(ObjectUtil.isNotEmpty(standardBabyNumber) ? Integer.valueOf(standardBabyNumber.count() + "") : 0);
/*计算合格率*/
resultCoveredCountVO.setPercentOfPass(BigDecimalUtils.rate(resultCoveredCountVO.getStandardBabyNumber() + "", resultCoveredCountVO.getMonitorBabyNumber() + ""));
return resultCoveredCountVO;
});