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.service.*;
import com.iailab.module.ansteel.common.constant.CommonConstant;
import com.iailab.module.data.api.ind.IndItemApi;
import com.iailab.module.data.api.point.DataPointApi;
import com.iailab.module.model.api.mcs.McsApi;
import com.iailab.module.model.api.mcs.dto.ChartParamDTO;
import org.apache.commons.lang3.StringUtils;
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 java.util.*;
import java.util.stream.Collectors;

/**
 * 干熄焦工序-异常溯源
 *
 * @author lirm
 * @Description
 * @createTime 2025年04月17日
 */
@Component("runCokingTraceModelGXJTask")
public class RunCokingTraceModelGXJTask implements ITask {

    private Logger logger = LoggerFactory.getLogger(getClass());

    @Autowired
    private CokingTraceReportService cokingTraceReportService;

    @Autowired
    private CokingTraceSuggestService cokingTraceSuggestService;

    @Autowired
    private CokingTraceDeviationService cokingTraceDeviationService;

    @Autowired
    private CokingAnalyIndService cokingAnalyIndService;

    @Autowired
    private CokingTraceIndService cokingTraceIndService;

    @Autowired
    private CokingTraceConfService cokingTraceConfService;

    @Autowired
    private McsApi mcsApi;

    @Autowired
    private DataPointApi dataPointApi;

    @Autowired
    private IndItemApi indItemApi;

    private final static String process = "干熄焦工序";

    private final static String reportName = "干熄焦工序异常溯源";

    private final static String indType = "干熄焦工序异常溯源";
    private static final HashMap<String, Object> steamFirstColoumMap = new HashMap<String, Object>() {{
        put("steamFirstColoum0", "一级指标-偏差值");
        put("steamFirstColoum1", "二级指标-偏差值");
        put("steamFirstColoum2", "影响因素1-偏差值");
        put("steamFirstColoum3", "影响因素2-偏差值");
        put("steamFirstColoum4", "影响因素3-偏差值");
        put("steamFirstColoum5", "影响因素n-偏差值");
    }};

    private static final HashMap<String, Object> steamSecondColoumMap = new HashMap<String, Object>() {{
        put("steamSecondColoum0", "一级指标-偏差值");
        put("steamSecondColoum1", "二级指标-偏差值");
        put("steamSecondColoum2", "影响因素1-偏差值");
        put("steamSecondColoum3", "影响因素2-偏差值");
        put("steamSecondColoum4", "影响因素3-偏差值");
        put("steamSecondColoum5", "影响因素n-偏差值");
    }};

    private static final String jsonStr="{\"result\":{" +
            "      \"steamTotal1\":\"2#干熄焦蒸汽发生量因空气导入量偏高、锅炉入口温度偏低而减少13t，经模型计算，建议调整空气导入量，预计可使主蒸汽流量指标升高20m3/h\",\n" +
            "      \"steamTotal2\":\"2#干熄焦蒸汽发生量因空气导入量偏高、锅炉入口温度偏低而减少13m3/h，经模型计算，建议调整循环风机转速，预计可使主蒸汽流量指标升高32m3/h\",\n" +
            "      \"steamFirstColoum0\":[[14,20],[14,30],[14,35]],\n" +
            "      \"steamFirstColoum1\":[[10,5],[11,15],[21,20]],\n" +
            "      \"steamFirstColoum2\":[[74,8],[36,18],[35,34]],\n" +
            "      \"steamFirstColoum3\":[[24,67],[6,12],[19,21]],\n" +
            "      \"steamFirstColoum4\":[[21,19],[22,31],[8,42]],\n" +
            "      \"steamFirstColoum5\":[[11,45],[9,62],[18,34]],\n" +
            "      \"steamSecondColoum0\":[[14,20],[14,50],[14,13]],\n" +
            "      \"steamSecondColoum1\":[[11,27],[10,35],[22,16]],\n" +
            "      \"steamSecondColoum2\":[[36,90],[35,34],[74,55]],\n" +
            "      \"steamSecondColoum3\":[[6,66],[19,73],[21,31]],\n" +
            "      \"steamSecondColoum4\":[[22,65],[8,22],[21,67]],\n" +
            "      \"steamSecondColoum5\":[[9,29],[11,56],[18,61]],\n" +
            "      \"steamHomePage\":[[663,617],[66.5,68.9]],\n" +
            "      \"steamHomeIndex\":[[90.4,90.2],[80.3,78],[103,100],[280,270]],\n" +
            "      \"steamHomeIndexInfo\":\"2月18日甲班：干熄焦产蒸汽量异常\",\n" +
            "      \"steamHomeErr\":\"2月18日循环风机转速数据异常（无数据）\"}}";

    @Override
    public void run(String params) {
        logger.info("RunCokingTraceModelGXJTask，参数为：{}", params);
        try {
            Calendar calendar = Calendar.getInstance();
            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();
            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 = "丙班";
            }

            // 调用模型
            JSONObject jsonObject = JSONObject.parseObject(jsonStr);
            JSONObject result = (JSONObject) JSON.toJSON(jsonObject.get("result"));

            if (Objects.isNull(result)) {
                logger.info("模型结果为空");
                return;
            }

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

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

            // 保存优化建议
            saveTraceSuggest(relId, process, clock, result);

            // 保存偏差值
            saveTraceDeviation(relId, process, clock, result);

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

        } catch (Exception ex) {
            logger.error("RunCokingTraceModelGXJTask运行异常");
            ex.printStackTrace();
        }
        logger.info("RunCokingTraceModelGXJTask运行完成");

    }

    public void saveTraceSuggest(String relId, String process, String clock, JSONObject result) {
        List<CokingTraceSuggestEntity> entityList = new ArrayList<>();
        for(int i=1;i<=2;i++){
            String steamTotal = result.get("steamTotal"+i).toString();
            if(StringUtils.isNotBlank(steamTotal)){
                CokingTraceSuggestEntity suggestEntity = new CokingTraceSuggestEntity();
                suggestEntity.setRelId(relId);
                suggestEntity.setProcess(process);
                suggestEntity.setSugObj(i+"#");
                suggestEntity.setClock(clock);
                suggestEntity.setContent(steamTotal);
                suggestEntity.setCreateDate(new Date());
                entityList.add(suggestEntity);
            }
        }
        cokingTraceSuggestService.insert(entityList);
    }

    @Transactional
    public void saveTraceDeviation(String relId, String process, String clock, JSONObject result) {
        List<ChartParamDTO> list = mcsApi.getChartParamList(CommonConstant.STEAM_INDEX_CHARTCODE);
        Map<String, String> steamIndexMaps = list.stream().collect(Collectors.toMap(ChartParamDTO::getParamCode, e -> e.getParamName()));
        for(int i=0;i<=5;i++){
            String steamFirstColoum = result.get("steamFirstColoum"+i).toString();
            if(StringUtils.isNotBlank(steamFirstColoum)){
                JSONArray responseArr = JSON.parseArray(steamFirstColoum);
                List<CokingTraceDeviationEntity> entityList = new ArrayList<>();
                for (int j = 0; j < responseArr.size(); j++) {
                    JSONArray element = JSON.parseArray(responseArr.get(j).toString());
                    CokingTraceDeviationEntity deviationEntity = new CokingTraceDeviationEntity();
                    deviationEntity.setRelId(relId);
                    deviationEntity.setProcess(process);
                    deviationEntity.setClock(clock);
                    deviationEntity.setSugObj("1#");
                    deviationEntity.setGroupName(steamFirstColoumMap.get("steamFirstColoum"+i).toString());
                    deviationEntity.setIndName(steamIndexMaps.get(element.get(0).toString()));
                    deviationEntity.setIndValue(element.get(1).toString());
                    deviationEntity.setCreateDate(new Date());
                    entityList.add(deviationEntity);
                }
                cokingTraceDeviationService.save(entityList);
            }
        }
        for(int i=0;i<=5;i++){
            String steamSecondColoum = result.get("steamSecondColoum"+i).toString();
            if(StringUtils.isNotBlank(steamSecondColoum)){
                JSONArray responseArr = JSON.parseArray(steamSecondColoum);
                List<CokingTraceDeviationEntity> entityList = new ArrayList<>();
                for (int j = 0; j < responseArr.size(); j++) {
                    JSONArray element = JSON.parseArray(responseArr.get(j).toString());
                    CokingTraceDeviationEntity deviationEntity = new CokingTraceDeviationEntity();
                    deviationEntity.setRelId(relId);
                    deviationEntity.setProcess(process);
                    deviationEntity.setClock(clock);
                    deviationEntity.setSugObj("2#");
                    deviationEntity.setGroupName(steamSecondColoumMap.get("steamSecondColoum" + i).toString());
                    deviationEntity.setIndName(steamIndexMaps.get(element.get(0).toString()));
                    deviationEntity.setIndValue(element.get(1).toString());
                    deviationEntity.setCreateDate(new Date());
                    entityList.add(deviationEntity);
                }
                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.save(entityList);
    }
}