From 76561071d3441cd8a551772d0f43d93c9f9a1e6a Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期二, 29 十月 2024 18:08:25 +0800
Subject: [PATCH] 修改

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TFoundationPersonServiceImpl.java |   88 +++++++++++++++++++++++---------------------
 1 files changed, 46 insertions(+), 42 deletions(-)

diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TFoundationPersonServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TFoundationPersonServiceImpl.java
index c5fbbd1..cc1982f 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TFoundationPersonServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TFoundationPersonServiceImpl.java
@@ -15,9 +15,11 @@
 import com.ruoyi.system.vo.TFoundationPersonVO;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
 
 import java.util.Comparator;
+import java.util.HashSet;
 import java.util.List;
 import java.util.stream.Collectors;
 
@@ -36,54 +38,45 @@
     private TFoundationConfigService foundationConfigService;
 
     @Override
-    public void add(TFoundationPersonDTO dto) {
-
-        long count1 = this.count(Wrappers.lambdaQuery(TFoundationPerson.class)
-                .eq(TFoundationPerson::getShopId, dto.getShopId())
-                .eq(TFoundationPerson::getMealCount, dto.getMealCount()));
-        if(count1>0){
-            throw new ServiceException("该店铺已存在该用餐人数设置");
+    @Transactional(rollbackFor = Exception.class)
+    public void add(List<TFoundationPersonDTO> dto) {
+        for (TFoundationPersonDTO tFoundationPersonDTO : dto) {
+            List<TFoundationConfig> foundationConfigs = tFoundationPersonDTO.getFoundationConfigs();
+            long count = foundationConfigs.stream().map(TFoundationConfig::getTypeId).distinct().count();
+            if(foundationConfigs.size()!=count){
+                throw new ServiceException("菜品重复设置");
+            }
+            this.save(tFoundationPersonDTO);
+            // 添加菜品
+            foundationConfigs.forEach(item -> {
+                item.setPersonId(tFoundationPersonDTO.getId());
+            });
+            foundationConfigService.saveBatch(foundationConfigs);
         }
-
-        List<TFoundationConfig> foundationConfigs = dto.getFoundationConfigs();
-        long count = foundationConfigs.stream().map(TFoundationConfig::getTypeId).distinct().count();
-        if(foundationConfigs.size()!=count){
-            throw new ServiceException("菜品重复设置");
-        }
-        // 添加人数
-        this.save(dto);
-        // 添加菜品
-        foundationConfigs.forEach(item -> {
-            item.setPersonId(dto.getId());
-        });
-        foundationConfigService.saveBatch(foundationConfigs);
     }
 
     @Override
-    public void edit(TFoundationPersonDTO dto) {
-        long count1 = this.count(Wrappers.lambdaQuery(TFoundationPerson.class)
-                .eq(TFoundationPerson::getShopId, dto.getShopId())
-                .eq(TFoundationPerson::getMealCount, dto.getMealCount())
-                .ne(TFoundationPerson::getId,dto.getId()));
-        if(count1>0){
-            throw new ServiceException("该店铺已存在该用餐人数设置");
+    @Transactional(rollbackFor = Exception.class)
+    public void edit(List<TFoundationPersonDTO> dto) {
+
+        for (TFoundationPersonDTO tFoundationPersonDTO : dto) {
+            List<TFoundationConfig> foundationConfigs = tFoundationPersonDTO.getFoundationConfigs();
+            long count = foundationConfigs.stream().map(TFoundationConfig::getTypeId).distinct().count();
+            if(foundationConfigs.size()!=count){
+                throw new ServiceException("菜品重复设置");
+            }
+            // 添加人数
+            this.updateById(tFoundationPersonDTO);
+            // 删除菜品
+            foundationConfigService.remove(Wrappers.lambdaQuery(TFoundationConfig.class)
+                    .eq(TFoundationConfig::getPersonId,tFoundationPersonDTO.getId()));
+            // 添加菜品
+            foundationConfigs.forEach(item -> {
+                item.setPersonId(tFoundationPersonDTO.getId());
+            });
+            foundationConfigService.saveBatch(foundationConfigs);
         }
 
-        List<TFoundationConfig> foundationConfigs = dto.getFoundationConfigs();
-        long count = foundationConfigs.stream().map(TFoundationConfig::getTypeId).distinct().count();
-        if(foundationConfigs.size()!=count){
-            throw new ServiceException("菜品重复设置");
-        }
-        // 添加人数
-        this.updateById(dto);
-        // 删除菜品
-        foundationConfigService.remove(Wrappers.lambdaQuery(TFoundationConfig.class)
-                .eq(TFoundationConfig::getPersonId,dto.getId()));
-        // 添加菜品
-        foundationConfigs.forEach(item -> {
-            item.setPersonId(dto.getId());
-        });
-        foundationConfigService.saveBatch(foundationConfigs);
     }
 
     @Override
@@ -106,4 +99,15 @@
         pageInfo.setRecords(list);
         return pageInfo;
     }
+
+    @Override
+    public boolean hasDuplicateId(List<TFoundationPersonDTO> list) {
+        HashSet<Integer> countSet = new HashSet<>();
+        for (TFoundationPersonDTO dto : list) {
+            if (!countSet.add(dto.getMealCount())) {
+                return true; // 发现重复的人数
+            }
+        }
+        return false; // 没有发现重复的人数
+    }
 }

--
Gitblit v1.7.1