From 206c1338ae6f5100cb408fb02a2095e211c0521f Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期一, 22 九月 2025 16:51:03 +0800
Subject: [PATCH] 资产入库联调修改

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/OaApprovalApplicationStorageServiceImpl.java |  270 ++++++++++++++++++++---------------------------------
 1 files changed, 104 insertions(+), 166 deletions(-)

diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/OaApprovalApplicationStorageServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/OaApprovalApplicationStorageServiceImpl.java
index a197e1c..6332ff5 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/OaApprovalApplicationStorageServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/OaApprovalApplicationStorageServiceImpl.java
@@ -12,23 +12,23 @@
 import com.ruoyi.system.emums.ApprovalTypeEnum;
 import com.ruoyi.system.emums.AssetTypeEnum;
 import com.ruoyi.system.mapper.OaApprovalApplicationStorageMapper;
-import com.ruoyi.system.model.AssetGeneralExt;
 import com.ruoyi.system.model.AssetMain;
-import com.ruoyi.system.model.AssetPropertyExt;
-import com.ruoyi.system.model.AssetVehicleExt;
 import com.ruoyi.system.model.AssetWarehouse;
 import com.ruoyi.system.model.OaApprovalApplicationStorage;
+import com.ruoyi.system.model.OaApprovalApplicationStorageAsset;
+import com.ruoyi.system.model.OaApprovalApplicationStorageAssetPropertyExt;
+import com.ruoyi.system.model.OaApprovalApplicationStorageAssetVehicleExt;
 import com.ruoyi.system.model.OaApprovalApplications;
 import com.ruoyi.system.model.OaApprovalFlowNode;
 import com.ruoyi.system.query.OaApprovalApplicationStoragePageQuery;
-import com.ruoyi.system.service.AssetGeneralExtService;
 import com.ruoyi.system.service.AssetMainService;
 import com.ruoyi.system.service.AssetPropertyExtService;
 import com.ruoyi.system.service.AssetTypeService;
 import com.ruoyi.system.service.AssetVehicleExtService;
 import com.ruoyi.system.service.AssetWarehouseService;
-import com.ruoyi.system.service.ISysDeptService;
-import com.ruoyi.system.service.ISysUserService;
+import com.ruoyi.system.service.OaApprovalApplicationStorageAssetPropertyExtService;
+import com.ruoyi.system.service.OaApprovalApplicationStorageAssetService;
+import com.ruoyi.system.service.OaApprovalApplicationStorageAssetVehicleExtService;
 import com.ruoyi.system.service.OaApprovalApplicationStorageService;
 import com.ruoyi.system.service.OaApprovalApplicationsService;
 import com.ruoyi.system.service.OaApprovalFlowNodeService;
@@ -43,7 +43,6 @@
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
 
-import java.math.BigDecimal;
 import java.time.LocalDate;
 import java.time.format.DateTimeFormatter;
 import java.util.ArrayList;
@@ -64,15 +63,15 @@
 public class OaApprovalApplicationStorageServiceImpl extends ServiceImpl<OaApprovalApplicationStorageMapper, OaApprovalApplicationStorage> implements OaApprovalApplicationStorageService {
 
     private final OaApprovalApplicationsService oaApprovalApplicationsService;
+    private final OaApprovalApplicationStorageAssetService oaApprovalApplicationStorageAssetService;
+    private final OaApprovalApplicationStorageAssetPropertyExtService oaApprovalApplicationStorageAssetPropertyExtService;
+    private final OaApprovalApplicationStorageAssetVehicleExtService oaApprovalApplicationStorageAssetVehicleExtService;
     private final AssetMainService assetMainService;
-    private final AssetGeneralExtService assetGeneralExtService;
     private final AssetPropertyExtService assetPropertyExtService;
     private final AssetVehicleExtService assetVehicleExtService;
     private final AssetTypeService assetTypeService;
     private final OaApprovalFlowNodeService oaApprovalFlowNodeService;
     private final OaApprovalTodoService oaApprovalTodoService;
-    private final ISysUserService sysUserService;
-    private final ISysDeptService sysDeptService;
     private final TDeptService tDeptService;
     private final AssetWarehouseService assetWarehouseService;
 
@@ -146,104 +145,78 @@
      * 保存通用资产数据
      */
     private void saveGeneralAssets(List<OaApprovalApplicationStorageGeneralDTO.GeneralAssetItemDTO> assetItems, Integer applicationId, Integer assetTypeId, LocalDate storageDate, OaApprovalApplicationStorageGeneralDTO baseDto) {
-        List<AssetMain> allAssetMains = new ArrayList<>();
+        List<OaApprovalApplicationStorageAsset> storageAssets = new ArrayList<>();
 
         for (OaApprovalApplicationStorageGeneralDTO.GeneralAssetItemDTO item : assetItems) {
             // 根据数量创建对应数量的资产记录
-            int quantity = item.getQuantity().intValue();
-            for (int i = 0; i < quantity; i++) {
-                AssetMain assetMain = buildAssetMain(item, applicationId, assetTypeId, storageDate, baseDto);
-                assetMain.setAssetMainType(AssetTypeEnum.GENERAL.getCode());
-                assetMain.setQuantity(BigDecimal.ONE); // 每个资产记录数量为1
-                assetMain.setTotalValue(item.getUnitPrice()); // 总价值等于单价
-                allAssetMains.add(assetMain);
-            }
+            OaApprovalApplicationStorageAsset assetMain = buildAssetMain(item, applicationId, assetTypeId, storageDate, baseDto);
+            assetMain.setAssetMainType(AssetTypeEnum.GENERAL.getCode());
+            storageAssets.add(assetMain);
         }
-        
         // 批量保存资产主表数据
-        assetMainService.saveBatch(allAssetMains);
+        oaApprovalApplicationStorageAssetService.saveBatch(storageAssets);
     }
 
     /**
      * 保存房产资产数据
      */
     private void savePropertyAssets(List<OaApprovalApplicationStoragePropertyDTO.PropertyAssetItemDTO> assetItems, Integer applicationId, Integer assetTypeId, LocalDate storageDate, OaApprovalApplicationStoragePropertyDTO baseDto) {
-        List<AssetMain> allAssetMains = new ArrayList<>();
-        List<AssetPropertyExt> allPropertyExts = new ArrayList<>();
+        List<OaApprovalApplicationStorageAsset> allAssetMains = new ArrayList<>();
+        List<OaApprovalApplicationStorageAssetPropertyExt> allPropertyExts = new ArrayList<>();
         
         for (OaApprovalApplicationStoragePropertyDTO.PropertyAssetItemDTO item : assetItems) {
-            // 根据数量创建对应数量的资产记录
-            int quantity = item.getQuantity().intValue();
-            for (int i = 0; i < quantity; i++) {
-                AssetMain assetMain = buildAssetMainFromProperty(item, applicationId, assetTypeId, storageDate, baseDto);
-                assetMain.setAssetMainType(AssetTypeEnum.HOUSE.getCode());
-                assetMain.setQuantity(BigDecimal.ONE); // 每个资产记录数量为1
-                assetMain.setTotalValue(item.getUnitPrice()); // 总价值等于单价
-                allAssetMains.add(assetMain);
-            }
+            OaApprovalApplicationStorageAsset assetMain = buildAssetMainFromProperty(item, applicationId, assetTypeId, storageDate, baseDto);
+            assetMain.setAssetMainType(AssetTypeEnum.HOUSE.getCode());
+            allAssetMains.add(assetMain);
         }
         
         // 批量保存资产主表数据
-        assetMainService.saveBatch(allAssetMains);
+        oaApprovalApplicationStorageAssetService.saveBatch(allAssetMains);
 
         // 为每个资产主表记录创建对应的扩展信息
-        int assetMainIndex = 0;
-        for (OaApprovalApplicationStoragePropertyDTO.PropertyAssetItemDTO item : assetItems) {
-            int quantity = item.getQuantity().intValue();
-            for (int i = 0; i < quantity; i++) {
-                AssetPropertyExt propertyExt = buildAssetPropertyExt(item, allAssetMains.get(assetMainIndex).getId());
+        for (OaApprovalApplicationStorageAsset allAssetMain : allAssetMains) {
+            for (OaApprovalApplicationStoragePropertyDTO.PropertyAssetItemDTO item : assetItems) {
+                OaApprovalApplicationStorageAssetPropertyExt propertyExt = buildAssetPropertyExt(item, allAssetMain.getId());
                 allPropertyExts.add(propertyExt);
-                assetMainIndex++;
             }
         }
-        
-        assetPropertyExtService.saveBatch(allPropertyExts);
+        oaApprovalApplicationStorageAssetPropertyExtService.saveBatch(allPropertyExts);
     }
 
     /**
      * 保存车辆资产数据
      */
     private void saveVehicleAssets(List<OaApprovalApplicationStorageVehicleDTO.VehicleAssetItemDTO> assetItems, Integer applicationId, Integer assetTypeId, LocalDate storageDate, OaApprovalApplicationStorageVehicleDTO baseDto) {
-        List<AssetMain> allAssetMains = new ArrayList<>();
-        List<AssetVehicleExt> allVehicleExts = new ArrayList<>();
-        
+        List<OaApprovalApplicationStorageAsset> allAssetMains = new ArrayList<>();
+        List<OaApprovalApplicationStorageAssetVehicleExt> allVehicleExts = new ArrayList<>();
+
         for (OaApprovalApplicationStorageVehicleDTO.VehicleAssetItemDTO item : assetItems) {
-            // 根据数量创建对应数量的资产记录
-            int quantity = item.getQuantity().intValue();
-            for (int i = 0; i < quantity; i++) {
-                AssetMain assetMain = buildAssetMainFromVehicle(item, applicationId, assetTypeId, storageDate, baseDto);
-                assetMain.setAssetMainType(AssetTypeEnum.VEHICLE.getCode());
-                assetMain.setQuantity(BigDecimal.ONE); // 每个资产记录数量为1
-                assetMain.setTotalValue(item.getUnitPrice()); // 总价值等于单价
-                allAssetMains.add(assetMain);
-            }
+            OaApprovalApplicationStorageAsset assetMain = buildAssetMainFromVehicle(item, applicationId, assetTypeId, storageDate, baseDto);
+            assetMain.setAssetMainType(AssetTypeEnum.VEHICLE.getCode());
+            allAssetMains.add(assetMain);
         }
         
         // 批量保存资产主表数据
-        assetMainService.saveBatch(allAssetMains);
+        oaApprovalApplicationStorageAssetService.saveBatch(allAssetMains);
 
         // 为每个资产主表记录创建对应的扩展信息
-        int assetMainIndex = 0;
-        for (OaApprovalApplicationStorageVehicleDTO.VehicleAssetItemDTO item : assetItems) {
-            int quantity = item.getQuantity().intValue();
-            for (int i = 0; i < quantity; i++) {
-                AssetVehicleExt vehicleExt = buildAssetVehicleExt(item, allAssetMains.get(assetMainIndex).getId());
+        for (OaApprovalApplicationStorageAsset allAssetMain : allAssetMains) {
+            for (OaApprovalApplicationStorageVehicleDTO.VehicleAssetItemDTO item : assetItems) {
+                OaApprovalApplicationStorageAssetVehicleExt vehicleExt = buildAssetVehicleExt(item, allAssetMain.getId());
                 allVehicleExts.add(vehicleExt);
-                assetMainIndex++;
             }
+
         }
-        
-        assetVehicleExtService.saveBatch(allVehicleExts);
+        oaApprovalApplicationStorageAssetVehicleExtService.saveBatch(allVehicleExts);
     }
 
     /**
      * 构建通用资产主表数据
      */
-    private AssetMain buildAssetMain(OaApprovalApplicationStorageGeneralDTO.GeneralAssetItemDTO item, Integer applicationId, Integer assetTypeId, LocalDate storageDate, OaApprovalApplicationStorageGeneralDTO baseDto) {
-        AssetMain assetMain = new AssetMain();
+    private OaApprovalApplicationStorageAsset buildAssetMain(OaApprovalApplicationStorageGeneralDTO.GeneralAssetItemDTO item, Integer applicationId, Integer assetTypeId, LocalDate storageDate, OaApprovalApplicationStorageGeneralDTO baseDto) {
+        OaApprovalApplicationStorageAsset assetMain = new OaApprovalApplicationStorageAsset();
         assetMain.setApprovalApplicationId(applicationId.intValue());
         assetMain.setAssetOriginalCode(item.getAssetOriginalCode());
-        assetMain.setAssetCode(generateAssetCode(assetTypeId, storageDate)); // 系统生成资产编码
         assetMain.setAssetName(item.getAssetName());
         assetMain.setCategory(item.getCategory());
         assetMain.setSpecificationModel(item.getSpecificationModel());
@@ -262,8 +235,8 @@
         }
         assetMain.setUserName(item.getUserName());
         // 根据整单 addressType 写入使用部门ID或仓库ID
-        applyAddressToAssetMain(assetMain, baseDto.getAddressType(), baseDto.getUseDeptName(), baseDto.getWarehouseName(), baseDto.getAddress());
-        assetMain.setAssetMainType(baseDto.getAddressType());
+        applyUseDeptInfoToAssetMain(assetMain, baseDto.getUseDeptName());
+        assetMain.setAssetMainType(AssetTypeEnum.GENERAL.getCode());
         assetMain.setAssetStatus(item.getAssetStatus());
         assetMain.setRemarks(item.getRemarks());
         assetMain.setAccountingStatus(item.getAccountingStatus());
@@ -278,11 +251,10 @@
     /**
      * 构建房产资产主表数据
      */
-    private AssetMain buildAssetMainFromProperty(OaApprovalApplicationStoragePropertyDTO.PropertyAssetItemDTO item, Integer applicationId, Integer assetTypeId, LocalDate storageDate, OaApprovalApplicationStoragePropertyDTO baseDto) {
-        AssetMain assetMain = new AssetMain();
+    private OaApprovalApplicationStorageAsset buildAssetMainFromProperty(OaApprovalApplicationStoragePropertyDTO.PropertyAssetItemDTO item, Integer applicationId, Integer assetTypeId, LocalDate storageDate, OaApprovalApplicationStoragePropertyDTO baseDto) {
+        OaApprovalApplicationStorageAsset assetMain = new OaApprovalApplicationStorageAsset();
         assetMain.setApprovalApplicationId(applicationId.intValue());
         assetMain.setAssetOriginalCode(item.getAssetOriginalCode());
-        assetMain.setAssetCode(generateAssetCode(assetTypeId, storageDate));
         assetMain.setAssetName(item.getAssetName());
         assetMain.setCategory(item.getCategory());
         assetMain.setSpecificationModel(item.getSpecificationModel());
@@ -299,8 +271,8 @@
             }
         }
         assetMain.setUserName(item.getUserName());
-        applyAddressToAssetMain(assetMain, baseDto.getAddressType(), baseDto.getUseDeptName(), baseDto.getWarehouseName(), baseDto.getAddress());
-        assetMain.setAssetMainType(baseDto.getAddressType());
+        applyUseDeptInfoToAssetMain(assetMain, baseDto.getUseDeptName());
+        assetMain.setAssetMainType(AssetTypeEnum.HOUSE.getCode());
         assetMain.setAssetStatus(item.getAssetStatus());
         assetMain.setRemarks(item.getRemarks());
         assetMain.setAccountingStatus(item.getAccountingStatus());
@@ -315,11 +287,10 @@
     /**
      * 构建车辆资产主表数据
      */
-    private AssetMain buildAssetMainFromVehicle(OaApprovalApplicationStorageVehicleDTO.VehicleAssetItemDTO item, Integer applicationId, Integer assetTypeId, LocalDate storageDate, OaApprovalApplicationStorageVehicleDTO baseDto) {
-        AssetMain assetMain = new AssetMain();
+    private OaApprovalApplicationStorageAsset buildAssetMainFromVehicle(OaApprovalApplicationStorageVehicleDTO.VehicleAssetItemDTO item, Integer applicationId, Integer assetTypeId, LocalDate storageDate, OaApprovalApplicationStorageVehicleDTO baseDto) {
+        OaApprovalApplicationStorageAsset assetMain = new OaApprovalApplicationStorageAsset();
         assetMain.setApprovalApplicationId(applicationId.intValue());
         assetMain.setAssetOriginalCode(item.getAssetOriginalCode());
-        assetMain.setAssetCode(generateAssetCode(assetTypeId, storageDate));
         assetMain.setAssetName(item.getAssetName());
         assetMain.setCategory(item.getCategory());
         assetMain.setSpecificationModel(item.getSpecificationModel());
@@ -336,8 +307,8 @@
             }
         }
         assetMain.setUserName(item.getUserName());
-        applyAddressToAssetMain(assetMain, baseDto.getAddressType(), baseDto.getUseDeptName(), baseDto.getWarehouseName(), baseDto.getAddress());
-        assetMain.setAssetMainType(baseDto.getAddressType());
+        applyUseDeptInfoToAssetMain(assetMain, baseDto.getUseDeptName());
+        assetMain.setAssetMainType(AssetTypeEnum.VEHICLE.getCode());
         assetMain.setAssetStatus(item.getAssetStatus());
         assetMain.setRemarks(item.getRemarks());
         assetMain.setAccountingStatus(item.getAccountingStatus());
@@ -352,59 +323,25 @@
     /**
      * 按整单地址类型设置资产主表位置信息(使用部门ID或仓库ID或地址)
      */
-    private void applyAddressToAssetMain(AssetMain assetMain, Integer addressType, String useDeptName, String warehouseName, String address) {
-        if (addressType == null) {
-            return;
-        }
-        switch (addressType) {
-            case 0:
-                // 使用部门
-                if (useDeptName != null && !useDeptName.isEmpty()) {
-                    TDept dept = tDeptService.lambdaQuery()
-                            .eq(TDept::getDeptName, useDeptName)
-                            .one();
-                    if (dept != null) {
-                        assetMain.setUseDeptId(dept.getId());
-                    }
-                }
-                break;
-            case 1:
-                if (warehouseName != null && !warehouseName.isEmpty()) {
-                    AssetWarehouse warehouse = assetWarehouseService.lambdaQuery()
-                            .eq(AssetWarehouse::getWarehouseName, warehouseName)
-                            .one();
-                    if (warehouse != null) {
-                        assetMain.setWarehouseId(warehouse.getId());
-                        assetMain.setWarehouseName(warehouse.getWarehouseName());
-                    }
-                }
-                break;
-            case 2:
-                if (address != null && !address.isEmpty()) {
-                    assetMain.setAddress(address);
-                }
-                break;
-            default:
-                // ignore
+    private void applyUseDeptInfoToAssetMain(OaApprovalApplicationStorageAsset assetMain, String useDeptName) {
+        // 使用部门
+        if (useDeptName != null && !useDeptName.isEmpty()) {
+            TDept dept = tDeptService.lambdaQuery()
+                    .eq(TDept::getDeptName, useDeptName)
+                    .one();
+            if (dept != null) {
+                assetMain.setUseDeptId(dept.getId());
+            }
         }
     }
 
-    /**
-     * 构建通用资产扩展数据
-     */
-    private AssetGeneralExt buildAssetGeneralExt(OaApprovalApplicationStorageGeneralDTO.GeneralAssetItemDTO item, Integer assetMainId) {
-        AssetGeneralExt generalExt = new AssetGeneralExt();
-        generalExt.setAssetMainId(assetMainId);
-        generalExt.setDisabled(false);
-        return generalExt;
-    }
 
     /**
      * 构建房产资产扩展数据
      */
-    private AssetPropertyExt buildAssetPropertyExt(OaApprovalApplicationStoragePropertyDTO.PropertyAssetItemDTO item, Integer assetMainId) {
-        AssetPropertyExt propertyExt = new AssetPropertyExt();
-        propertyExt.setAssetMainId(assetMainId);
+    private OaApprovalApplicationStorageAssetPropertyExt buildAssetPropertyExt(OaApprovalApplicationStoragePropertyDTO.PropertyAssetItemDTO item, Integer assetMainId) {
+        OaApprovalApplicationStorageAssetPropertyExt propertyExt = new OaApprovalApplicationStorageAssetPropertyExt();
+        propertyExt.setStorageAssetId(assetMainId);
         propertyExt.setRegion(item.getRegion());
         propertyExt.setDesignPurpose(item.getDesignPurpose());
         propertyExt.setBuilding(item.getBuilding());
@@ -428,9 +365,9 @@
     /**
      * 构建车辆资产扩展数据
      */
-    private AssetVehicleExt buildAssetVehicleExt(OaApprovalApplicationStorageVehicleDTO.VehicleAssetItemDTO item, Integer assetMainId) {
-        AssetVehicleExt vehicleExt = new AssetVehicleExt();
-        vehicleExt.setAssetMainId(assetMainId);
+    private OaApprovalApplicationStorageAssetVehicleExt buildAssetVehicleExt(OaApprovalApplicationStorageVehicleDTO.VehicleAssetItemDTO item, Integer assetMainId) {
+        OaApprovalApplicationStorageAssetVehicleExt vehicleExt = new OaApprovalApplicationStorageAssetVehicleExt();
+        vehicleExt.setStorageAssetId(assetMainId);
         vehicleExt.setLicensePlate(item.getLicensePlate());
         vehicleExt.setVinCode(item.getVinCode());
         vehicleExt.setEngineNumber(item.getEngineNumber());
@@ -598,16 +535,15 @@
         OaApprovalApplicationStorageGeneralDetailVO vo = new OaApprovalApplicationStorageGeneralDetailVO();
         fillBaseFields(vo, app, storage);
 
-        List<AssetMain> mains = assetMainService.lambdaQuery()
-                .eq(AssetMain::getApprovalApplicationId, app.getId())
-                .eq(AssetMain::getAssetTypeId, storage.getAssetTypeId())
+        List<OaApprovalApplicationStorageAsset> mains = oaApprovalApplicationStorageAssetService.lambdaQuery()
+                .eq(OaApprovalApplicationStorageAsset::getApprovalApplicationId, app.getId())
+                .eq(OaApprovalApplicationStorageAsset::getAssetTypeId, storage.getAssetTypeId())
                 .list();
         setAddressInfo(vo, mains);
 
         if (!CollectionUtils.isEmpty(mains)) {
-            List<Integer> mainIds = mains.stream().map(AssetMain::getId).collect(Collectors.toList());
             // 批量查询权属部门,避免循环内查询
-            List<Integer> ownerIds = mains.stream().map(AssetMain::getOwnershipDeptId)
+            List<Integer> ownerIds = mains.stream().map(OaApprovalApplicationStorageAsset::getOwnershipDeptId)
                     .filter(id -> id != null).distinct().collect(Collectors.toList());
             Map<Integer, TDept> ownerDeptMap = ownerIds.isEmpty() ? java.util.Collections.emptyMap()
                     : tDeptService.listByIds(ownerIds).stream().collect(Collectors.toMap(TDept::getId, d -> d));
@@ -654,18 +590,18 @@
         OaApprovalApplicationStoragePropertyDetailVO vo = new OaApprovalApplicationStoragePropertyDetailVO();
         fillBaseFields(vo, app, storage);
 
-        List<AssetMain> mains = assetMainService.lambdaQuery()
-                .eq(AssetMain::getApprovalApplicationId, app.getId())
-                .eq(AssetMain::getAssetTypeId, storage.getAssetTypeId())
+        List<OaApprovalApplicationStorageAsset> mains = oaApprovalApplicationStorageAssetService.lambdaQuery()
+                .eq(OaApprovalApplicationStorageAsset::getApprovalApplicationId, app.getId())
+                .eq(OaApprovalApplicationStorageAsset::getAssetTypeId, storage.getAssetTypeId())
                 .list();
         setAddressInfo(vo, mains);
 
         if (!CollectionUtils.isEmpty(mains)) {
-            List<Integer> mainIds = mains.stream().map(AssetMain::getId).collect(Collectors.toList());
-            List<AssetPropertyExt> exts = assetPropertyExtService.lambdaQuery().in(AssetPropertyExt::getAssetMainId, mainIds).list();
-            Map<Integer, AssetPropertyExt> extMap = exts.stream().collect(Collectors.toMap(AssetPropertyExt::getAssetMainId, e -> e));
+            List<Integer> mainIds = mains.stream().map(OaApprovalApplicationStorageAsset::getId).collect(Collectors.toList());
+            List<OaApprovalApplicationStorageAssetPropertyExt> exts = oaApprovalApplicationStorageAssetPropertyExtService.lambdaQuery().in(OaApprovalApplicationStorageAssetPropertyExt::getStorageAssetId, mainIds).list();
+            Map<Integer, OaApprovalApplicationStorageAssetPropertyExt> extMap = exts.stream().collect(Collectors.toMap(OaApprovalApplicationStorageAssetPropertyExt::getStorageAssetId, e -> e));
             // 批量查询权属部门
-            List<Integer> ownerIds = mains.stream().map(AssetMain::getOwnershipDeptId)
+            List<Integer> ownerIds = mains.stream().map(OaApprovalApplicationStorageAsset::getOwnershipDeptId)
                     .filter(id -> id != null).distinct().collect(Collectors.toList());
             Map<Integer, TDept> ownerDeptMap = ownerIds.isEmpty() ? java.util.Collections.emptyMap()
                     : tDeptService.listByIds(ownerIds).stream().collect(Collectors.toMap(TDept::getId, d -> d));
@@ -691,7 +627,7 @@
                 item.setAccountingVoucherNo(m.getAccountingVoucherNo());
                 item.setAccountingSubject(m.getAccountingSubject());
                 item.setAccountingAmount(m.getAccountingAmount());
-                AssetPropertyExt ext = extMap.get(m.getId());
+                OaApprovalApplicationStorageAssetPropertyExt ext = extMap.get(m.getId());
                 if (ext != null) {
                     item.setRegion(ext.getRegion());
                     item.setDesignPurpose(ext.getDesignPurpose());
@@ -731,18 +667,18 @@
         OaApprovalApplicationStorageVehicleDetailVO vo = new OaApprovalApplicationStorageVehicleDetailVO();
         fillBaseFields(vo, app, storage);
 
-        List<AssetMain> mains = assetMainService.lambdaQuery()
-                .eq(AssetMain::getApprovalApplicationId, app.getId())
-                .eq(AssetMain::getAssetTypeId, storage.getAssetTypeId())
+        List<OaApprovalApplicationStorageAsset> mains = oaApprovalApplicationStorageAssetService.lambdaQuery()
+                .eq(OaApprovalApplicationStorageAsset::getApprovalApplicationId, app.getId())
+                .eq(OaApprovalApplicationStorageAsset::getAssetTypeId, storage.getAssetTypeId())
                 .list();
         setAddressInfo(vo, mains);
 
         if (!CollectionUtils.isEmpty(mains)) {
-            List<Integer> mainIds = mains.stream().map(AssetMain::getId).collect(Collectors.toList());
-            List<AssetVehicleExt> exts = assetVehicleExtService.lambdaQuery().in(AssetVehicleExt::getAssetMainId, mainIds).list();
-            Map<Integer, AssetVehicleExt> extMap = exts.stream().collect(Collectors.toMap(AssetVehicleExt::getAssetMainId, e -> e));
+            List<Integer> mainIds = mains.stream().map(OaApprovalApplicationStorageAsset::getId).collect(Collectors.toList());
+            List<OaApprovalApplicationStorageAssetVehicleExt> exts = oaApprovalApplicationStorageAssetVehicleExtService.lambdaQuery().in(OaApprovalApplicationStorageAssetVehicleExt::getStorageAssetId, mainIds).list();
+            Map<Integer, OaApprovalApplicationStorageAssetVehicleExt> extMap = exts.stream().collect(Collectors.toMap(OaApprovalApplicationStorageAssetVehicleExt::getStorageAssetId, e -> e));
             // 批量查询权属部门
-            List<Integer> ownerIds = mains.stream().map(AssetMain::getOwnershipDeptId)
+            List<Integer> ownerIds = mains.stream().map(OaApprovalApplicationStorageAsset::getOwnershipDeptId)
                     .filter(id -> id != null).distinct().collect(Collectors.toList());
             Map<Integer, TDept> ownerDeptMap = ownerIds.isEmpty() ? java.util.Collections.emptyMap()
                     : tDeptService.listByIds(ownerIds).stream().collect(Collectors.toMap(TDept::getId, d -> d));
@@ -768,7 +704,7 @@
                 item.setAccountingVoucherNo(m.getAccountingVoucherNo());
                 item.setAccountingSubject(m.getAccountingSubject());
                 item.setAccountingAmount(m.getAccountingAmount());
-                AssetVehicleExt ext = extMap.get(m.getId());
+                OaApprovalApplicationStorageAssetVehicleExt ext = extMap.get(m.getId());
                 if (ext != null) {
                     item.setLicensePlate(ext.getLicensePlate());
                     item.setVinCode(ext.getVinCode());
@@ -829,46 +765,48 @@
         }
     }
 
-    private void setAddressInfo(Object vo, List<AssetMain> mains) {
+    private void setAddressInfo(Object vo, List<OaApprovalApplicationStorageAsset> mains) {
         if (CollectionUtils.isEmpty(mains)) {
             return;
         }
-        AssetMain sample = mains.get(0);
+        OaApprovalApplicationStorageAsset sample = mains.get(0);
         Integer addressType = null;
         String useDeptName = null;
         String warehouseName = null;
         String address = null;
-        if (sample.getUseDeptId() != null) {
-            addressType = 0;
-            TDept dept = tDeptService.getById(sample.getUseDeptId());
-            useDeptName = dept != null ? dept.getDeptName() : null;
-        } else if (sample.getWarehouseId() != null) {
-            addressType = 1;
-            AssetWarehouse wh = assetWarehouseService.getById(sample.getWarehouseId());
-            warehouseName = wh != null ? wh.getWarehouseName() : null;
-        } else if (sample.getAddress() != null && !sample.getAddress().isEmpty()) {
-            addressType = 2;
-            address = sample.getAddress();
+        switch (sample.getAddressType()) {
+            case 0:
+                TDept dept = tDeptService.getById(sample.getUseDeptId());
+                useDeptName = dept != null ? dept.getDeptName() : null;
+                break;
+            case 1:
+                AssetWarehouse wh = assetWarehouseService.getById(sample.getWarehouseId());
+                warehouseName = wh != null ? wh.getWarehouseName() : null;
+                break;
+            case 2:
+                break;
+            default:
+                return;
         }
 
         if (vo instanceof OaApprovalApplicationStorageGeneralDetailVO) {
             OaApprovalApplicationStorageGeneralDetailVO v = (OaApprovalApplicationStorageGeneralDetailVO) vo;
-            v.setAddressType(addressType);
+            v.setAddressType(sample.getAddressType());
             v.setUseDeptName(useDeptName);
             v.setWarehouseName(warehouseName);
-            v.setAddress(address);
+            v.setAddress(sample.getAddress());
         } else if (vo instanceof OaApprovalApplicationStoragePropertyDetailVO) {
             OaApprovalApplicationStoragePropertyDetailVO v = (OaApprovalApplicationStoragePropertyDetailVO) vo;
-            v.setAddressType(addressType);
+            v.setAddressType(sample.getAddressType());
             v.setUseDeptName(useDeptName);
             v.setWarehouseName(warehouseName);
-            v.setAddress(address);
+            v.setAddress(sample.getAddress());
         } else if (vo instanceof OaApprovalApplicationStorageVehicleDetailVO) {
             OaApprovalApplicationStorageVehicleDetailVO v = (OaApprovalApplicationStorageVehicleDetailVO) vo;
-            v.setAddressType(addressType);
+            v.setAddressType(sample.getAddressType());
             v.setUseDeptName(useDeptName);
             v.setWarehouseName(warehouseName);
-            v.setAddress(address);
+            v.setAddress(sample.getAddress());
         }
     }
 }

--
Gitblit v1.7.1