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.*;
import com.iailab.module.ansteel.coking.service.*;
import com.iailab.module.ansteel.common.constant.CommonConstant;
import com.iailab.module.ansteel.common.enums.ProcessConfDataTypeEnum;
import com.iailab.module.data.api.ind.IndItemApi;
import com.iailab.module.data.api.ind.dto.ApiIndItemValueDTO;
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 org.springframework.util.CollectionUtils;

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

/**
 * 备煤工序异常溯源
 *
 * @author lirm
 * @Description
 * @createTime 2025年04月22日
 */
@Component("runCokingTraceModelBMTask")
public class RunCokingTraceModelBMTask 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 CokingTraceChartService cokingTraceChartService;

    @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 SugObj = "BM";

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

    private static final HashMap<String, Object> coalColoumMap = new HashMap<String, Object>() {{
        put("coalColoum0", "一级指标-偏差值");
        put("coalColoum1", "二级指标-偏差值");
        put("coalColoum2", "影响因素1-偏差值");
    }};

    private static final HashMap<String, Object> historyPointMap = new HashMap<String, Object>() {{
        put("F0000101008", "备煤耗电");
    }};

    private static final String jsonStr = "{\n" +
            "    \"result\": {\n" +
            "      \"coalPrepElec\":[1600.8,1613.5],     \n" +
            "      \"coalPrepElecIndex\":[[0.8,0.73],[723,608],[1782,1782],[752,743],[729,783]],\n" +
            "      \"coalPrepElecTotal1\":\"1#粉碎机耗电因粉碎机偏高增加220KW/h，经模型计算，建建议调整单班用煤量，预计可使粉碎机耗电量指标降低170KW/h\",\n" +
            "      \"coalPrepElecTotal2\":\"1#粉碎机耗电量数据异常\",\n" +
            "     \"coalColoum0\":[[0,450],[0,137],[0,342]],\n" +
            "     \"coalColoum1\":[[2,306],[3,134],[1,142]],\n" +
            "     \"coalColoum2\":[[4,142],[4,132],[4,349]],\n" +
            "\"coalPrepElecHomePage\":[[50,42],[10,10]],\n" +
            "\"coalHomeIndex\":[[90.4,90.2],[80.3,78],[103,100],[280,270]],\n" +
            "\"coalIndexInfo\":\"2月18日甲班：备煤耗电异常\",\n" +
            "\"coalErr\":\"2月18日粉碎机耗电数据异常（无数据）\"\n" +
            "    }" +
            "}";

    @Override
    public void run(String params) {
        logger.info("RunCokingTraceModelBMTask，参数为：{}", 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);

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

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

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

            // 保存溯源指标
            this.saveTraceInd(relId, result);

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

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


    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");
        }
        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);
    }

    @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()));
        for (int i = 0; i <= 2; i++) {
            String coalColoum = result.get("coalColoum" + i).toString();
            if (StringUtils.isNotBlank(coalColoum)) {
                JSONArray responseArr = JSON.parseArray(coalColoum);
                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(SugObj);
                    deviationEntity.setGroupName(coalColoumMap.get("coalColoum" + 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");
        }
        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);
    }

    public void saveTraceInd(String relId, JSONObject result) {
        Map<String, Object> map = new HashMap<>();
        map.put("indType", indType);
        List<CokingTraceConfEntity> list = cokingTraceConfService.list(map);
        if (CollectionUtils.isEmpty(list)) {
            logger.info("ConfLis is Empty");
            return;
        }
        List<CokingTraceIndEntity> entityList = new ArrayList<>();
        for (CokingTraceConfEntity conf : list) {
            if (StringUtils.isBlank(conf.getDataType())) {
                logger.info("DataType is Empty");
                continue;
            }
            if (StringUtils.isBlank(conf.getPointNo())) {
                logger.info("PointNo is Empty");
                continue;
            }
            String value = "";
            switch (ProcessConfDataTypeEnum.getEumByCode(conf.getDataType())) {
                case DATAPOINT:
                    List<String> points = new ArrayList<>();
                    points.add(conf.getPointNo());
                    Map<String, Object> pointsRealValue = dataPointApi.queryPointsRealValue(points);
                    value = pointsRealValue.get(conf.getPointNo()).toString();
                    break;
                case IND:
                    List<ApiIndItemValueDTO> indValues = indItemApi.queryIndItemDefaultValue(conf.getPointNo());
                    if (!CollectionUtils.isEmpty(indValues)) {
                        value = indValues.get(indValues.size() - 1).getDataValue().toString();
                    }
                    break;
                default:
                    break;
            }
            CokingTraceIndEntity cokingTraceIndEntity = new CokingTraceIndEntity();
            cokingTraceIndEntity.setRelId(relId);
            cokingTraceIndEntity.setTitle(conf.getIndType());
            cokingTraceIndEntity.setClock(relId);
            cokingTraceIndEntity.setIndCode(conf.getIndCode());
            cokingTraceIndEntity.setIndName(conf.getIndName());
            cokingTraceIndEntity.setIndValue(value);
            cokingTraceIndEntity.setIndUnit(conf.getIndUnit());
            entityList.add(cokingTraceIndEntity);
        }
        cokingTraceIndService.save(entityList);
    }

    public void saveTraceChart(String relId, String clock, Date startDate, Date endDate) {
        // 查询需要保存chart的配置
        Map<String, Object> map = new HashMap<>();
        map.put("indType", indType);
        map.put("ext1", "chart");
        List<CokingTraceConfEntity> list = cokingTraceConfService.list(map);
        if (CollectionUtils.isEmpty(list)) {
            logger.info("TraceChartConfLis is Empty");
        }

        List<CokingTraceChartEntity> entityList = new ArrayList<>();
        list.forEach(value -> {
            CokingTraceChartEntity cokingTraceChartEntity = new CokingTraceChartEntity();
            cokingTraceChartEntity.setRelId(relId);
            cokingTraceChartEntity.setName(value.toString());
            cokingTraceChartEntity.setClock(clock);
            cokingTraceChartEntity.setDataType(value.getDataType());
            cokingTraceChartEntity.setDataNo(value.getPointNo());
            cokingTraceChartEntity.setStartTime(startDate);
            cokingTraceChartEntity.setEndTime(endDate);
            cokingTraceChartEntity.setCreateDate(new Date());
            entityList.add(cokingTraceChartEntity);
        });
        cokingTraceChartService.insert(entityList);
    }
}