feat(mes): 新增独立能耗管理功能

云

15 小时以前 d6b144421f86bcaaef5f21ddda193ab6f58b0d57

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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package cn.iocoder.yudao.module.mes.service.energy;
 
import cn.iocoder.yudao.framework.common.exception.ErrorCode;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.mes.controller.admin.energy.vo.*;
import cn.iocoder.yudao.module.mes.dal.dataobject.energy.EnergyRecordDO;
import cn.iocoder.yudao.module.mes.dal.mysql.energy.EnergyRecordMapper;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import java.math.RoundingMode;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
 
@Service
public class EnergyRecordService {
    private static final ErrorCode NOT_EXISTS = new ErrorCode(1_040_950_000, "能耗记录不存在或已删除");
    private static final ErrorCode ID_REQUIRED = new ErrorCode(1_040_950_001, "修改能耗记录必须提供编号");
    @Resource
    private EnergyRecordMapper recordMapper;
 
    public Long create(EnergyRecordSaveReqVO req) {
        EnergyRecordDO record = toRecord(req);
        record.setId(null);
        recordMapper.insert(record);
        return record.getId();
    }
 
    public void update(EnergyRecordSaveReqVO req) {
        if (req.getId() == null) {
            throw exception(ID_REQUIRED);
        }
        get(req.getId());
        if (recordMapper.updateById(toRecord(req)) == 0) {
            throw exception(NOT_EXISTS);
        }
    }
 
    public void delete(Long id) {
        get(id);
        if (recordMapper.deleteById(id) == 0) {
            throw exception(NOT_EXISTS);
        }
    }
 
    public EnergyRecordDO get(Long id) {
        EnergyRecordDO record = recordMapper.selectById(id);
        if (record == null) {
            throw exception(NOT_EXISTS);
        }
        return record;
    }
 
    public PageResult<EnergyRecordDO> page(EnergyRecordPageReqVO req) {
        return recordMapper.selectPage(req);
    }
 
    static EnergyRecordDO toRecord(EnergyRecordSaveReqVO req) {
        EnergyRecordDO record = BeanUtils.toBean(req, EnergyRecordDO.class);
        record.setUsageArea(req.getUsageArea().trim());
        record.setProductCategory(req.getProductCategory().trim());
        record.setUsagePurpose(req.getUsagePurpose().trim());
        record.setRecorder(req.getRecorder().trim());
        // 空备注也可在修改时清除。
        record.setRemark(req.getRemark() == null ? "" : req.getRemark().trim());
        record.setAmount(req.getQuantity().multiply(req.getUnitPrice()).setScale(2, RoundingMode.HALF_UP));
        return record;
    }
 
    public EnergyAnalysisRespVO analysis(EnergyAnalysisReqVO req) {
        return aggregate(recordMapper.selectAnalysis(req));
    }
 
    static EnergyAnalysisRespVO aggregate(List<EnergyRecordMapper.Aggregate> rows) {
        EnergyAnalysisRespVO result = new EnergyAnalysisRespVO();
        Map<String, EnergyAnalysisRespVO.Summary> summary = new LinkedHashMap<>();
        for (String type : List.of("WATER", "ELECTRICITY", "STEAM")) {
            EnergyAnalysisRespVO.Summary item = new EnergyAnalysisRespVO.Summary();
            item.setEnergyType(type);
            summary.put(type, item);
        }
        Map<List<String>, EnergyAnalysisRespVO.Bucket> trend = new LinkedHashMap<>();
        Map<List<String>, EnergyAnalysisRespVO.Bucket> areas = new LinkedHashMap<>();
        for (EnergyRecordMapper.Aggregate row : rows) {
            add(summary.get(row.getEnergyType()), row);
            add(bucket(trend, row.getPeriod(), row.getEnergyType()), row);
            add(bucket(areas, row.getUsageArea(), row.getEnergyType()), row);
            result.setTotalAmount(result.getTotalAmount().add(row.getAmount()));
            result.setRecordCount(result.getRecordCount() + row.getRecordCount());
        }
        result.setSummary(List.copyOf(summary.values()));
        result.setTrend(List.copyOf(trend.values()));
        result.setAreas(List.copyOf(areas.values()));
        return result;
    }
 
    private static EnergyAnalysisRespVO.Bucket bucket(Map<List<String>, EnergyAnalysisRespVO.Bucket> map,
                                                       String label, String type) {
        return map.computeIfAbsent(List.of(label, type), key -> {
            EnergyAnalysisRespVO.Bucket item = new EnergyAnalysisRespVO.Bucket();
            item.setLabel(label);
            item.setEnergyType(type);
            return item;
        });
    }
 
    private static void add(EnergyAnalysisRespVO.Summary item, EnergyRecordMapper.Aggregate row) {
        item.setQuantity(item.getQuantity().add(row.getQuantity()));
        item.setAmount(item.getAmount().add(row.getAmount()));
        item.setRecordCount(item.getRecordCount() + row.getRecordCount());
    }
}