package com.iailab.module.shasteel.job.task;

import com.alibaba.fastjson.JSONArray;
import com.iailab.module.model.api.mcs.McsApi;
import com.iailab.module.model.api.mcs.dto.ScheduleSuggestRespDTO;
import com.iailab.module.model.api.mcs.dto.StScheduleSchemeDTO;
import com.iailab.module.model.api.mdk.MdkApi;
import com.iailab.module.model.api.mdk.dto.MdkScheduleReqDTO;
import com.iailab.module.model.api.mdk.dto.MdkScheduleRespDTO;
import com.iailab.module.model.enums.CommonConstant;
import com.iailab.module.model.enums.ScheduleTriggerMethodEnum;
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.util.CollectionUtils;

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

/**
 * 06
 * 氧气班平衡计划
 * 法夜空建议送出量	faAdvice
 * 盈德低压建议送出量	yingDiAdvice
 * 盈德中压建议送出量	yingZhongAdvice
 * 宏昌建议送出量	hongAdvice
 * 润忠建议送出量	RunAdvice
 * 调整值	adviceValues
 *
 * 05
 * 氧气系统优化调度决策
 * 盈德中压建议调整量	yingdiAdvice
 * 法夜空建议调整量	faAdvice
 * 宏昌建议调整量	hongAdvice
 * 润忠建议调整量	RunAdvice
 *
 * 01
 * 氮气预警模型
 * 预警信号	advice
 *
 * @author PanZhibao
 * @Description
 * @createTime 2025年01月07日
 */
@Component("runScheduleOnaOptTask")
public class RunScheduleOnaOptTask implements ITask {
    private Logger logger = LoggerFactory.getLogger(getClass());

    @Autowired
    private McsApi mcsApi;

    @Autowired
    private MdkApi mdkApi;

    private Map<String, StScheduleSchemeDTO> schemeMap = new HashMap<>();

    /**
     * 调度调整方向
     * 1：调整值为正数；-1：调整值为负数
     */
    private static Map<String, Integer> gasAdjustMode = new HashMap<>();

    /**
     * 调度建议时间
     */
    private static Map<String, Date> gasLastSugTime = new HashMap<>();

    private static final String CODE06 = "06";

    private static final String CODE05 = "05";

    private static final String CODE01 = "01";

    @Override
    public void run(String params) {
        logger.info("RunScheduleOnaOptTask，参数为：{}", params);
        try {
            Calendar calendar = Calendar.getInstance();
            calendar.set(Calendar.MILLISECOND, 0);
            calendar.set(Calendar.SECOND, 0);
            Date scheduleTime = calendar.getTime();
            if (StringUtils.isEmpty(params)) {
                logger.info("参数为空");
                return;
            }

            // 根据条件查找方案
            List<StScheduleSchemeDTO> schemeList = mcsApi.listScheduleScheme(ScheduleTriggerMethodEnum.JOB.getCode(), params);
            if (CollectionUtils.isEmpty(schemeList)) {
                logger.info("方案为空");
                return;
            }
            schemeMap.clear();
            Map<String, MdkScheduleRespDTO> resultMap = new HashMap<String, MdkScheduleRespDTO>();
            schemeList.forEach(scheme -> {
                schemeMap.put(scheme.getCode(), scheme);
                MdkScheduleReqDTO dto = new MdkScheduleReqDTO();
                dto.setScheduleTime(scheduleTime);
                dto.setScheduleCode(scheme.getCode());
                MdkScheduleRespDTO mdkScheduleRespDTO = mdkApi.doSchedule(dto);
                resultMap.put(scheme.getCode(), mdkScheduleRespDTO);
            });
            if (CollectionUtils.isEmpty(resultMap)) {
                logger.info("resultMap为空");
                return;
            }
            for (Map.Entry<String, MdkScheduleRespDTO> entry : resultMap.entrySet()) {
                if (!CommonConstant.MDK_STATUS_100.equals(entry.getValue().getStatusCode())) {
                    logger.info("调度结果异常，STATUS=" + entry.getValue().getStatusCode());
                    continue;
                }

                if (entry.getKey().equals(CODE06)) {
                    // 氧气班平衡计划，建议值写入测点
                    mdkApi.scheduleModelOut(entry.getValue());
                    logger.info("氧气班平衡计划下发参数: {}", entry.getValue());
                } else if (entry.getKey().equals(CODE05)) {
                    // 氧气系统优化调度决策
                    StScheduleSchemeDTO scheduleScheme = schemeMap.get(CODE05);
                    String scheduleObj = scheduleScheme.getScheduleObj();
                    String yingdiAdvice = entry.getValue().getResult().get("yingdiAdvice").toString();
                    String faAdvice = entry.getValue().getResult().get("faAdvice").toString();
                    String hongAdvice = entry.getValue().getResult().get("hongAdvice").toString();
                    String runAdvice = entry.getValue().getResult().get("runAdvice").toString();
                    /*JSONArray adviceValues = JSONArray.parseArray(entry.getValue().getResult().get("adviceValues").toString());
                    List<BigDecimal> adviceValueList = new ArrayList<>();
                    if (!CollectionUtils.isEmpty(adviceValues)) {
                        for (int i = 0; i < adviceValues.size(); i++) {
                            adviceValueList.add(new BigDecimal(adviceValues.get(i).toString()));
                        }
                    }*/
                    if(!yingdiAdvice.equals("压力正常")){
                        saveScheduleSuggest("盈德中压调整建议", yingdiAdvice, scheduleObj, scheduleTime);
                    }
                    if(!faAdvice.equals("压力正常")){
                        saveScheduleSuggest("法夜空调整建议", faAdvice, scheduleObj, scheduleTime);
                    }
                    if(!hongAdvice.equals("压力正常")){
                        saveScheduleSuggest("宏昌调整建议", hongAdvice, scheduleObj, scheduleTime);
                    }
                    if(!runAdvice.equals("压力正常")){
                        saveScheduleSuggest("润忠调整建议", runAdvice, scheduleObj, scheduleTime);
                    }
                } else if (entry.getKey().equals(CODE01)) {
                    StScheduleSchemeDTO scheduleScheme = schemeMap.get(CODE05);
                    String scheduleObj = scheduleScheme.getScheduleObj();
                    String advice = entry.getValue().getResult().get("advice").toString();
                    if(!"压力正常".equals(advice)){
                        saveScheduleSuggest("氮气预警", advice, scheduleObj, scheduleTime);
                    }
                }
            }
            logger.info(params + "调度方案执行完成");
        } catch (Exception ex) {
            logger.error("RunScheduleOnaOptTask运行异常");
            ex.printStackTrace();
        }
        logger.info("RunScheduleOnaOptTask运行完成");
    }

    private void saveScheduleSuggest(String title, String content, String scheduleObj, Date scheduleTime) {
        if (StringUtils.isBlank(content)) {
            logger.info(title + "调整值为空，不产生建议");
            return;
        }

        ScheduleSuggestRespDTO suggest = new ScheduleSuggestRespDTO();
        suggest.setTitle(title);
        suggest.setContent(content);
        suggest.setScheduleObj(scheduleObj);
        suggest.setScheduleTime(scheduleTime);
        suggest.setSort(0);
        mcsApi.createScheduleSuggest(suggest);
    }

    private void saveScheduleSuggest(String title, String content, String scheduleObj, Date scheduleTime, BigDecimal adjustValue, int adjustRate) {
        if (StringUtils.isBlank(content)) {
            logger.info(title + "调整值为空，不产生建议");
            return;
        }
        if (adjustValue == null || adjustValue.doubleValue() == 0) {
            logger.info(title + "调整值为空，不产生建议");
            return;
        }
        if (gasAdjustMode.containsKey(title) && gasAdjustMode.get(title).equals(adjustValue.compareTo(BigDecimal.ZERO)) &&
                gasLastSugTime.containsKey(title) && (scheduleTime.getTime() - gasLastSugTime.get(title).getTime() <= Integer.valueOf(adjustRate).longValue() * 60 * 1000)) {
            logger.info("调整建议在有效期内");
            return;
        }
        gasAdjustMode.put(title,  adjustValue.compareTo(BigDecimal.ZERO));
        gasLastSugTime.put(title, scheduleTime);

        ScheduleSuggestRespDTO suggest = new ScheduleSuggestRespDTO();
        suggest.setTitle(title);
        suggest.setContent(content);
        suggest.setScheduleObj(scheduleObj);
        suggest.setScheduleTime(scheduleTime);
        suggest.setSort(0);
        mcsApi.createScheduleSuggest(suggest);
    }
}