update

Lius

昨天 f73f8cb6753f8d8ab2c689b4640cd2688e6f6f62

 lxzn-module-mdc/src/main/java/org/jeecg/modules/mdc/service/impl/MdcEquipmentDayScheduleServiceImpl.java

@@ -1,6 +1,7 @@
package org.jeecg.modules.mdc.service.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.modules.mdc.entity.MdcEquipmentDaySchedule;
@@ -9,11 +10,17 @@
import org.jeecg.modules.mdc.vo.MdcEquipmentDayScheduleVo;
import org.jeecg.modules.system.entity.MdcProduction;
import org.jeecg.modules.system.service.IMdcProductionService;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @Author: Lius
@@ -52,4 +59,43 @@
    public MdcEquipmentDaySchedule selectLast(String equipmentId) {
        return this.baseMapper.selectLast(equipmentId);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean saveOrUpdateBatchByMesId(List<MdcEquipmentDaySchedule> entityList) {
        if (CollectionUtils.isEmpty(entityList)) {
            return true;
        }

        // 按mesId分组处理
        Map<String, MdcEquipmentDaySchedule> mesIdMap = entityList.stream()
                .collect(Collectors.toMap(MdcEquipmentDaySchedule::getMesId,
                        mdcEquipmentDaySchedule -> mdcEquipmentDaySchedule,       // 值：用户对象本身
                        (existing, replacement) -> existing));

        // 批量查询已存在的mesId
        List<MdcEquipmentDaySchedule> existList = this.lambdaQuery()
                .in(MdcEquipmentDaySchedule::getMesId, mesIdMap.keySet())
                .list();

        // 分离需要新增和更新的数据
        List<MdcEquipmentDaySchedule> toInsert = new ArrayList<>();
        List<MdcEquipmentDaySchedule> toUpdate = new ArrayList<>();

        existList.forEach(exist -> {
            MdcEquipmentDaySchedule newData = mesIdMap.get(exist.getMesId());
            // 保留原ID
            BeanUtils.copyProperties(newData, exist);
            toUpdate.add(exist);
            mesIdMap.remove(exist.getMesId());
        });

        toInsert.addAll(mesIdMap.values());

        // 执行批量操作
        boolean insertResult = toInsert.isEmpty() || this.saveBatch(toInsert);
        boolean updateResult = toUpdate.isEmpty() || this.updateBatchById(toUpdate);

        return insertResult && updateResult;
    }
}