排版

sd-business/src/main/java/com/sd/business/controller/inventory/InventoryController.java

@@ -5,6 +5,7 @@ import com.ruoyi.common.annotation.NonInterception;
 import com.sd.business.entity.inventory.dto.InventorySelectDto;
 import com.sd.business.entity.inventory.dto.QuantityByWarehouseDto;
 import com.sd.business.entity.inventory.dto.QuantityDto;
+import com.sd.business.entity.inventory.po.Inventory;
 import com.sd.business.entity.inventory.vo.InventoryVo;
 import com.sd.business.entity.inventory.vo.QuantityByWarehouseVo;
 import com.sd.business.service.inventory.InventoryService;
@@ -68,4 +69,13 @@ public class InventoryController {
         return inventoryService.getQuantityByWarehouse(dto);
     }
 
+    /**
+     * 查询库存
+     */
+    @PostMapping("/getInventoryMap")
+    public Map<Long, List<Inventory>> getInventoryMap(@Validated @RequestBody QuantityDto dto) {
+        return inventoryService.getInventoryMap(dto);
+    }
+
+
 }

sd-business/src/main/java/com/sd/business/service/inventory/InventoryService.java

@@ -55,4 +55,9 @@ public interface InventoryService extends BaseService<Inventory> {
      */
     void out(InOutStorageDto dto);
 
+    /**
+     * 库存map
+     */
+    Map<Long, List<Inventory>> getInventoryMap(QuantityDto dto);
+
 }

sd-business/src/main/java/com/sd/business/service/inventory/impl/InventoryServiceImpl.java

@@ -239,6 +239,13 @@ public class InventoryServiceImpl extends ServiceImpl<InventoryMapper, Inventory
         }
     }
 
+    @Override
+    public Map<Long, List<Inventory>> getInventoryMap(QuantityDto dto) {
+        return mapKGroup(Inventory::getBomSpecId, q -> q
+                .eq(Inventory::getWarehouseId, dto.getWarehouseId())
+                .in(Inventory::getBomSpecId, dto.getBomSpecIdList()));
+    }
+
     /**
      * 通过仓库id获取bom规格库存
      */