From 92c85d94810a6b516a87e897244a8b86af778219 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=B9=E9=B9=8F=E9=A3=9E?= <rakor2010@gmail.com>
Date: Tue, 16 Jun 2026 16:06:29 +0800
Subject: [PATCH] =?UTF-8?q?feat(inventory):=20=E6=B7=BB=E5=8A=A0=E5=BA=93?=
 =?UTF-8?q?=E5=AD=98=E6=95=B0=E9=87=8F=E6=9F=A5=E8=AF=A2=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 新增 MaterialInventoryVO 视图对象用于返回库存数量信息
- 实现按工厂分组汇总库存数量的功能
- 添加 getTotalNum 和 getItems 方法返回总计和明细数据
- 在 InventoryController 中新增 getCount 接口提供库存查询服务
- 使用 Stream API 进行数据分组和数量累加计算
- 集成 BigDecimal 类型确保数值计算精度
---
 .../admin/controller/InventoryController.java | 36 +++++++++++++++
 .../admin/pojo/vo/MaterialInventoryVO.java    | 46 +++++++++++++++++++
 2 files changed, 82 insertions(+)
 create mode 100644 nflg-wms-admin/src/main/java/com/nflg/wms/admin/pojo/vo/MaterialInventoryVO.java

diff --git a/nflg-wms-admin/src/main/java/com/nflg/wms/admin/controller/InventoryController.java b/nflg-wms-admin/src/main/java/com/nflg/wms/admin/controller/InventoryController.java
index d8c57492..a5e245fb 100644
--- a/nflg-wms-admin/src/main/java/com/nflg/wms/admin/controller/InventoryController.java
+++ b/nflg-wms-admin/src/main/java/com/nflg/wms/admin/controller/InventoryController.java
@@ -4,6 +4,7 @@ import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.util.IdUtil;
 import cn.hutool.core.util.NumberUtil;
 import cn.hutool.core.util.StrUtil;
+import com.nflg.wms.admin.pojo.vo.MaterialInventoryVO;
 import com.nflg.wms.admin.repository.InventoryCheckTaskScanRecordResitory;
 import com.nflg.wms.admin.service.QmsService;
 import com.nflg.wms.common.constant.STATE;
@@ -344,4 +345,39 @@ public class InventoryController extends BaseController {
             }
         }
     }
+
+    /**
+     * 获取库存数量
+     * @param materialNo 物料编号
+     */
+    @GetMapping("getCount")
+    public ApiResult<MaterialInventoryVO> getCount(@RequestParam String materialNo) {
+        List<WmsInventory> inventories = inventoryService.lambdaQuery()
+                .eq(WmsInventory::getMaterialNo, materialNo)
+                .list();
+
+        // 按工厂分组汇总数量
+        List<MaterialInventoryVO.Item> items = inventories.stream()
+                .collect(Collectors.groupingBy(
+                        WmsInventory::getFactoryNo,
+                        Collectors.reducing(BigDecimal.ZERO, WmsInventory::getNum, BigDecimal::add))
+                )
+                .entrySet()
+                .stream()
+                .map(entry -> new MaterialInventoryVO.Item()
+                        .setFactoryNo(entry.getKey())
+                        .setNum(entry.getValue())
+                )
+                .collect(Collectors.toList());
+
+        BigDecimal totalNum = inventories.stream()
+                .map(WmsInventory::getNum)
+                .filter(Objects::nonNull)
+                .reduce(BigDecimal.ZERO, BigDecimal::add);
+
+        MaterialInventoryVO vo = new MaterialInventoryVO()
+                .setTotalNum(totalNum)
+                .setItems(items);
+        return ApiResult.success(vo);
+    }
 }
\ No newline at end of file
diff --git a/nflg-wms-admin/src/main/java/com/nflg/wms/admin/pojo/vo/MaterialInventoryVO.java b/nflg-wms-admin/src/main/java/com/nflg/wms/admin/pojo/vo/MaterialInventoryVO.java
new file mode 100644
index 00000000..9cc43b0e
--- /dev/null
+++ b/nflg-wms-admin/src/main/java/com/nflg/wms/admin/pojo/vo/MaterialInventoryVO.java
@@ -0,0 +1,46 @@
+package com.nflg.wms.admin.pojo.vo;
+
+import lombok.Getter;
+import lombok.Setter;
+import lombok.ToString;
+import lombok.experimental.Accessors;
+
+import java.math.BigDecimal;
+import java.util.List;
+
+/**
+ * 物料库存数量视图对象
+ */
+@Getter
+@Setter
+@ToString
+@Accessors(chain = true)
+public class MaterialInventoryVO {
+
+    /**
+     * 库存总数量（所有库存地点合计）
+     */
+    private BigDecimal totalNum;
+
+    /**
+     * 各库存地点的数量明细
+     */
+    private List<Item> items;
+
+    @Getter
+    @Setter
+    @ToString
+    @Accessors(chain = true)
+    public static class Item {
+
+        /**
+         * 工厂编号
+         */
+        private String factoryNo;
+
+        /**
+         * 数量
+         */
+        private BigDecimal num;
+    }
+}