| | |
|---|
| | | @Override |
|---|
| | | public List<FactoryDto> selectStandardTreeList() { |
|---|
| | | List<FactoryDto> factoryDtos = standardTreeMapper.selectStandardTreeList(); |
|---|
| | | // 收集所有 sampleTypeDto 的 value |
|---|
| | | Set<String> sampleTypeValues = new HashSet<>(); |
|---|
| | | for (FactoryDto factoryDto : factoryDtos) { |
|---|
| | | for (LaboratoryDto laboratoryDto : factoryDto.getChildren()) { |
|---|
| | | laboratoryDto.getChildren().sort((o1, o2) -> (o1.getSort() == null ? 0 : o1.getSort()) |
|---|
| | | - (o2.getSort() == null ? 0 : o2.getSort())); |
|---|
| | | for (SampleTypeDto sampleTypeDto : laboratoryDto.getChildren()) { |
|---|
| | | sampleTypeValues.add(sampleTypeDto.getValue()); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | // 批量查询所有 sampleTypeDto 的数据 |
|---|
| | | List<SampleDto> standardList = standardTreeMapper.getStandardTree3Batch(sampleTypeValues); |
|---|
| | | Map<String, List<SampleDto>> standardTreeMap = standardList.stream().collect(Collectors.groupingBy(SampleDto::getValue)); |
|---|
| | | // 将查询结果分配到对应的 sampleTypeDto |
|---|
| | | for (FactoryDto factoryDto : factoryDtos) { |
|---|
| | | for (LaboratoryDto laboratoryDto : factoryDto.getChildren()) { |
|---|
| | | for (SampleTypeDto sampleTypeDto : laboratoryDto.getChildren()) { |
|---|
| | | List<SampleDto> standardTreeList = standardTreeMap.get(sampleTypeDto.getValue()); |
|---|
| | | if (standardTreeList != null) { |
|---|
| | | sampleTypeDto.getChildren().addAll(standardTreeList); |
|---|
| | | } |
|---|
| | | // if (sampleTypeDto.getChildren().size() == 0) { |
|---|
| | | sampleTypeDto.getChildren().addAll(standardTreeMapper.getStandardTree3(sampleTypeDto.getValue())); |
|---|
| | | // } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
