xuhy
2024-09-24 ae4c93dd2e2aef79332360fb5dec13c5b2c961f3
ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TFoundationPersonServiceImpl.java
@@ -2,6 +2,7 @@
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.system.domain.TFoundationConfig;
import com.ruoyi.system.domain.TFoundationPerson;
import com.ruoyi.system.dto.TFoundationPersonDTO;
@@ -33,10 +34,14 @@
    @Override
    public void add(TFoundationPersonDTO dto) {
        List<TFoundationConfig> foundationConfigs = dto.getFoundationConfigs();
        long count = foundationConfigs.stream().map(TFoundationConfig::getTypeId).distinct().count();
        if(foundationConfigs.size()!=count){
            throw new ServiceException("菜品重复设置");
        }
        // 添加人数
        this.save(dto);
        // 添加菜品
        List<TFoundationConfig> foundationConfigs = dto.getFoundationConfigs();
        foundationConfigs.forEach(item -> {
            item.setPersonId(dto.getId());
        });
@@ -45,13 +50,17 @@
    @Override
    public void edit(TFoundationPersonDTO dto) {
        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()));
        // 添加菜品
        List<TFoundationConfig> foundationConfigs = dto.getFoundationConfigs();
        foundationConfigs.forEach(item -> {
            item.setPersonId(dto.getId());
        });