List<String> landsForVillages = feignBasicController.getLandsForVillages(id); List<DevInfo> devByBaseId = feignHdController.getDevByBaseId(userInfo.getId(), classify); List<String> machineIds = devByBaseId.stream().map(DevInfo::getId).collect(Collectors.toList()); Set<String> machineIdList = new HashSet<>(); // json转化为list for (String machineId : machineIds) { List<String> landIdsList = baseMapper.getLandIdsById(machineId); for (String landIdStr : landIdsList) { if(StringUtils.isNotBlank(landIdStr)){ JSONArray arrayStr = JSON.parseArray(landIdStr); for (Object o : arrayStr) { if (landsForVillages.contains(o)) { // 村下面的工作设备 machineIdList.add(machineId); } } } } } 优化一下代码
时间: 2024-01-17 21:06:09 浏览: 262
检测USB驱动是否已经 安装(C语言).txt
可以尝试使用 Java 8 的 Stream API 重构代码,如下所示:
```
Set<String> machineIdList = devByBaseId.stream()
.map(DevInfo::getId)
.flatMap(machineId -> baseMapper.getLandIdsById(machineId).stream())
.filter(StringUtils::isNotBlank)
.flatMap(landIdStr -> JSON.parseArray(landIdStr).stream())
.distinct()
.filter(landsForVillages::contains)
.map(machineIdList::add)
.collect(Collectors.toSet());
```
这段代码首先使用 `flatMap` 将 `devByBaseId` 中的每个 `DevInfo` 转换为其对应的 `machineId`,然后对于每个 `machineId`,调用 `baseMapper.getLandIdsById` 获取其对应的 `landIdsList`,并将其转换为一个 `Stream<String>`。接着对于每个 `landIdStr`,使用 `JSON.parseArray` 将其转换为一个 `JSONArray`,然后将其中的每个元素转换为 `String`,并去重。最后,使用 `filter` 过滤出 `landsForVillages` 中包含的元素,并将其对应的 `machineId` 添加到 `machineIdList` 中,最终使用 `collect` 将其转换为一个 `Set<String>`。
阅读全文