ansteel-biz/src/main/java/com/iailab/module/ansteel/job/task/RunCokingTraceModelBMTask.java

@@ -1,26 +1,24 @@
package com.iailab.module.ansteel.job.task;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.iailab.framework.common.util.date.DateUtils;
import com.iailab.module.ansteel.coking.entity.CokingAnalyIndEntity;
import com.iailab.module.ansteel.coking.entity.CokingTraceDeviationEntity;
import com.iailab.module.ansteel.coking.entity.CokingTraceSuggestEntity;
import com.iailab.module.ansteel.coking.entity.CokingTraceChartEntity;
import com.iailab.module.ansteel.coking.entity.CokingTraceConfEntity;
import com.iailab.module.ansteel.coking.service.*;
import com.iailab.module.ansteel.common.constant.CommonConstant;
import com.iailab.module.ansteel.common.enums.TraceProcessTypeEnum;
import com.iailab.module.model.api.mcs.McsApi;
import com.iailab.module.model.api.mcs.dto.ChartParamDTO;
import com.iailab.module.model.api.mdk.MdkApi;
import org.apache.commons.lang3.StringUtils;
import com.iailab.module.model.api.mdk.dto.MdkScheduleReqDTO;
import com.iailab.module.model.api.mdk.dto.MdkScheduleRespDTO;
import com.iailab.module.model.api.mdk.dto.StScheduleRecordVO;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;

import java.math.BigDecimal;
import java.util.*;
import java.util.stream.Collectors;

@@ -51,10 +49,11 @@
    private CokingTraceIndService cokingTraceIndService;

    @Autowired
    private CokingTraceConfService cokingTraceConfService;
    private CokingTraceChartService cokingTraceChartService;

    @Autowired
    private CokingTraceChartService cokingTraceChartService;
    private CokingTraceConfService cokingTraceConfService;


    @Autowired
    private McsApi mcsApi;
@@ -62,7 +61,7 @@
    @Autowired
    private MdkApi mdkApi;

    private final static String process = "备煤工序";
    private final static String process = TraceProcessTypeEnum.BM.getProcess();

    private final static String reportName = "备煤工序异常溯源";

@@ -70,22 +69,9 @@

    private final static String indType = "备煤工序异常溯源";

    private final static String coalRow = "coalRow";

    private static final String jsonStr = "{\n" + "    " +
            "\"result\": {\n" +
            "\"coalHomeIndexInfo\":\"2025-04-28 甲班 备煤耗电偏高\"," +
            "\"coalPrepElec\":[1600.8,1613.5]," +
            "\"coalPrepElecTotal1\":\"备煤耗电量偏高，经模型计算，原因和调整建议如下：煤量异常，当前值2000, 建议调整煤量至区间[765.0,1020.0]\"," +
            "\"coalPrepElecTime\":[[0.3,0.3],[1700.31,1900.2],[120.3]]," +
            "\"coalPrepElecIndex\":[133527.2,283517.6,83451.5,233461.2,83564.4]," +
            "\"coalRow0\":[[0.0,1000.3],[1.0,120.2],[7.0,1150.32]]," +
            "\"coalRow1\":[[0.0,1000.5],[2.0,200.56],[7.0,120.2]]," +
            "\"coalRow2\":[[0.0,1000.6],[3.0,261.7],[7.0,170.52]]," +
            "\"coalPrepElecHomePage\":[503000.6,84.04]," +
            "\"coalHomeIndex\":[30.0,4.6,523000.6]" +
            "  }" +
            "}";
    private String scheduleCode = "";
    private final String finalResultStrKey = "finalResultStr";
    private final String resultListKey = "resultList";

    @Override
    public void run(String params) {
@@ -95,49 +81,31 @@
            calendar.set(Calendar.MILLISECOND, 0);
            calendar.set(Calendar.SECOND, 0);
            calendar.set(Calendar.MINUTE, 0);
            int hour = calendar.get(Calendar.HOUR_OF_DAY);
            Date startDate = calendar.getTime();

            calendar.set(Calendar.HOUR_OF_DAY, 0);
            Date endDate = calendar.getTime();
            String clock = "";
            String analyClass = "";
            if (hour >= 8 && hour < 16) {
                calendar.set(Calendar.HOUR_OF_DAY, 0);
                startDate = calendar.getTime();
                calendar.add(Calendar.HOUR_OF_DAY, 8);
                endDate = calendar.getTime();
                clock = DateUtils.format(startDate).concat("-").concat("A");
                analyClass = "甲班";
            } else if (hour >= 16 && hour < 24) {
                calendar.set(Calendar.HOUR_OF_DAY, 8);
                startDate = calendar.getTime();
                calendar.add(Calendar.HOUR_OF_DAY, 8);
                endDate = calendar.getTime();
                clock = DateUtils.format(startDate).concat("-").concat("B");
                analyClass = "乙班";
            } else if (hour >= 0 && hour < 8) {
                calendar.add(Calendar.DAY_OF_YEAR, -1);
                calendar.set(Calendar.HOUR_OF_DAY, 16);
                startDate = calendar.getTime();
                calendar.add(Calendar.HOUR_OF_DAY, 8);
                endDate = calendar.getTime();
                clock = DateUtils.format(startDate).concat("-").concat("B");
                analyClass = "丙班";
            calendar.add(Calendar.DAY_OF_YEAR, -1);
            Date startDate = calendar.getTime();
            String clock = DateUtils.format(startDate);

            calendar.add(Calendar.DAY_OF_YEAR, 1);
            calendar.add(Calendar.MINUTE, -3);
            Date collectStartDate = calendar.getTime();

            // 查询模型结果
            List<StScheduleRecordVO> lastScheduleData = mcsApi.getLastScheduleData(scheduleCode, 1);
            if (CollectionUtils.isEmpty(lastScheduleData)) {
                logger.info("模型结果为空");
                return;
            }
            StScheduleRecordVO stScheduleRecordVO = lastScheduleData.get(0);
            if (stScheduleRecordVO == null) {
                logger.info("模型结果为空");
                return;
            }
            String jsonStr = stScheduleRecordVO.getResultData();

            // 调用模型
     /*       MdkScheduleReqDTO dto = new MdkScheduleReqDTO();
            dto.setScheduleTime(calendar.getTime());
            dto.setScheduleCode(params);
            MdkScheduleRespDTO mdkScheduleRespDTO = mdkApi.doSchedule(dto);
            logger.info(params + "调度方案执行完成，" + mdkScheduleRespDTO);
            Map<String, Object> result = mdkScheduleRespDTO.getResult();

            JSONObject jsonObject = new JSONObject(result);
            JSONObject result2 = (JSONObject) JSON.toJSON(jsonObject.get("result"));*/

            JSONObject jsonObject = JSONObject.parseObject(jsonStr);
            JSONObject result = (JSONObject) JSON.toJSON(jsonObject.get("result"));

            JSONObject result = JSONObject.parseObject(jsonStr);
            if (Objects.isNull(result)) {
                logger.info("模型结果为空");
                return;
@@ -145,23 +113,25 @@

            // 保存报告
            String analyDate = DateUtils.format(startDate);
            String analyContent = result.getString("coalHomeIndexInfo");
            String relId = cokingTraceReportService.save(process, reportName, analyDate, analyClass, clock, analyContent);
            String analyContent = "备煤异常";
            String relId = cokingTraceReportService.save(process, reportName, analyDate, clock, analyContent);

            // 保存一级分析指标
            saveAnalyInd(relId, process, analyDate, analyClass, analyContent);
            cokingAnalyIndService.saveAnalyInd(relId, process, analyDate, analyContent);

            // 保存优化建议
            saveTraceSuggest(relId, process, clock, result);
            String suggest = result.getString(finalResultStrKey);
            cokingTraceSuggestService.saveTraceSuggest(relId, process, clock, suggest, SugObj);

            // 保存偏差值
            saveTraceDeviation(relId, process, clock, result);
            String jsonString = result.getString(resultListKey);
            cokingTraceDeviationService.saveTraceDeviation(relId, process, clock, jsonString, SugObj);

            // 保存溯源指标
            cokingTraceIndService.saveTraceInd(relId, indType, clock);
            cokingTraceIndService.saveTraceInd(relId, indType, clock, collectStartDate, endDate);

            // 保存溯源折线图
            cokingTraceChartService.save(relId, clock, indType, startDate, endDate);
            saveTraceChart(relId, clock, indType, startDate, endDate);

        } catch (Exception ex) {
            logger.error("RunCokingTraceModelBMTask运行异常");
@@ -170,106 +140,57 @@
        logger.info("RunCokingTraceModelBMTask运行完成");
    }

    public void saveTraceChart(String relId, String clock, String indType, Date startDate, Date endDate) {
        logger.info("查询需要保存chart的配置");

    public void saveTraceSuggest(String relId, String process, String clock, JSONObject result) {
        String coalTotal = result.get("coalPrepElecTotal1").toString();
        if (StringUtils.isBlank(coalTotal)) {
            logger.info("coalPrepElecTotal1 is null");
        Calendar calendar = Calendar.getInstance();
        calendar.set(Calendar.MILLISECOND, 0);
        calendar.set(Calendar.SECOND, 0);
        calendar.set(Calendar.MINUTE, 0);
        calendar.set(Calendar.HOUR_OF_DAY, 0);
        calendar.set(Calendar.DAY_OF_MONTH, 1);
        Date monthStartDate = calendar.getTime();

        Map<String, Object> map = new HashMap<>();
        map.put("indType", indType);
        map.put("ext1", "chart");
        List<CokingTraceConfEntity> list = cokingTraceConfService.list(map);
        if (CollectionUtils.isEmpty(list)) {
            return;
        }
        CokingTraceSuggestEntity suggestEntity = new CokingTraceSuggestEntity();
        suggestEntity.setRelId(relId);
        suggestEntity.setProcess(process);
        suggestEntity.setSugObj(SugObj);
        suggestEntity.setClock(clock);
        suggestEntity.setContent(coalTotal);
        suggestEntity.setCreateDate(new Date());
        cokingTraceSuggestService.save(suggestEntity);
    }
        logger.info("list.size={}", list.size());

    @Transactional
    public void saveTraceDeviation(String relId, String process, String clock, JSONObject result) {
        List<ChartParamDTO> list = mcsApi.getChartParamList(CommonConstant.COAL_INDEX_CHARTCODE);
        Map<String, String> steamIndexMaps = list.stream().collect(Collectors.toMap(ChartParamDTO::getParamCode, e -> e.getParamName()));
        List<CokingTraceChartEntity> entityList = new ArrayList<>();

        List<String> coalRowKeys = new ArrayList<>();
        result.forEach((key, value) -> {
            if (StringUtils.isNotBlank(key) && key.contains(coalRow)) {
                coalRowKeys.add(key);
            }
        List<CokingTraceConfEntity> listMin = list.stream().filter(e -> e.getIndCode().equals("BMSY025")).collect(Collectors.toList());

        listMin.forEach(value -> {
            CokingTraceChartEntity cokingTraceChartEntity = new CokingTraceChartEntity();
            cokingTraceChartEntity.setRelId(relId);
            cokingTraceChartEntity.setName(value.getIndName());
            cokingTraceChartEntity.setClock(clock);
            cokingTraceChartEntity.setDataType(value.getDataType());
            cokingTraceChartEntity.setDataNo(value.getPointNo());
            cokingTraceChartEntity.setStartTime(startDate);
            cokingTraceChartEntity.setEndTime(endDate);
            cokingTraceChartEntity.setCreateDate(new Date());
            entityList.add(cokingTraceChartEntity);
        });

        List<CokingTraceDeviationEntity> entityList = new ArrayList<>();
        for (String key : coalRowKeys) {
            JSONArray rowArr = JSON.parseArray(result.get(key).toString());
            if (CollectionUtils.isEmpty(rowArr)) {
                continue;
            }
            CokingTraceDeviationEntity entity = new CokingTraceDeviationEntity();
            entity.setRelId(relId);
            entity.setProcess(process);
            entity.setClock(clock);
            entity.setSugObj(SugObj);
        List<CokingTraceConfEntity> listMonth = list.stream().filter(e -> e.getIndCode().equals("BMSY026")).collect(Collectors.toList());

            entity.setInd1Name(steamIndexMaps.get(new BigDecimal(rowArr.getJSONArray(0).get(0).toString()).setScale(0,BigDecimal.ROUND_HALF_UP).toString()));
            entity.setInd1Value(rowArr.getJSONArray(0).get(1).toString());
            entity.setInd1Unit("");

            entity.setInd2Name(steamIndexMaps.get(new BigDecimal(rowArr.getJSONArray(1).get(0).toString()).setScale(0,BigDecimal.ROUND_HALF_UP).toString()));
            entity.setInd2Value(rowArr.getJSONArray(1).get(1).toString());
            entity.setInd2Unit("");

            if (rowArr.size() > 2) {
                entity.setFac1Name(steamIndexMaps.get(new BigDecimal(rowArr.getJSONArray(2).get(0).toString()).setScale(0,BigDecimal.ROUND_HALF_UP).toString()));
                entity.setFac1Value(rowArr.getJSONArray(2).get(1).toString());
                entity.setFac1Unit("");
            }
            if (rowArr.size() > 3) {
                entity.setFac2Name(steamIndexMaps.get(new BigDecimal(rowArr.getJSONArray(3).get(0).toString()).setScale(0,BigDecimal.ROUND_HALF_UP).toString()));
                entity.setFac2Value(rowArr.getJSONArray(3).get(1).toString());
                entity.setFac2Unit("");
            }
            if (rowArr.size() > 4) {
                entity.setFac3Name(steamIndexMaps.get(new BigDecimal(rowArr.getJSONArray(4).get(0).toString()).setScale(0,BigDecimal.ROUND_HALF_UP).toString()));
                entity.setFac3Value(rowArr.getJSONArray(4).get(1).toString());
                entity.setFac3Unit("");
            }
            if (rowArr.size() > 5) {
                entity.setFac4Name(steamIndexMaps.get(new BigDecimal(rowArr.getJSONArray(5).get(0).toString()).setScale(0,BigDecimal.ROUND_HALF_UP).toString()));
                entity.setFac4Value(rowArr.getJSONArray(5).get(1).toString());
                entity.setFac4Unit("");
            }
            if (rowArr.size() > 6) {
                entity.setFac5Name(steamIndexMaps.get(new BigDecimal(rowArr.getJSONArray(6).get(0).toString()).setScale(0,BigDecimal.ROUND_HALF_UP).toString()));
                entity.setFac5Value(rowArr.getJSONArray(6).get(1).toString());
                entity.setFac5Unit("");
            }
            entityList.add(entity);
        }

        cokingTraceDeviationService.save(entityList);
    }

    public void saveAnalyInd(String relId, String process, String analyDate, String analyClass, String analyContent) {
        if (StringUtils.isBlank(analyContent)) {
            logger.info("analyContent is null");
            return;
        }
        String[] analyContentArr = analyContent.split(";");
        List<CokingAnalyIndEntity> entityList = new ArrayList<>();
        for (int i = 0; i < analyContentArr.length; i++) {
            CokingAnalyIndEntity analyIndEntity = new CokingAnalyIndEntity();
            analyIndEntity.setRelId(relId);
            analyIndEntity.setAnalyType(process);
            analyIndEntity.setAnalyDate(analyDate);
            analyIndEntity.setAnalyClass(analyClass);
            analyIndEntity.setAnalyContent(analyContentArr[i]);
            analyIndEntity.setSort(i + 1);
            entityList.add(analyIndEntity);
        }
        // 清理旧数据
        cokingAnalyIndService.delete(process, analyDate, analyClass);
        cokingAnalyIndService.save(entityList);
        listMonth.forEach(value -> {
            CokingTraceChartEntity cokingTraceChartEntity = new CokingTraceChartEntity();
            cokingTraceChartEntity.setRelId(relId);
            cokingTraceChartEntity.setName(value.getIndName());
            cokingTraceChartEntity.setClock(clock);
            cokingTraceChartEntity.setDataType(value.getDataType());
            cokingTraceChartEntity.setDataNo(value.getPointNo());
            cokingTraceChartEntity.setStartTime(monthStartDate);
            cokingTraceChartEntity.setEndTime(endDate);
            cokingTraceChartEntity.setCreateDate(new Date());
            entityList.add(cokingTraceChartEntity);
        });
        cokingTraceChartService.insert(entityList);
    }
}