feat 功能修正

xiaoyi

5 天以前 c9d4ba37c1f681b12e24014013fa9a28734bad42

 yudao-module-bi/src/main/java/cn/iocoder/yudao/module/bi/service/decision/BiForecastServiceImpl.java

@@ -1,14 +1,22 @@
package cn.iocoder.yudao.module.bi.service.decision;

import cn.hutool.core.date.DateUtil;
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.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_NOT_EXISTS;

/**
 * 负荷预测 Service 实现
@@ -21,8 +29,15 @@
@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) {
@@ -48,4 +63,32 @@
        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));
    }

}