From dac03b5ff8efb1b733d37dce7553ed0aff32c780 Mon Sep 17 00:00:00 2001
From: liding <756868258@qq.com>
Date: 星期五, 09 五月 2025 17:47:02 +0800
Subject: [PATCH] 1.销售台账合同金额(所有产品含税总价)

---
 src/main/java/com/ruoyi/sales/service/impl/SalesLedgerServiceImpl.java |  110 +++++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 82 insertions(+), 28 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..52ec03e 100644
--- a/src/main/java/com/ruoyi/sales/service/impl/SalesLedgerServiceImpl.java
+++ b/src/main/java/com/ruoyi/sales/service/impl/SalesLedgerServiceImpl.java
@@ -1,11 +1,13 @@
 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.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.sales.dto.SalesLedgerDto;
 import com.ruoyi.sales.mapper.SalesLedgerMapper;
 import com.ruoyi.sales.mapper.SalesLedgerProductMapper;
@@ -19,10 +21,13 @@
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.lang.reflect.Field;
+import java.math.BigDecimal;
 import java.time.LocalDate;
 import java.time.format.DateTimeFormatter;
 import java.util.*;
 import java.util.concurrent.TimeUnit;
+import java.util.function.Function;
 import java.util.stream.Collectors;
 
 /**
@@ -48,33 +53,34 @@
     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);
-
-            List<SalesLedgerProduct> ledgerProducts = productMap.getOrDefault(ledger.getId(), Collections.emptyList());
-            if (!ledgerProducts.isEmpty()) {
-                dto.setHasChildren(true);
-                dto.setProductData(ledgerProducts);
-            }
-            return dto;
-        }).collect(Collectors.toList());
-    }
-
-    @Override
-    public SalesLedger selectSalesLedgerById(Long id) {
-        return salesLedgerMapper.selectById(id);
+        // 3. 杞崲 DTO
+        SalesLedgerDto resultDto = new SalesLedgerDto();
+        BeanUtils.copyProperties(salesLedger, resultDto);
+        if (!products.isEmpty()) {
+            resultDto.setHasChildren(true);
+            resultDto.setProductData(products);
+        }
+        return resultDto;
     }
 
     @Override
@@ -120,10 +126,19 @@
         }
 
         // 4. 澶勭悊瀛愯〃鏁版嵁
-        if (salesLedgerDto.getProductData() != null && !salesLedgerDto.getProductData().isEmpty()) {
-            handleSalesLedgerProducts(salesLedger.getId(), salesLedgerDto.getProductData());
-        }
+        List<SalesLedgerProduct> productList = salesLedgerDto.getProductData();
+        if (productList != null && !productList.isEmpty()) {
+            handleSalesLedgerProducts(salesLedger.getId(), productList);
 
+            // 鉁� 璋冪敤閫氱敤鏂规硶鏇存柊涓昏〃閲戦
+            updateMainContractAmount(
+                    salesLedger.getId(),
+                    productList,
+                    SalesLedgerProduct::getTaxInclusiveTotalPrice,
+                    salesLedgerMapper,
+                    SalesLedger.class
+            );
+        }
         return 1; // 鎿嶄綔鎴愬姛杩斿洖1
     }
 
@@ -138,11 +153,15 @@
 
         // 鎵ц鏇存柊鎿嶄綔
         if (!updateList.isEmpty()) {
-            salesLedgerProductMapper.updateBatchSomeColumn(updateList);
+            for (SalesLedgerProduct product : updateList) {
+                salesLedgerProductMapper.updateById(product);
+            }
         }
         // 鎵ц鎻掑叆鎿嶄綔
         if (!insertList.isEmpty()) {
-            salesLedgerProductMapper.insertBatchSomeColumn(insertList);
+            for (SalesLedgerProduct salesLedgerProduct : insertList) {
+                salesLedgerProductMapper.insert(salesLedgerProduct);
+            }
         }
     }
 
@@ -213,4 +232,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