From 4ad572c4e51e44e3134def3c70879b3a698ab1d8 Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期一, 29 九月 2025 19:56:16 +0800
Subject: [PATCH] 资产变更

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetMainServiceImpl.java |  238 +++++++++++++++++++++++++++++++++++++++++++++-------------
 1 files changed, 183 insertions(+), 55 deletions(-)

diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetMainServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetMainServiceImpl.java
index 055b0e6..3a144ff 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetMainServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/AssetMainServiceImpl.java
@@ -1,11 +1,12 @@
 package com.ruoyi.system.service.impl;
 
 import cn.hutool.core.bean.BeanUtil;
-import org.springframework.util.CollectionUtils;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ruoyi.common.core.domain.entity.TDept;
 import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.system.mapper.AssetMainMapper;
 import com.ruoyi.system.mapper.AssetPropertyExtMapper;
@@ -15,7 +16,10 @@
 import com.ruoyi.system.mapper.OaApprovalApplicationStorageAssetVehicleExtMapper;
 import com.ruoyi.system.model.AssetMain;
 import com.ruoyi.system.model.AssetPropertyExt;
+import com.ruoyi.system.model.AssetType;
 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;
@@ -24,19 +28,29 @@
 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.OaApprovalApplicationStorageAssetPropertyExtService;
 import com.ruoyi.system.service.OaApprovalApplicationStorageAssetService;
 import com.ruoyi.system.service.OaApprovalApplicationStorageAssetVehicleExtService;
+import com.ruoyi.system.service.OaApprovalApplicationStorageService;
+import com.ruoyi.system.service.TDeptService;
 import com.ruoyi.system.vo.asset.AssetMainPageVO;
+import com.ruoyi.system.vo.asset.AssetMainVO;
 import lombok.RequiredArgsConstructor;
 import org.springframework.context.annotation.Lazy;
 import org.springframework.stereotype.Service;
+import org.springframework.util.CollectionUtils;
 
-import java.math.BigDecimal;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
 
 /**
  * 资产主表 服务实现类
@@ -46,6 +60,7 @@
 public class AssetMainServiceImpl extends ServiceImpl<AssetMainMapper, AssetMain> implements AssetMainService {
     private final AssetTypeService assetTypeService;
     private final OaApprovalApplicationStorageAssetService oaApprovalApplicationStorageAssetService;
+    private final OaApprovalApplicationStorageService oaApprovalApplicationStorageService;
     private final OaApprovalApplicationStorageAssetPropertyExtService oaApprovalApplicationStorageAssetPropertyExtService;
     private final OaApprovalApplicationStorageAssetVehicleExtService oaApprovalApplicationStorageAssetVehicleExtService;
     private final AssetVehicleExtService assetVehicleExtService;
@@ -57,6 +72,8 @@
     private final OaApprovalApplicationStorageAssetVehicleExtMapper oaApprovalApplicationStorageAssetVehicleExtMapper;
     private final AssetPropertyExtMapper assetPropertyExtMapper;
     private final AssetVehicleExtMapper assetVehicleExtMapper;
+    private final TDeptService tDeptService;
+    private final AssetWarehouseService assetWarehouseService;
     @Override
     public IPage<AssetMainPageVO> getApprovedPageList(AssetMainPageQuery pageQuery) {
         Page<AssetMainPageVO> page = new Page<>(pageQuery.getPageNum(), pageQuery.getPageSize());
@@ -92,31 +109,56 @@
     public void saveAssetByApprovalApplicationId(Integer approvalApplicationId) {
         // 1. 根据approvalApplicationId查询审批通过的资产列表
         List<OaApprovalApplicationStorageAsset> storageAssets = oaApprovalApplicationStorageAssetMapper
-                .selectList(new QueryWrapper<OaApprovalApplicationStorageAsset>()
-                        .eq("approval_application_id", approvalApplicationId));
+                .selectList(new LambdaQueryWrapper<OaApprovalApplicationStorageAsset>()
+                        .eq(OaApprovalApplicationStorageAsset::getApprovalApplicationId, approvalApplicationId));
         
         if (CollectionUtils.isEmpty(storageAssets)) {
             return;
         }
         
+        List<AssetMain> assetMainList = new ArrayList<>();
+        Map<Integer, Integer> storageAssetIdToAssetMainIdMap = new HashMap<>();
+        Map<Integer, Integer> storageAssetIdToAssetTypeMap = new HashMap<>();
+        
+        // 预查询当天已存在的资产编码数量
+        LocalDate currentDate = LocalDate.now();
+        Map<Integer, Long> assetTypeCountMap = new HashMap<>();
+        
         for (OaApprovalApplicationStorageAsset storageAsset : storageAssets) {
+            Integer assetTypeId = storageAsset.getAssetTypeId();
             Integer assetMainType = storageAsset.getAssetMainType();
-            BigDecimal quantity = storageAsset.getQuantity();
+            Integer quantity = storageAsset.getQuantity();
             
             // 根据数量生成对应条数的资产记录
-            int assetCount = quantity != null ? quantity.intValue() : 1;
+            int assetCount = quantity != null ? quantity : 1;
+            
+            // 获取该类型资产的编码前缀和当前数量
+            String typeCodePrefix = assetTypeService.getAssetCodePrefix(assetTypeId);
+            String dateStr = currentDate.format(DateTimeFormatter.ofPattern("yyyyMMdd"));
+            String prefix = typeCodePrefix + dateStr + "-";
+            
+            // 获取当天该类型资产的当前数量(只查询一次)
+            Long currentCount = assetTypeCountMap.computeIfAbsent(assetTypeId, k -> 
+                this.lambdaQuery()
+                    .like(AssetMain::getAssetCode, prefix)
+                    .ge(AssetMain::getCreateTime, currentDate.atStartOfDay())
+                    .lt(AssetMain::getCreateTime, currentDate.plusDays(1).atStartOfDay())
+                    .count()
+            );
             
             for (int i = 0; i < assetCount; i++) {
                 // 创建资产主表记录
                 AssetMain assetMain = new AssetMain();
-                BeanUtil.copyProperties(storageAsset, assetMain);
+                BeanUtil.copyProperties(storageAsset, assetMain, "id");
                 
-                // 为每个资产生成编码
-                String assetCode = generateAssetCode(storageAsset.getAssetTypeId(), LocalDate.now());
+                // 生成资产编码,使用递增的序号
+                int sequence = (currentCount != null ? currentCount.intValue() : 0) + i + 1;
+                String sequenceStr = String.format("%04d", sequence);
+                String assetCode = prefix + sequenceStr;
                 assetMain.setAssetCode(assetCode);
                 
                 // 单个资产的数量设为1
-                assetMain.setQuantity(BigDecimal.ONE);
+                assetMain.setQuantity(1);
                 // 单个资产的总价值等于单价
                 assetMain.setTotalValue(storageAsset.getUnitPrice());
                 
@@ -125,63 +167,149 @@
                 assetMain.setCreateBy(SecurityUtils.getUsername());
                 assetMain.setDisabled(false);
                 
-                // 保存资产主表
-                assetMainMapper.insert(assetMain);
+                assetMainList.add(assetMain);
                 
-                // 2. 根据资产类型保存扩展信息
-                if (assetMainType != null) {
-                    if (assetMainType == 1) {
-                        // 房产资产 - 保存房产扩展信息
-                        savePropertyExtension(storageAsset.getId(), assetMain.getId());
-                    } else if (assetMainType == 2) {
-                        // 车辆资产 - 保存车辆扩展信息
-                        saveVehicleExtension(storageAsset.getId(), assetMain.getId());
-                    }
-                    // assetMainType == 0 为通用资产,无需保存扩展信息
-                }
+                // 记录storageAssetId和资产类型的映射关系,用于后续创建扩展信息
+                storageAssetIdToAssetTypeMap.put(storageAsset.getId(), assetMainType);
+            }
+            
+            // 更新计数器,确保下一个storageAsset的编码序号正确
+            if (currentCount != null) {
+                assetTypeCountMap.put(assetTypeId, currentCount + assetCount);
             }
         }
-    }
-    
-    /**
-     * 保存房产资产扩展信息
-     */
-    private void savePropertyExtension(Integer storageAssetId, Integer assetMainId) {
-        // 查询房产扩展信息
-        OaApprovalApplicationStorageAssetPropertyExt propertyExt = oaApprovalApplicationStorageAssetPropertyExtMapper
-                .selectOne(new QueryWrapper<OaApprovalApplicationStorageAssetPropertyExt>()
-                        .eq("storage_asset_id", storageAssetId));
         
-        if (propertyExt != null) {
-            AssetPropertyExt assetPropertyExt = new AssetPropertyExt();
-            BeanUtil.copyProperties(propertyExt, assetPropertyExt);
-            assetPropertyExt.setAssetMainId(assetMainId);
-            assetPropertyExt.setCreateTime(LocalDateTime.now());
-            assetPropertyExt.setCreateBy(SecurityUtils.getUsername());
-            assetPropertyExt.setDisabled(false);
+        // 批量保存资产主表
+        if (!assetMainList.isEmpty()) {
+            this.saveBatch(assetMainList);
             
-            assetPropertyExtMapper.insert(assetPropertyExt);
+            // 建立storageAssetId到assetMainId的映射关系
+            int assetMainIndex = 0;
+            for (OaApprovalApplicationStorageAsset storageAsset : storageAssets) {
+                int quantity = storageAsset.getQuantity() != null ? storageAsset.getQuantity() : 1;
+                for (int i = 0; i < quantity; i++) {
+                    if (assetMainIndex < assetMainList.size()) {
+                        storageAssetIdToAssetMainIdMap.put(storageAsset.getId(), assetMainList.get(assetMainIndex).getId());
+                        assetMainIndex++;
+                    }
+                }
+            }
+            
+            // 准备并保存扩展信息
+            saveExtensionInfoBatch(storageAssetIdToAssetMainIdMap, storageAssetIdToAssetTypeMap);
         }
     }
     
     /**
-     * 保存车辆资产扩展信息
+     * 批量保存扩展信息
      */
-    private void saveVehicleExtension(Integer storageAssetId, Integer assetMainId) {
-        // 查询车辆扩展信息
-        OaApprovalApplicationStorageAssetVehicleExt vehicleExt = oaApprovalApplicationStorageAssetVehicleExtMapper
-                .selectOne(new QueryWrapper<OaApprovalApplicationStorageAssetVehicleExt>()
-                        .eq("storage_asset_id", storageAssetId));
+    private void saveExtensionInfoBatch(Map<Integer, Integer> storageAssetIdToAssetMainIdMap, Map<Integer, Integer> storageAssetIdToAssetTypeMap) {
+        List<AssetPropertyExt> propertyExtList = new ArrayList<>();
+        List<AssetVehicleExt> vehicleExtList = new ArrayList<>();
         
-        if (vehicleExt != null) {
-            AssetVehicleExt assetVehicleExt = new AssetVehicleExt();
-            BeanUtil.copyProperties(vehicleExt, assetVehicleExt);
-            assetVehicleExt.setAssetMainId(assetMainId);
-            assetVehicleExt.setCreateTime(LocalDateTime.now());
-            assetVehicleExt.setCreateBy(SecurityUtils.getUsername());
-            assetVehicleExt.setDisabled(false);
+        // 遍历所有需要扩展信息的storageAsset
+        for (Map.Entry<Integer, Integer> entry : storageAssetIdToAssetMainIdMap.entrySet()) {
+            Integer storageAssetId = entry.getKey();
+            Integer assetMainId = entry.getValue();
+            Integer assetMainType = storageAssetIdToAssetTypeMap.get(storageAssetId);
             
-            assetVehicleExtMapper.insert(assetVehicleExt);
+            if (assetMainType == null) {
+                continue;
+            }
+            
+            if (assetMainType == 1) {
+                // 房产资产 - 查询并准备房产扩展信息
+                OaApprovalApplicationStorageAssetPropertyExt propertyExt = oaApprovalApplicationStorageAssetPropertyExtMapper
+                        .selectOne(new LambdaQueryWrapper<OaApprovalApplicationStorageAssetPropertyExt>()
+                                .eq(OaApprovalApplicationStorageAssetPropertyExt::getStorageAssetId, storageAssetId));
+                
+                if (propertyExt != null) {
+                    AssetPropertyExt assetPropertyExt = new AssetPropertyExt();
+                    BeanUtil.copyProperties(propertyExt, assetPropertyExt,"id");
+                    assetPropertyExt.setAssetMainId(assetMainId);
+                    assetPropertyExt.setCreateTime(LocalDateTime.now());
+                    assetPropertyExt.setCreateBy(SecurityUtils.getUsername());
+                    assetPropertyExt.setDisabled(false);
+                    
+                    propertyExtList.add(assetPropertyExt);
+                }
+            } else if (assetMainType == 2) {
+                // 车辆资产 - 查询并准备车辆扩展信息
+                OaApprovalApplicationStorageAssetVehicleExt vehicleExt = oaApprovalApplicationStorageAssetVehicleExtMapper
+                        .selectOne(new QueryWrapper<OaApprovalApplicationStorageAssetVehicleExt>()
+                                .eq("storage_asset_id", storageAssetId));
+                
+                if (vehicleExt != null) {
+                    AssetVehicleExt assetVehicleExt = new AssetVehicleExt();
+                    BeanUtil.copyProperties(vehicleExt, assetVehicleExt,"id");
+                    assetVehicleExt.setAssetMainId(assetMainId);
+                    assetVehicleExt.setCreateTime(LocalDateTime.now());
+                    assetVehicleExt.setCreateBy(SecurityUtils.getUsername());
+                    assetVehicleExt.setDisabled(false);
+                    
+                    vehicleExtList.add(assetVehicleExt);
+                }
+            }
+            // assetMainType == 0 为通用资产,无需保存扩展信息
         }
+        
+        // 批量保存扩展信息
+        if (!propertyExtList.isEmpty()) {
+            assetPropertyExtService.saveBatch(propertyExtList);
+        }
+        if (!vehicleExtList.isEmpty()) {
+            assetVehicleExtService.saveBatch(vehicleExtList);
+        }
+    }
+    
+    
+    @Override
+    public List<AssetMainVO> getListByIds(List<Integer> assetMainIds) {
+        if (!CollectionUtils.isEmpty(assetMainIds)) {
+            List<AssetMain> assetMains = this.lambdaQuery()
+                    .in(AssetMain::getId, assetMainIds)
+                    .list();
+            //查询资产类型名称
+            List<Integer> assetTypeIds = assetMains.stream().map(AssetMain::getAssetTypeId).distinct().collect(Collectors.toList());
+            Map<Integer, String> typeNameMap = assetTypeIds.isEmpty() ? Collections.emptyMap() : assetTypeService.listByIds(assetTypeIds).stream()
+                    .collect(Collectors.toMap(AssetType::getId, AssetType::getTypeName));
+            //查询批量查询部门名称
+            //使用部门
+            Set<Integer> useDeptIds = assetMains.stream().filter(item -> item.getAddressType().equals(0))
+                    .map(AssetMain::getUseDeptId).collect(Collectors.toSet());
+            //归属部门
+            Set<Integer> deptIds = assetMains.stream().map(AssetMain::getOwnershipDeptId).collect(Collectors.toSet());
+            deptIds.addAll(useDeptIds);
+            Map<Integer, String> deptNameMap = deptIds.isEmpty() ? Collections.emptyMap() : tDeptService.listByIds(deptIds).stream()
+                    .collect(Collectors.toMap(TDept::getId, TDept::getDeptName));
+            //查询资产入库时间
+            List<Integer> applicationIds = assetMains.stream().map(AssetMain::getApprovalApplicationId).distinct().collect(Collectors.toList());
+            Map<Integer, LocalDate> storageTimeMap = oaApprovalApplicationStorageService.lambdaQuery().in(OaApprovalApplicationStorage::getApprovalApplicationId, applicationIds).list().stream().collect(Collectors.toMap(OaApprovalApplicationStorage::getApprovalApplicationId, OaApprovalApplicationStorage::getStorageTime));
+            //查询使用仓库
+            Set<Integer> warehouseIds = assetMains.stream().filter(item -> item.getAddressType().equals(1)).map(AssetMain::getWarehouseId).collect(Collectors.toSet());
+            Map<Integer, String> warehouseNameMap = warehouseIds.isEmpty() ? Collections.emptyMap() : assetWarehouseService.listByIds(warehouseIds).stream()
+                    .collect(Collectors.toMap(AssetWarehouse::getId, AssetWarehouse::getWarehouseName));
+            for (AssetMain assetMain : assetMains) {
+                switch (assetMain.getAddressType()) {
+                    case 0:
+                        assetMain.setUseDeptOrLocation(deptNameMap.get(assetMain.getUseDeptId()));
+                        break;
+                    case 1:
+                        assetMain.setUseDeptOrLocation(warehouseNameMap.get(assetMain.getWarehouseId()));
+                        break;
+                    case 2:
+                        assetMain.setUseDeptOrLocation(assetMain.getAddress());
+                        break;
+                }
+            }
+            List<AssetMainVO> voList = BeanUtil.copyToList(assetMains, AssetMainVO.class);
+            for (AssetMainVO assetMainVO : voList) {
+                assetMainVO.setAssetTypeName(typeNameMap.get(assetMainVO.getAssetTypeId()));
+                assetMainVO.setOwnershipDeptName(deptNameMap.get(assetMainVO.getOwnershipDeptId()));
+                assetMainVO.setStorageTime(storageTimeMap.get(assetMainVO.getApprovalApplicationId()));
+            }
+            return voList;
+        }
+        return Collections.emptyList();
     }
 }

--
Gitblit v1.7.1