xiaoyi
2 天以前 1dfdc3b0611211feba0570ec6d8cfad078c37135
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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
package cn.iocoder.yudao.module.bi.service.decision;
 
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateUtil;
import cn.iocoder.yudao.module.bi.controller.admin.decision.vo.DecisionForecastHistorySaveReqVO;
import cn.iocoder.yudao.module.bi.controller.admin.decision.vo.DecisionForecastWorkOrderCreateReqVO;
import cn.iocoder.yudao.module.bi.dal.dataobject.decision.BiForecastDataDO;
import cn.iocoder.yudao.module.bi.dal.mysql.decision.BiForecastDataMapper;
import cn.iocoder.yudao.module.mes.api.workorder.MesProWorkOrderApi;
import cn.iocoder.yudao.module.mes.api.workorder.dto.MesProWorkOrderCreateReqDTO;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.time.LocalDateTime;
import java.util.List;
 
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.bi.enums.ErrorCodeConstants.BI_FORECAST_NO_HISTORY;
import static cn.iocoder.yudao.module.bi.enums.ErrorCodeConstants.BI_FORECAST_NOT_EXISTS;
 
/**
 * 负荷预测 Service 实现
 *
 * 本期使用简单移动平均占位,预留 ML 模型接口
 *
 * @author 超级管理员
 */
@Slf4j
@Service
public class BiForecastServiceImpl implements BiForecastService {
 
    /**
     * 工单来源类型:负荷预测(与字典 mes_pro_work_order_source_type 的 value=4 一致)
     */
    private static final Integer WORK_ORDER_SOURCE_TYPE_FORECAST = 4;
 
    @Resource
    private BiForecastDataMapper forecastDataMapper;
    @Resource
    private MesProWorkOrderApi mesProWorkOrderApi;
 
    @Override
    public List<BiForecastDataDO> getForecastList(String forecastCode, LocalDateTime beginTime, LocalDateTime endTime) {
        return forecastDataMapper.selectListByCodeAndTimeRange(forecastCode, beginTime, endTime);
    }
 
    /**
     * 移动平均窗口大小
     */
    private static final int SMA_WINDOW = 12;
 
    @Override
    public Long createForecastHistory(DecisionForecastHistorySaveReqVO reqVO) {
        BiForecastDataDO data = BiForecastDataDO.builder()
                .forecastCode(reqVO.getForecastCode())
                .forecastName(reqVO.getForecastName())
                .pointTime(reqVO.getPointTime())
                .forecastValue(reqVO.getForecastValue())
                .actualValue(reqVO.getActualValue())
                .dimension(reqVO.getDimension())
                .dimensionValue(reqVO.getDimensionValue())
                .remark(reqVO.getRemark())
                .modelVersion("manual-history")
                .build();
        forecastDataMapper.insert(data);
        return data.getId();
    }
 
    @Override
    public void generateForecast(String forecastCode, LocalDateTime pointTime) {
        // 基于同预测编码的历史预测数据做简单移动平均;无历史数据时不伪造零值
        List<BiForecastDataDO> history = forecastDataMapper.selectRecentByCode(forecastCode, SMA_WINDOW);
        if (CollUtil.isEmpty(history)) {
            throw exception(BI_FORECAST_NO_HISTORY);
        }
        BigDecimal sum = history.stream()
                .map(BiForecastDataDO::getForecastValue)
                .filter(value -> value != null)
                .reduce(BigDecimal.ZERO, BigDecimal::add);
        BigDecimal avg = sum.divide(BigDecimal.valueOf(history.size()), 4, RoundingMode.HALF_UP);
        BigDecimal bound = avg.multiply(new BigDecimal("0.10"));
 
        BiForecastDataDO forecast = BiForecastDataDO.builder()
                .forecastCode(forecastCode)
                .forecastName(history.get(0).getForecastName())
                .pointTime(pointTime)
                .forecastValue(avg)
                .lowerBound(avg.subtract(bound))
                .upperBound(avg.add(bound))
                .confidenceLevel(new BigDecimal("95.00"))
                .modelVersion("v1.0-sma")
                .build();
        forecastDataMapper.insert(forecast);
    }
 
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Long createWorkOrderFromForecast(DecisionForecastWorkOrderCreateReqVO reqVO) {
        // 1. 校验预测数据存在
        BiForecastDataDO forecast = forecastDataMapper.selectById(reqVO.getForecastId());
        if (forecast == null) {
            throw exception(BI_FORECAST_NOT_EXISTS);
        }
        // 2. 组装工单 DTO 并调用 MES API 创建(无产品工单,跳 BOM)
        MesProWorkOrderCreateReqDTO dto = new MesProWorkOrderCreateReqDTO();
        dto.setCode(generateWorkOrderCode());
        dto.setName(reqVO.getWorkOrderName() != null
                ? reqVO.getWorkOrderName()
                : ("负荷预测-" + forecast.getForecastName()));
        dto.setType(reqVO.getWorkOrderType());
        dto.setOrderSourceType(WORK_ORDER_SOURCE_TYPE_FORECAST);
        dto.setOrderSourceCode(forecast.getForecastCode());
        dto.setRequestDate(reqVO.getRequestDate());
        dto.setQuantity(reqVO.getQuantity());
        dto.setRemark(reqVO.getRemark());
        return mesProWorkOrderApi.createWorkOrder(dto);
    }
 
    private String generateWorkOrderCode() {
        return "GZ" + DateUtil.format(LocalDateTime.now(), "yyyyMMddHHmmss")
                + String.format("%03d", (int) (Math.random() * 1000));
    }
 
}