diff --git a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsDeviceUserPOServiceImpl.java b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsDeviceUserPOServiceImpl.java index f13f3cb..7e5e877 100644 --- a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsDeviceUserPOServiceImpl.java +++ b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsDeviceUserPOServiceImpl.java @@ -162,7 +162,13 @@ public class CsDeviceUserPOServiceImpl extends ServiceImpl deviceTree = iCsLedgerService.getDeviceTree(); - List collect1 = deviceTree.stream().filter(temp -> temp.getId().equals(id)).map(CsLedgerVO::getChildren).flatMap(Collection::stream).map(CsLedgerVO::getChildren).flatMap(Collection::stream).map(CsLedgerVO::getId).collect(Collectors.toList()); + //由于多加了一程便携式设备 + List collect1 = deviceTree.stream() + .map(CsLedgerVO::getChildren).flatMap(Collection::stream) + .filter(temp -> temp.getId().equals(id)) + .map(CsLedgerVO::getChildren).flatMap(Collection::stream) + .map(CsLedgerVO::getChildren).flatMap(Collection::stream) + .map(CsLedgerVO::getId).collect(Collectors.toList()); //求交集 device.retainAll(collect1); if(CollectionUtils.isEmpty(device)){ diff --git a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsLedgerServiceImpl.java b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsLedgerServiceImpl.java index a78e839..13bc0b0 100644 --- a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsLedgerServiceImpl.java +++ b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsLedgerServiceImpl.java @@ -194,10 +194,35 @@ public class CsLedgerServiceImpl extends ServiceImpl i } ). sorted(Comparator.comparing(CsLedgerVO::getSort)).collect(Collectors.toList()); - + checkDevSetData(deviceList); projectList.forEach(pro -> pro.setChildren(getChildren(pro, deviceList))); engineeringList.forEach(eng -> eng.setChildren(getChildren(eng, projectList))); - return engineeringList; + //整理整个树并新增便携式设备树 + CsLedgerVO portable = new CsLedgerVO(); + portable.setLevel(0); + portable.setName(DataParam.portableDev); + portable.setPid("0"); + portable.setId(IdUtil.simpleUUID()); + List portables = wlRecordMapper.getAll(); + checkDevSetData(portables); + for(CsLedgerVO c : portables){ + c.setPid(portable.getId()); + CsEquipmentDeliveryPO csEquipmentDeliveryPO = csEquipmentDeliveryMapper.selectById(c.getId()); + c.setComFlag(Integer.parseInt(csEquipmentDeliveryPO.getStatus())); + } + portable.setChildren(portables); + + CsLedgerVO government = new CsLedgerVO(); + government.setLevel(0); + government.setName(DataParam.governmentDev); + government.setPid("0"); + government.setId(IdUtil.simpleUUID()); + government.setChildren(engineeringList); + List tree = new ArrayList<>(); + tree.add(portable); + tree.add(government); + return tree; + //return engineeringList; } @Override