Lius
昨天 f73f8cb6753f8d8ab2c689b4640cd2688e6f6f62
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package org.jeecg.modules.mdc.service.impl;
 
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.modules.mdc.entity.MdcEquipmentDaySummary;
import org.jeecg.modules.mdc.mapper.MdcEquipmentDaySummaryMapper;
import org.jeecg.modules.mdc.service.IMdcEquipmentDaySummaryService;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
 
/**
 * @Author: Lius
 * @CreateTime: 2025-02-26
 * @Description: 设备日汇总impl
 */
@Service
public class MdcEquipmentDaySummaryServiceImpl extends ServiceImpl<MdcEquipmentDaySummaryMapper, MdcEquipmentDaySummary> implements IMdcEquipmentDaySummaryService {
    @Override
    public MdcEquipmentDaySummary statisticsQty(String productionId) {
        return this.baseMapper.statisticsQty(productionId);
    }
 
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean saveOrUpdateBatchByMesId(List<MdcEquipmentDaySummary> entityList) {
        if (CollectionUtils.isEmpty(entityList)) {
            return true;
        }
 
        // 按mesId分组处理
        Map<String, MdcEquipmentDaySummary> mesIdMap = entityList.stream()
                .collect(Collectors.toMap(MdcEquipmentDaySummary::getMesId,
                        mdcEquipmentDaySummary -> mdcEquipmentDaySummary,       // 值:用户对象本身
                        (existing, replacement) -> existing));
 
        // 批量查询已存在的mesId
        List<MdcEquipmentDaySummary> existList = this.lambdaQuery()
                .in(MdcEquipmentDaySummary::getMesId, mesIdMap.keySet())
                .list();
 
 
        // 分离需要新增和更新的数据
        List<MdcEquipmentDaySummary> toInsert = new ArrayList<>();
        List<MdcEquipmentDaySummary> toUpdate = new ArrayList<>();
 
        existList.forEach(exist -> {
            MdcEquipmentDaySummary 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;
    }
}