diff --git a/fw-u8/src/main/java/com/hzya/frame/u8/base/inventory/service/impl/InventoryExtServiceImpl.java b/fw-u8/src/main/java/com/hzya/frame/u8/base/inventory/service/impl/InventoryExtServiceImpl.java
index ecd98f4f..6cbd5bfb 100644
--- a/fw-u8/src/main/java/com/hzya/frame/u8/base/inventory/service/impl/InventoryExtServiceImpl.java
+++ b/fw-u8/src/main/java/com/hzya/frame/u8/base/inventory/service/impl/InventoryExtServiceImpl.java
@@ -42,6 +42,7 @@ public class  InventoryExtServiceImpl extends BaseService<InventoryExtEntity,Str
 		if (null != entity && StrUtil.isNotEmpty(entity.getcInvCode())){
 			InventoryExtEntity extEntity = new InventoryExtEntity();
 			extEntity.setCInvCode(entity.getCInvCode());
+			extEntity.setDataSourceCode(entity.getDataSourceCode());
 			List<InventoryExtEntity> query = inventoryextDao.query(extEntity);
 			if (CollectionUtils.isNotEmpty(query)){
 				if (query.size() > 1){
diff --git a/fw-u8/src/main/java/com/hzya/frame/u8/base/inventory/service/impl/InventorySubServiceImpl.java b/fw-u8/src/main/java/com/hzya/frame/u8/base/inventory/service/impl/InventorySubServiceImpl.java
index 5553df53..0a6fc842 100644
--- a/fw-u8/src/main/java/com/hzya/frame/u8/base/inventory/service/impl/InventorySubServiceImpl.java
+++ b/fw-u8/src/main/java/com/hzya/frame/u8/base/inventory/service/impl/InventorySubServiceImpl.java
@@ -41,6 +41,7 @@ public class InventorySubServiceImpl extends BaseService<InventorySubEntity,Stri
         if (null != entity && StrUtil.isNotEmpty(entity.getcInvSubCode())){
             InventorySubEntity subEntity = new InventorySubEntity();
             subEntity.setcInvSubCode(entity.getcInvSubCode());
+            subEntity.setDataSourceCode(entity.getDataSourceCode());
             List<InventorySubEntity> query = inventorySubDao.query(subEntity);
             if (CollectionUtils.isNotEmpty(query)){
                 if (query.size() > 1){