From 96c5a333ec502b7d545648e8996b8b9c3e39ec2c Mon Sep 17 00:00:00 2001
From: skyouc
Date: 星期三, 26 三月 2025 15:12:58 +0800
Subject: [PATCH] 删除下载安全策略

---
 rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java |   25 ++++++++++++++++++++++---
 1 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java b/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java
index fc56f98..e9edfaa 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java
@@ -2,17 +2,22 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.vincent.rsf.framework.exception.CoolException;
+import com.vincent.rsf.server.manager.entity.Loc;
 import com.vincent.rsf.server.manager.entity.WarehouseAreas;
 import com.vincent.rsf.server.manager.mapper.WarehouseMapper;
 import com.vincent.rsf.server.manager.entity.Warehouse;
 import com.vincent.rsf.server.manager.service.WarehouseAreasService;
 import com.vincent.rsf.server.manager.service.WarehouseService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+import java.util.stream.Collectors;
 
 @Service("warehouseService")
 public class WarehouseServiceImpl extends ServiceImpl<WarehouseMapper, Warehouse> implements WarehouseService {
@@ -21,19 +26,33 @@
     private WarehouseAreasService warehouseAreasService;
 
     @Override
-    public List<Warehouse> getAllWarehouseAreas() {
-        List<Warehouse> warehouses = this.list(new LambdaQueryWrapper<>());
+    public List<Warehouse> getAllWarehouseAreas(Map<String, Object> param) {
+        List<Warehouse> filters = new ArrayList<>();
+        List<Long> wareIds = new ArrayList<>();
+        Object name = param.get("name");
+        boolean isEmpty = !Objects.isNull(name) && !StringUtils.isBlank(name.toString());
+        if (isEmpty) {
+            filters = this.baseMapper.selectByName(name.toString());
+            wareIds =  filters.stream().map(Warehouse::getId).collect(Collectors.toList());
+        }
+        List<Warehouse> warehouses  = this.list(new LambdaQueryWrapper<Warehouse>().in(!wareIds.isEmpty(), Warehouse::getId, wareIds));
         if (warehouses.isEmpty()) {
             throw new CoolException("浠撳簱涓虹┖锛岃娣诲姞浠撳簱鍚庡啀鎿嶄綔锛侊紒");
         }
-        List<WarehouseAreas> warehouseAreas = warehouseAreasService.list(new LambdaQueryWrapper<>());
+        List<Long> list = warehouses.stream().map(Warehouse::getId).collect(Collectors.toList());
+        List<WarehouseAreas> warehouseAreas = warehouseAreasService.list(new LambdaQueryWrapper<WarehouseAreas>()
+                .like(isEmpty, WarehouseAreas::getName, !Objects.isNull(name) ? name.toString() : "")
+                .in(!warehouses.isEmpty(), WarehouseAreas::getWarehouseId, list)
+        );
         if (warehouseAreas.isEmpty()) {
             return warehouses;
         }
         warehouses.forEach(warehouse -> {
+            warehouse.setFlagWare(1);
             List<WarehouseAreas> areas = new ArrayList<>();
             warehouseAreas.forEach(warehouseAreas1 -> {
                 if (warehouse.getId().equals(warehouseAreas1.getWarehouseId())) {
+                    warehouseAreas1.setFlagWare(0);
                     areas.add(warehouseAreas1);
                 }
             });

--
Gitblit v1.9.1