diff --git a/api-admin/src/main/java/com/glxp/sale/admin/controller/inventory/InvProductsController.java b/api-admin/src/main/java/com/glxp/sale/admin/controller/inventory/InvProductsController.java
index 90a0610..5dec685 100644
--- a/api-admin/src/main/java/com/glxp/sale/admin/controller/inventory/InvProductsController.java
+++ b/api-admin/src/main/java/com/glxp/sale/admin/controller/inventory/InvProductsController.java
@@ -47,10 +47,7 @@ import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RestController;
 
 import javax.annotation.Resource;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 import java.util.stream.Collectors;
 
 
diff --git a/api-admin/src/main/resources/mybatis/mapper/inventory/InvProductDao.xml b/api-admin/src/main/resources/mybatis/mapper/inventory/InvProductDao.xml
index 6f142f9..a528401 100644
--- a/api-admin/src/main/resources/mybatis/mapper/inventory/InvProductDao.xml
+++ b/api-admin/src/main/resources/mybatis/mapper/inventory/InvProductDao.xml
@@ -239,6 +239,9 @@
             <if test="invStorageCode != '' and invStorageCode != null">
                 AND invStorageCode like concat('%', #{invStorageCode,jdbcType=VARCHAR}, '%')
             </if>
+            <if test="invWarehouseCode != '' and invWarehouseCode != null">
+                AND invWarehouseCode like concat('%', #{invWarehouseCode,jdbcType=VARCHAR}, '%')
+            </if>
             <if test="batchNo != '' and batchNo != null">
                 AND ip.batchNo like concat('%', #{batchNo,jdbcType=VARCHAR}, '%')
             </if>