From 53bdb26a0ae994418e92d93aab23d6f5e7225f37 Mon Sep 17 00:00:00 2001
From: liding <756868258@qq.com>
Date: 星期一, 19 五月 2025 16:20:56 +0800
Subject: [PATCH] 产品

---
 src/main/java/com/ruoyi/sales/service/impl/SalesLedgerServiceImpl.java |  338 +++++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 285 insertions(+), 53 deletions(-)

diff --git a/src/main/java/com/ruoyi/sales/service/impl/SalesLedgerServiceImpl.java b/src/main/java/com/ruoyi/sales/service/impl/SalesLedgerServiceImpl.java
index a1f884d..557b898 100644
--- a/src/main/java/com/ruoyi/sales/service/impl/SalesLedgerServiceImpl.java
+++ b/src/main/java/com/ruoyi/sales/service/impl/SalesLedgerServiceImpl.java
@@ -1,28 +1,48 @@
 package com.ruoyi.sales.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.basic.mapper.CustomerMapper;
 import com.ruoyi.basic.pojo.Customer;
 import com.ruoyi.common.exception.base.BaseException;
+import com.ruoyi.common.utils.StringUtils;
+import com.ruoyi.other.mapper.TempFileMapper;
+import com.ruoyi.other.pojo.TempFile;
 import com.ruoyi.sales.dto.SalesLedgerDto;
+import com.ruoyi.sales.mapper.CommonFileMapper;
 import com.ruoyi.sales.mapper.SalesLedgerMapper;
 import com.ruoyi.sales.mapper.SalesLedgerProductMapper;
+import com.ruoyi.sales.pojo.CommonFile;
 import com.ruoyi.sales.pojo.SalesLedger;
 import com.ruoyi.sales.pojo.SalesLedgerProduct;
 import com.ruoyi.sales.service.ISalesLedgerService;
-import lombok.AllArgsConstructor;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.io.FilenameUtils;
 import org.springframework.beans.BeanUtils;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.data.redis.core.script.DefaultRedisScript;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.io.IOException;
+import java.lang.reflect.Field;
+import java.math.BigDecimal;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardCopyOption;
 import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.time.YearMonth;
 import java.time.format.DateTimeFormatter;
 import java.util.*;
 import java.util.concurrent.TimeUnit;
+import java.util.function.Function;
 import java.util.stream.Collectors;
 
 /**
@@ -32,14 +52,22 @@
  * @date 2025-05-08
  */
 @Service
-@AllArgsConstructor
+@RequiredArgsConstructor
+@Slf4j
 public class SalesLedgerServiceImpl extends ServiceImpl<SalesLedgerMapper, SalesLedger> implements ISalesLedgerService {
 
-    private SalesLedgerMapper salesLedgerMapper;
+    private final SalesLedgerMapper salesLedgerMapper;
 
-    private CustomerMapper customerMapper;
+    private final CustomerMapper customerMapper;
 
-    private SalesLedgerProductMapper salesLedgerProductMapper;
+    private final SalesLedgerProductMapper salesLedgerProductMapper;
+
+    private final CommonFileMapper commonFileMapper;
+
+    private final TempFileMapper tempFileMapper;
+
+    @Value("${file.upload-dir}")
+    private String uploadDir;
 
     private static final String LOCK_PREFIX = "contract_no_lock:";
     private static final long LOCK_WAIT_TIMEOUT = 10; // 閿佺瓑寰呰秴鏃舵椂闂达紙绉掞級
@@ -48,33 +76,105 @@
     private final RedisTemplate<String, String> redisTemplate;
 
     @Override
-    public List<SalesLedger> selectSalesLedgerList(SalesLedger salesLedger) {
-        return salesLedgerMapper.selectList(new LambdaQueryWrapper<>());
+    public List<SalesLedger> selectSalesLedgerList(SalesLedgerDto salesLedgerDto) {
+        LambdaQueryWrapper<SalesLedger> queryWrapper = new LambdaQueryWrapper<>();
+        if (StringUtils.isNotBlank(salesLedgerDto.getCustomerName())) {
+            queryWrapper.eq(SalesLedger::getCustomerName, salesLedgerDto.getCustomerName());
+        }
+        return salesLedgerMapper.selectList(queryWrapper);
     }
 
-    public List<SalesLedgerDto> getSalesLedgerWithProducts() {
-        List<SalesLedger> ledgers = salesLedgerMapper.selectList(new LambdaQueryWrapper<>());
-        List<SalesLedgerProduct> products = salesLedgerProductMapper.selectList(new LambdaQueryWrapper<>());
+    public SalesLedgerDto getSalesLedgerWithProducts(SalesLedgerDto salesLedgerDto) {
+        // 1. 鏌ヨ涓昏〃
+        SalesLedger salesLedger = salesLedgerMapper.selectById(salesLedgerDto.getId());
+        if (salesLedger == null) {
+            throw new BaseException("鍙拌处涓嶅瓨鍦�");
+        }
 
-        Map<Long, List<SalesLedgerProduct>> productMap = products.stream()
-                .collect(Collectors.groupingBy(SalesLedgerProduct::getSalesLedgerId));
+        // 2. 鏌ヨ瀛愯〃
+        LambdaQueryWrapper<SalesLedgerProduct> productWrapper = new LambdaQueryWrapper<>();
+        productWrapper.eq(SalesLedgerProduct::getSalesLedgerId, salesLedger.getId());
+        List<SalesLedgerProduct> products = salesLedgerProductMapper.selectList(productWrapper);
 
-        return ledgers.stream().map(ledger -> {
-            SalesLedgerDto dto = new SalesLedgerDto();
-            org.springframework.beans.BeanUtils.copyProperties(ledger, dto);
+        // 3.鏌ヨ涓婁紶鏂囦欢
+        LambdaQueryWrapper<CommonFile> salesLedgerFileWrapper = new LambdaQueryWrapper<>();
+        salesLedgerFileWrapper.eq(CommonFile::getCommonId, salesLedger.getId());
+        List<CommonFile> salesLedgerFiles = commonFileMapper.selectList(salesLedgerFileWrapper);
 
-            List<SalesLedgerProduct> ledgerProducts = productMap.getOrDefault(ledger.getId(), Collections.emptyList());
-            if (!ledgerProducts.isEmpty()) {
-                dto.setHasChildren(true);
-                dto.setProductData(ledgerProducts);
-            }
-            return dto;
-        }).collect(Collectors.toList());
+        // 4. 杞崲 DTO
+        SalesLedgerDto resultDto = new SalesLedgerDto();
+        BeanUtils.copyProperties(salesLedger, resultDto);
+        if (!products.isEmpty()) {
+            resultDto.setHasChildren(true);
+            resultDto.setProductData(products);
+            resultDto.setSalesLedgerFiles(salesLedgerFiles);
+        }
+        return resultDto;
     }
 
     @Override
-    public SalesLedger selectSalesLedgerById(Long id) {
-        return salesLedgerMapper.selectById(id);
+    public List<Map<String, Object>> getSalesNo() {
+        LambdaQueryWrapper<SalesLedger> queryWrapper = Wrappers.lambdaQuery();
+        queryWrapper.select(SalesLedger::getId, SalesLedger::getSalesContractNo);
+
+        // 鑾峰彇鍘熷鏌ヨ缁撴灉
+        List<Map<String, Object>> result = salesLedgerMapper.selectMaps(queryWrapper);
+
+        // 灏嗕笅鍒掔嚎鍛藉悕杞崲涓洪┘宄板懡鍚�
+        return result.stream().map(map -> map.entrySet().stream()
+                .collect(Collectors.toMap(
+                        entry -> underlineToCamel(entry.getKey()),
+                        Map.Entry::getValue))
+        ).collect(Collectors.toList());
+    }
+
+    @Override
+    public BigDecimal getContractAmount() {
+        LocalDate now = LocalDate.now();
+        YearMonth currentMonth = YearMonth.from(now);
+
+        // 鍒涘缓LambdaQueryWrapper
+        LambdaQueryWrapper<SalesLedger> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.ge(SalesLedger::getEntryDate, currentMonth.atDay(1).atStartOfDay())  // 澶т簬绛変簬鏈湀绗竴澶�
+                .lt(SalesLedger::getEntryDate, currentMonth.plusMonths(1).atDay(1).atStartOfDay()); // 灏忎簬涓嬫湀绗竴澶�
+
+        // 鎵ц鏌ヨ骞惰绠楁�诲拰
+        List<SalesLedger> salesLedgers = salesLedgerMapper.selectList(queryWrapper);
+
+        BigDecimal totalContractAmount = salesLedgers.stream()
+                .map(SalesLedger::getContractAmount)
+                .filter(Objects::nonNull)
+                .reduce(BigDecimal.ZERO, BigDecimal::add);
+
+        return totalContractAmount;
+    }
+
+    @Override
+    public List getTopFiveList() {
+        
+        return null;
+    }
+
+    /**
+     * 涓嬪垝绾垮懡鍚嶈浆椹煎嘲鍛藉悕
+     */
+    private String underlineToCamel(String param) {
+        if (param == null || "".equals(param.trim())) {
+            return "";
+        }
+        int len = param.length();
+        StringBuilder sb = new StringBuilder(len);
+        for (int i = 0; i < len; i++) {
+            char c = param.charAt(i);
+            if (c == '_') {
+                if (++i < len) {
+                    sb.append(Character.toUpperCase(param.charAt(i)));
+                }
+            } else {
+                sb.append(Character.toLowerCase(c));
+            }
+        }
+        return sb.toString();
     }
 
     @Override
@@ -96,38 +196,129 @@
         return salesLedgerMapper.deleteBatchIds(idList);
     }
 
+    @Override
     @Transactional(rollbackFor = Exception.class)
     public int addOrUpdateSalesLedger(SalesLedgerDto salesLedgerDto) {
-        // 1. 鏍¢獙瀹㈡埛淇℃伅
-        Customer customer = customerMapper.selectById(salesLedgerDto.getCustomerId());
-        if (customer == null) {
-            throw new BaseException("瀹㈡埛涓嶅瓨鍦�");
+        try {
+            // 1. 鏍¢獙瀹㈡埛淇℃伅
+            Customer customer = customerMapper.selectById(salesLedgerDto.getCustomerId());
+            if (customer == null) {
+                throw new BaseException("瀹㈡埛涓嶅瓨鍦�");
+            }
+
+            // 2. DTO杞珽ntity
+            SalesLedger salesLedger = convertToEntity(salesLedgerDto);
+            salesLedger.setCustomerName(customer.getCustomerName());
+            salesLedger.setTenantId(customer.getTenantId());
+
+            // 3. 鏂板鎴栨洿鏂颁富琛�
+            if (salesLedger.getId() == null) {
+                String contractNo = generateSalesContractNo();
+                salesLedger.setSalesContractNo(contractNo);
+                salesLedgerMapper.insert(salesLedger);
+            } else {
+                salesLedgerMapper.updateById(salesLedger);
+            }
+
+            // 4. 澶勭悊瀛愯〃鏁版嵁
+            List<SalesLedgerProduct> productList = salesLedgerDto.getProductData();
+            if (productList != null && !productList.isEmpty()) {
+                handleSalesLedgerProducts(salesLedger.getId(), productList, salesLedgerDto.getType());
+                updateMainContractAmount(
+                        salesLedger.getId(),
+                        productList,
+                        SalesLedgerProduct::getTaxInclusiveTotalPrice,
+                        salesLedgerMapper,
+                        SalesLedger.class
+                );
+            }
+
+            // 5. 杩佺Щ涓存椂鏂囦欢鍒版寮忕洰褰�
+            if (salesLedgerDto.getTempFileIds() != null && !salesLedgerDto.getTempFileIds().isEmpty()) {
+                migrateTempFilesToFormal(salesLedger.getId(), salesLedgerDto.getTempFileIds());
+            }
+
+            return 1;
+        } catch (IOException e) {
+            throw new BaseException("鏂囦欢杩佺Щ澶辫触: " + e.getMessage());
         }
-
-        // 2. DTO杞珽ntity
-        SalesLedger salesLedger = convertToEntity(salesLedgerDto);
-        salesLedger.setCustomerName(customer.getCustomerName());
-        salesLedger.setTenantId(customer.getTenantId());
-
-        // 3. 鏂板鎴栨洿鏂颁富琛�
-        if (salesLedger.getId() == null) {
-            // 鐢熸垚鍚堝悓缂栧彿
-            String contractNo = generateSalesContractNo();
-            salesLedger.setSalesContractNo(contractNo);
-            salesLedgerMapper.insert(salesLedger);
-        } else {
-            salesLedgerMapper.updateById(salesLedger);
-        }
-
-        // 4. 澶勭悊瀛愯〃鏁版嵁
-        if (salesLedgerDto.getProductData() != null && !salesLedgerDto.getProductData().isEmpty()) {
-            handleSalesLedgerProducts(salesLedger.getId(), salesLedgerDto.getProductData());
-        }
-
-        return 1; // 鎿嶄綔鎴愬姛杩斿洖1
     }
 
-    private void handleSalesLedgerProducts(Long salesLedgerId, List<SalesLedgerProduct> products) {
+    // 鏂囦欢杩佺Щ鏂规硶
+
+    /**
+     * 灏嗕复鏃舵枃浠惰縼绉诲埌姝e紡鐩綍
+     *
+     * @param businessId  涓氬姟ID锛堥攢鍞彴璐D锛�
+     * @param tempFileIds 涓存椂鏂囦欢ID鍒楄〃
+     * @throws IOException 鏂囦欢鎿嶄綔寮傚父
+     */
+    private void migrateTempFilesToFormal(Long businessId, List<String> tempFileIds) throws IOException {
+        if (CollectionUtils.isEmpty(tempFileIds)) {
+            return;
+        }
+
+        // 鏋勫缓姝e紡鐩綍璺緞锛堟寜涓氬姟绫诲瀷鍜屾棩鏈熷垎缁勶級
+        String formalDir = uploadDir + LocalDate.now().format(DateTimeFormatter.ISO_LOCAL_DATE);
+
+        Path formalDirPath = Paths.get(formalDir);
+
+        // 纭繚姝e紡鐩綍瀛樺湪锛堥�掑綊鍒涘缓锛�
+        if (!Files.exists(formalDirPath)) {
+            Files.createDirectories(formalDirPath);
+        }
+
+        for (String tempFileId : tempFileIds) {
+            // 鏌ヨ涓存椂鏂囦欢璁板綍
+            TempFile tempFile = tempFileMapper.selectById(tempFileId);
+            if (tempFile == null) {
+                log.warn("涓存椂鏂囦欢涓嶅瓨鍦紝璺宠繃澶勭悊: {}", tempFileId);
+                continue;
+            }
+
+            // 鏋勫缓姝e紡鏂囦欢鍚嶏紙鍖呭惈涓氬姟ID鍜屾椂闂存埑锛岄伩鍏嶅啿绐侊級
+            String originalFilename = tempFile.getOriginalName();
+            String fileExtension = FilenameUtils.getExtension(originalFilename);
+            String formalFilename = businessId + "_" +
+                    System.currentTimeMillis() + "_" +
+                    UUID.randomUUID().toString().substring(0, 8) +
+                    (StringUtils.hasText(fileExtension) ? "." + fileExtension : "");
+
+            Path formalFilePath = formalDirPath.resolve(formalFilename);
+
+            try {
+                // 鎵ц鏂囦欢杩佺Щ锛堜娇鐢ㄥ師瀛愭搷浣滅‘淇濆畨鍏ㄦ�э級
+                Files.move(
+                        Paths.get(tempFile.getTempPath()),
+                        formalFilePath,
+                        StandardCopyOption.REPLACE_EXISTING,
+                        StandardCopyOption.ATOMIC_MOVE
+                );
+                log.info("鏂囦欢杩佺Щ鎴愬姛: {} -> {}", tempFile.getTempPath(), formalFilePath);
+
+                // 鏇存柊鏂囦欢璁板綍锛堝叧鑱斿埌涓氬姟ID锛�
+                CommonFile fileRecord = new CommonFile();
+                fileRecord.setCommonId(businessId);
+                fileRecord.setName(originalFilename);
+                fileRecord.setUrl(formalFilePath.toString());
+                fileRecord.setCreateTime(LocalDateTime.now());
+                fileRecord.setType("1");
+                commonFileMapper.insert(fileRecord);
+
+                // 鍒犻櫎涓存椂鏂囦欢璁板綍
+                tempFileMapper.deleteById(tempFile);
+
+                log.info("鏂囦欢杩佺Щ鎴愬姛: {} -> {}", tempFile.getTempPath(), formalFilePath);
+            } catch (IOException e) {
+                log.error("鏂囦欢杩佺Щ澶辫触: {}", tempFile.getTempPath(), e);
+                // 鍙�夋嫨鍥炴粴浜嬪姟鎴栬褰曞け璐ユ枃浠�
+                throw new IOException("鏂囦欢杩佺Щ寮傚父", e);
+            }
+        }
+    }
+
+
+    private void handleSalesLedgerProducts(Long salesLedgerId, List<SalesLedgerProduct> products, Integer type) {
         // 鎸塈D鍒嗙粍锛屽尯鍒嗘柊澧炲拰鏇存柊鐨勮褰�
         Map<Boolean, List<SalesLedgerProduct>> partitionedProducts = products.stream()
                 .peek(p -> p.setSalesLedgerId(salesLedgerId))
@@ -138,11 +329,17 @@
 
         // 鎵ц鏇存柊鎿嶄綔
         if (!updateList.isEmpty()) {
-            salesLedgerProductMapper.updateBatchSomeColumn(updateList);
+            for (SalesLedgerProduct product : updateList) {
+                product.setType(type);
+                salesLedgerProductMapper.updateById(product);
+            }
         }
         // 鎵ц鎻掑叆鎿嶄綔
         if (!insertList.isEmpty()) {
-            salesLedgerProductMapper.insertBatchSomeColumn(insertList);
+            for (SalesLedgerProduct salesLedgerProduct : insertList) {
+                salesLedgerProduct.setType(type);
+                salesLedgerProductMapper.insert(salesLedgerProduct);
+            }
         }
     }
 
@@ -213,4 +410,39 @@
         }
         return next;
     }
+
+    public <T, S> void updateMainContractAmount(
+            Long mainId,
+            List<T> subList,
+            Function<T, BigDecimal> amountGetter,
+            BaseMapper<S> mainMapper,
+            Class<S> mainEntityClass) {
+
+        if (mainId == null || subList == null || subList.isEmpty()) {
+            return;
+        }
+
+        // 璁$畻瀛愯〃閲戦鎬诲拰
+        BigDecimal totalAmount = subList.stream()
+                .map(amountGetter)
+                .filter(Objects::nonNull)
+                .reduce(BigDecimal.ZERO, BigDecimal::add);
+
+        // 鏋勯�犱富琛ㄦ洿鏂板璞★紙鏀寔浠绘剰涓昏〃绫诲瀷锛�
+        try {
+            S entity = mainEntityClass.getDeclaredConstructor().newInstance();
+            Field idField = mainEntityClass.getDeclaredField("id");
+            idField.setAccessible(true);
+            idField.set(entity, mainId);
+
+            // 璁剧疆 contractAmount 瀛楁锛屾敞鎰忚繖閲屽亣璁惧瓧娈靛悕涓� "contractAmount"
+            Field amountField = mainEntityClass.getDeclaredField("contractAmount");
+            amountField.setAccessible(true);
+            amountField.set(entity, totalAmount);
+
+            mainMapper.updateById(entity);
+        } catch (Exception e) {
+            throw new RuntimeException("鍔ㄦ�佹洿鏂颁富琛ㄩ噾棰濆け璐�", e);
+        }
+    }
 }

--
Gitblit v1.9.3