将下面代码进行优化,使用Stream流处理 List<PtmItemVo> ptmItemVoList = baseMapper.selectVoList(wrapper); List<PtmItemVo> list = new ArrayList<>(); for (PtmItemVo item : ptmItemVoList) { if ("dwgc".equals(item.getType())) { List<PtmItemVo> zdwgcList = findListByParentId(item.getId()); for (PtmItemVo zdwgcEntity : zdwgcList) { if ("zdwgc".equals(zdwgcEntity.getType())) { list.add(zdwgcEntity); } else if (!list.contains(item)) { list.add(item); } } } }
时间: 2024-02-10 11:13:06 浏览: 72
可以使用Stream流处理来优化这段代码,如下所示:
```
List<PtmItemVo> list = baseMapper.selectVoList(wrapper)
.stream()
.filter(item -> "dwgc".equals(item.getType()))
.flatMap(item -> {
List<PtmItemVo> zdwgcList = findListByParentId(item.getId());
return zdwgcList.stream()
.filter(zdwgcEntity -> "zdwgc".equals(zdwgcEntity.getType()))
.findFirst()
.map(Stream::of)
.orElse(Stream.of(item));
})
.collect(Collectors.toList());
```
首先,将List转换为Stream流,然后使用filter方法过滤出type属性为"dwgc"的元素。接着,使用flatMap方法将每个元素转换为一个Stream流,其中找到type属性为"zdwgc"的元素并返回,否则返回原元素。最后,使用collect方法将Stream流转换为List集合。
相关问题
优化下面的代码 public Page<FinanceCodeDTO> selectFinanceCodePage(FinanceCodeListVO financeCodeListVO) { //根据类型关联查询数据信息 Page<FinanceCodeDTO> page = new Page<>(financeCodeListVO.getPageNo(),financeCodeListVO.getPageSize()); Page<FinanceCodeDTO> dtoPage = financeCodeMapper.financeCodeList(page, financeCodeListVO); List<FinanceCodeDTO> dtos = dtoPage.getRecords(); List<String> contractIds = dtos.stream().map(FinanceCodePO::getContractIds).collect(Collectors.toList()); List<Integer> idList = new ArrayList<>(); contractIds.forEach(ids -> { if (ToolUtil.isNotEmpty(ids) && !ToolUtil.equals("[]",ids)){ List<Integer> list = JSONObject.parseArray(ids, Integer.class); idList.addAll(list); } }); if (ToolUtil.isNotEmpty(idList)){ CustomerContractPO customerContractPO = new CustomerContractPO(); List<CustomerContractPO> customerContractPOS = customerContractPO.selectList(new QueryWrapper<>(customerContractPO).lambda().in(CustomerContractPO::getCustomerContractId, idList)); Map<Integer, String> map = customerContractPOS.stream().collect(Collectors.toMap(CustomerContractPO::getCustomerContractId, CustomerContractPO::getContractName)); dtos.forEach(dto ->{ if (ToolUtil.isNotEmpty(dto.getContractIds())&& !ToolUtil.equals("[]",dto.getContractIds())){ List<Integer> list = JSONObject.parseArray(dto.getContractIds(), Integer.class); List<String> names = new ArrayList<>(); list.forEach(id ->{ names.add(map.get(id)); }); dto.setContractName(String.join(",",names)); } }); dtoPage.setRecords(dtos); } return dtoPage ; }
&& ids.contains(",")) { String[] idsArr = ids.split(","); Arrays.stream(idsArr).forEach(id -> { idList.add(Integer.parseInt(id)); }); } }); List<ContractDTO> contractDTOList = contractMapper.selectContractByIdList(idList); dtos.forEach(dto -> { dto.setContractList(contractDTOList); }); return dtoPage; }答:根据给定的代码,优化的思路是把contractIds通过Arrays.stream()和Integer.parseInt()转换为idList,并通过contractMapper.selectContractByIdList()获取contractDTOList,最后把contractDTOList赋给dtos中每个dto对象。
帮我解释一下下面的代码 public PageResponse<QuestionVo> getQuestionVoByIds(List<Integer> ids) { List<Question> questions = questionMapper.selectBatchIds(ids); List<Answer> answers = answerMapper.selectList(new QueryWrapper<Answer>().in("question_id", ids)); List<QuestionVo> questionVos = questions.stream() .map(question -> { Answer currentQuestionAnswer = answers.stream() .filter(answer -> answer.getQuestionId().equals(question.getId())) .findFirst() .orElse(null); return buildQuestionVoByQuestionAndAnswer(question, currentQuestionAnswer); }).collect(Collectors.toList()); return PageResponse.<QuestionVo>builder() .data(questionVos) .total(questionVos.size()) .build();
这段代码是一个方法,它接受一个整数列表作为参数,返回一个包含QuestionVo对象的PageResponse对象。
首先,代码从数据库中获取与给定id列表匹配的Question对象列表。接着,它使用给定的ids列表查询数据库中与这些ids匹配的Answer对象列表。
然后,代码使用流操作对问题列表进行映射。对于每个问题,它通过筛选与该问题id相匹配的答案对象,并将其构建为QuestionVo对象。最后,使用Collectors.toList()将所有的QuestionVo对象收集到一个列表中。
最后,通过PageResponse.builder()创建一个PageResponse对象,并设置data属性为问题Vo对象列表,total属性为问题Vo对象列表的大小,然后返回该PageResponse对象。
阅读全文