房间选择设备

server-service/src/main/java/com/makeit/service/platform/space/impl/PlatRoomBedDeviceServiceImpl.java

@@ -97,19 +97,23 @@ public class PlatRoomBedDeviceServiceImpl extends ServiceImpl<PlatRoomBedDeviceM
         PlatSpaceDeviceQueryDTO dto = pageReqDTO.getData();
 
         LambdaQueryWrapper<PlatRoomBedDevice> queryWrapper1 = new LambdaQueryWrapper<>();
-//        queryWrapper1.select(PlatRoomBedDevice::getDeviceId);
-
         List<PlatRoomBedDevice> list = list(queryWrapper1);
         List<PlatRoomBedDevice> list1 = new ArrayList<>();
-        list.forEach(item->{
-            if(!item.getRoomId().equals(dto.getRoomId())
-                    || (item.getRoomId().equals(dto.getRoomId()) && StringUtil.isNotEmpty(item.getBedId()))){
-                list1.add(item);
-            }
-        });
+        if (CommonEnum.NO.getValue().equals(dto.getIsRoom())) {
+
+            list.forEach(item->{
+                if(!item.getRoomId().equals(dto.getRoomId())
+                        || (item.getRoomId().equals(dto.getRoomId()) && StringUtil.isNotEmpty(item.getBedId()))){
+                    list1.add(item);
+                }
+            });
+        }
 
 
         List<String> listEquipmentIds = list1.stream().map(item -> item.getDeviceId()).collect(Collectors.toList());
+        if (CommonEnum.YES.getValue().equals(dto.getIsRoom())) {
+            listEquipmentIds = list.stream().map(item -> item.getDeviceId()).collect(Collectors.toList());
+        }
 
         String orgId = "";
         if (StringUtil.isNotEmpty(dto.getRoomId())) {