package com.moral.api.service.impl;

import com.alibaba.fastjson.JSON;
import com.moral.api.entity.*;
import com.moral.api.pojo.dto.dataDisplay.MonitorPointDataDisplayDTO;
import com.moral.api.pojo.form.dataDisplay.MonitorPointDataDisplayForm;
import com.moral.api.service.*;
import com.moral.constant.Constants;
import com.moral.util.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.ObjectUtils;

import java.util.*;

/**
 * @ClassName DataDisplayServiceImpl
 * @Description TODO
 * @Author 陈凯裕
 * @Date 2021/9/26 9:55
 * @Version TODO
 **/
@Service
public class DataDisplayServiceImpl implements DataDisplayService {

    @Autowired
    DeviceService deviceService;
    @Autowired
    HistoryDailyService historyDailyService;
    @Autowired
    HistoryHourlyService historyHourlyService;
    @Autowired
    HistoryWeeklyService historyWeeklyService;
    @Autowired
    HistoryMonthlyService historyMonthlyService;

    @Override
    public List<MonitorPointDataDisplayDTO> getMonitorPointDisplayData(MonitorPointDataDisplayForm form) {
        //取参
        Integer monitorPointId = form.getMonitorPointId();
        String reportType = form.getReportType();
        Date startTime = form.getStartTime();
        Date endTime = form.getEndTime();
        //根据站点id获取站点下所有的设备
        List<Device> devices = deviceService.getDevicesByMonitorPointId(monitorPointId);
        Map<String, Device> deviceMap = new HashMap<>();
        List<String> macs = new ArrayList<>();
        devices.forEach(value -> {
            macs.add(value.getMac());
            deviceMap.put(value.getMac(), value);
        });
        List<MonitorPointDataDisplayDTO> dtos = new ArrayList<>();
        if (reportType == null) {//处理自定义请求
            Map<String, List<HistoryDaily>> macDataMap = new HashMap<>();//key为mac
            //获取数据
            macs.forEach(value -> {
                //查询数据
                List<HistoryDaily> datas = historyDailyService.getHistoryDailyByMacAndTimeSlot(value, startTime, endTime);
                //如果查询到的数据不为空
                if (datas.size() != 0)
                    macDataMap.put(value, datas);
            });
            //如果都没有数据则不进行计算
            if (macDataMap.size() != 0)
                dtos = calculateCustomData(macDataMap, deviceMap, startTime, endTime);
        }
        //处理小时报表请求，默认取上一个小时的值
        else if (reportType.equals(Constants.HOURYLYREPORT)) {
            Map<String, HistoryHourly> macDataMap = new HashMap<>();
            //获取数据
            macs.forEach(value -> {
                List<HistoryHourly> datas = historyHourlyService.getValueByMacAndTime(value, startTime, startTime);
                if (datas.size() != 0)
                    macDataMap.put(value, datas.get(0));
            });
            if (macDataMap.size() != 0)
                dtos = calculateReportData(macDataMap, deviceMap, startTime);
        }
        //处理日报请求，默认获取昨天的值
        else if (reportType.equals(Constants.DAILYREPORT)) {
            //获取数据
            Map<String, HistoryDaily> macDataMap = historyDailyService.getHistoryDailyByMacsAndDate(macs, startTime);
            if (macDataMap.size() != 0)
                dtos = calculateReportData(macDataMap, deviceMap, startTime);
        }
        //处理周报请求，默认获取上周的值
        else if (reportType.equals(Constants.WEEKLYREPORT)) {
            //获取数据
            Map<String, HistoryWeekly> macDataMap = historyWeeklyService.getHistoryWeeklyByMacsAndDate(macs, startTime);
            if (macDataMap.size() != 0)
                dtos = calculateReportData(macDataMap, deviceMap, startTime, endTime);
        }
        //处理月报请求
        else if (reportType.equals(Constants.MONTHLYREPORT)) {
            //月报需要获取一个月每天的数据用来算综合指数，其他数据从月表中取出
            Map<String, List<HistoryDaily>> macDataDailyMap = new HashMap<>();
            Map<String, HistoryMonthly> macDataMonthlyMap = historyMonthlyService.getHistoryMonthlyByMacsAndDate(macs, startTime);
            macs.forEach(mac -> {
                List<HistoryDaily> dailyDatas = historyDailyService.getHistoryDailyByMacAndTimeSlot(mac, startTime, endTime);
                if (!ObjectUtils.isEmpty(dailyDatas))
                    macDataDailyMap.put(mac, dailyDatas);
            });
            if (macDataMonthlyMap.size() != 0)
                dtos = calculateReportData(macDataMonthlyMap, deviceMap, startTime);
            //注入综合指数
            injectComprehensiveIndex(dtos, macDataDailyMap);
        }
        return dtos;
    }


    /**
     * @Description: 计算自定义报表的数据
     * @Param: [macDatasMap] key为mac,value为数据集合，数据为日数据
     * @return: void
     * @Author: 陈凯裕
     * @Date: 2021/9/26
     */
    private List<MonitorPointDataDisplayDTO> calculateCustomData(Map<String, List<HistoryDaily>> macDataMap, Map<String, Device> deviceMap, Date startDate, Date endDate) {
        List<MonitorPointDataDisplayDTO> dtos = new ArrayList<>();
        macDataMap.forEach((key, value) -> {
            MonitorPointDataDisplayDTO dto = new MonitorPointDataDisplayDTO();
            dto.setMac(key);
            //获取设备名称
            String deviceName = deviceMap.get(key).getName();
            //拼接时间
            String startTime = DateUtils.dateToDateString(startDate, "yyyy-MM-dd");
            String endTime = DateUtils.dateToDateString(endDate, "yyyy-MM-dd");
            String time = startTime + " - " + endTime;
            //计算数据个数
            Double AQINum = 0d;
            Double PM25Num = 0d;
            Double PM10Num = 0d;
            Double SO2Num = 0d;
            Double NO2Num = 0d;
            Double CONum = 0d;
            Double O3Num = 0d;
            Double TVOCNum = 0d;
            //计算AQI,PM2.5,PM10,SO2,NO2,CO,O3,TVOC总值，
            Double AQISum = 0d;
            Double PM25Sum = 0d;
            Double PM10Sum = 0d;
            Double SO2Sum = 0d;
            Double NO2Sum = 0d;
            Double COSum = 0d;
            Double O3Sum = 0d;
            Double TVOCSum = 0d;
            for (HistoryDaily historyDaily : value) {
                //获取数据map
                Map<String, Object> valueMap = JSON.parseObject(historyDaily.getValue(), Map.class);
                Object SO2 = valueMap.get("a21026");
                Object NO2 = valueMap.get("a21004");
                Object PM10 = valueMap.get("a34002");
                Object PM25 = valueMap.get("a34004");
                Object CO = valueMap.get("a21005");
                Object O3 = valueMap.get("a05024");
                Object TVOC = valueMap.get("a99054");
                if (SO2 != null) {
                    SO2Sum = MathUtils.add(SO2Sum, Double.valueOf(SO2.toString()));
                    SO2Num++;
                }
                if (NO2 != null) {
                    NO2Sum = MathUtils.add(NO2Sum, Double.valueOf(NO2.toString()));
                    NO2Num++;
                }
                if (PM10 != null) {
                    PM10Sum = MathUtils.add(PM10Sum, Double.valueOf(PM10.toString()));
                    PM10Num++;
                }
                if (CO != null) {
                    COSum = MathUtils.add(COSum, Double.valueOf(CO.toString()));
                    CONum++;
                }
                if (PM25 != null) {
                    PM25Sum = MathUtils.add(PM25Sum, Double.valueOf(PM25.toString()));
                    PM25Num++;
                }
                if (O3 != null) {
                    O3Sum = MathUtils.add(O3Sum, Double.valueOf(O3.toString()));
                    O3Num++;
                }
                if (TVOC != null) {
                    TVOCSum = MathUtils.add(TVOCSum, Double.valueOf(TVOC.toString()));
                    TVOCNum++;
                }
                //加AQI
                int aqi = AQIUtils.hourlyAqi(valueMap);
                if (aqi != 0) {
                    AQISum += Double.valueOf(aqi);
                    AQINum++;
                }
            }
            //计算aqi均值
            if (AQINum != 0d) {
                Double AQID = AmendUtils.sciCal(AQISum / AQINum, 0);
                int AQIAvg = new Double(AQID).intValue();
                dto.setAQI(AQIAvg);
            }
            //计算PM2.5均值
            if (PM25Num != 0d) {
                Double PM25D = AmendUtils.sciCal(PM25Sum / PM25Num, 0);
                int PM25Avg = new Double(PM25D).intValue();
                dto.setA34004(PM25Avg);
            }
            //计算PM10均值
            if (PM10Num != 0d) {
                Double PM10D = AmendUtils.sciCal(PM10Sum / PM10Num, 0);
                int PM10Avg = new Double(PM10D).intValue();
                dto.setA34002(PM10Avg);
            }
            //计算SO2均值
            if (SO2Num != 0d) {
                Double SO2D = AmendUtils.sciCal(SO2Sum / SO2Num, 0);
                int SO2Avg = new Double(SO2D).intValue();
                dto.setA21026(SO2Avg);
            }
            //计算NO2均值
            if (NO2Num != 0d) {
                Double NO2D = AmendUtils.sciCal(NO2Sum / NO2Num, 0);
                int NO2Avg = new Double(NO2D).intValue();
                dto.setA21004(NO2Avg);
            }
            //计算CO均值，保留两位小数
            if (CONum != 0d) {
                Double COAvg = AmendUtils.sciCal(COSum / CONum, 2);
                dto.setA21005(COAvg);
            }
            //计算O3均值
            if (O3Num != 0d) {
                Double O3D = AmendUtils.sciCal(O3Sum / O3Num, 0);
                int O3Avg = new Double(O3D).intValue();
                dto.setA05024(O3Avg);
            }
            //计算TVOC均值,保留两位小数
            if (TVOCNum != 0d) {
                Double TVOCAvg = AmendUtils.sciCal(TVOCSum / TVOCNum, 2);
                dto.setA99054(TVOCAvg);
            }
            dto.setDeviceName(deviceName);
            dto.setTime(time);
            dtos.add(dto);
        });
        return dtos;
    }


    /**
     * @Description: 计算时报，日报, 周报,月报的数据(只包含6参和AQI，TVOC)
     * @Param: [macDataMap, deviceMap, startDate]
     * @return: java.util.List<com.moral.api.pojo.dto.dataDisplay.MonitorPointDataDisplayDTO>
     * @Author: 陈凯裕
     * @Date: 2021/9/28
     */
    private <T> List<MonitorPointDataDisplayDTO> calculateReportData(Map<String, T> macDataMap, Map<String, Device> deviceMap, Date... date) {
        List<MonitorPointDataDisplayDTO> dtos = new ArrayList<>();
        macDataMap.forEach((key, valueObject) -> {
            MonitorPointDataDisplayDTO dto = new MonitorPointDataDisplayDTO();
            dto.setMac(key);
            //数据的map
            Map<String, Object> valueMap;
            //获取对象的value属性值
            Object valueO = ClassUtils.getPropertyValue(valueObject, "value");
            if (valueO == null)
                return;
            String value = (String) valueO;
            valueMap = JSON.parseObject(value, Map.class);
            //获取设备名称
            String deviceName = deviceMap.get(key).getName();
            dto.setDeviceName(deviceName);
            //拼接时间,时报的时间需要精确到小时，日报精确到当天，周报需要含有开始和结束时间
            String startTime = "";
            if (valueObject instanceof HistoryHourly && date.length == 1)
                startTime = DateUtils.dateToDateString(date[0], "yyyy-MM-dd HH:mm:ss");
            else if (valueObject instanceof HistoryDaily && date.length == 1)
                startTime = DateUtils.dateToDateString(date[0], "yyyy-MM-dd");
            else if (valueObject instanceof HistoryWeekly && date.length == 2)
                startTime = DateUtils.dateToDateString(date[0], "yyyy-MM-dd") + " -- " + DateUtils.dateToDateString(date[1], "yyyy-MM-dd");
            else if (valueObject instanceof HistoryMonthly && date.length == 1)
                startTime = DateUtils.dateToDateString(date[0], "yyyy-MM");
            dto.setTime(startTime);
            //注入数据
            injectDataToDto(valueMap, false, dto);
            dtos.add(dto);
        });
        return dtos;
    }


    /**
     * @Description: 向DTO中注入因子数据，如果数据需要检测标记位，则flag传入true，否则传入false
     * @Param: [valueMap, flag, dto]
     * @return: void
     * @Author: 陈凯裕
     * @Date: 2021/9/28
     */
    private void injectDataToDto(Map<String, Object> valueMap, boolean flag, MonitorPointDataDisplayDTO dto) {
        //注入CO
        String COFlag = (String) valueMap.get("a21005-Flag");
        Object COObject = valueMap.get("a21005");
        if ((flag == false || Constants.NORMAL_FLAG.equals(COFlag)) && COObject != null) {
            Double CO = Double.valueOf(COObject.toString());
            dto.setA21005(AmendUtils.sciCal(CO, 2));
        }
        //注入SO2
        String SO2Flag = (String) valueMap.get("a21026-Flag");
        Object SO2Object = valueMap.get("a21026");
        if ((flag == false || Constants.NORMAL_FLAG.equals(SO2Flag)) && SO2Object != null) {
            Double SO2 = Double.valueOf(SO2Object.toString());
            dto.setA21026(new Double(AmendUtils.sciCal(SO2, 0)).intValue());
        }
        //注入NO2
        String NO2Flag = (String) valueMap.get("a21004-Flag");
        Object NO2Object = valueMap.get("a21004");
        if ((flag == false || Constants.NORMAL_FLAG.equals(NO2Flag)) && NO2Object != null) {
            Double NO2 = Double.valueOf(NO2Object.toString());
            dto.setA21004(new Double(AmendUtils.sciCal(NO2, 0)).intValue());
        }
        //注入PM10
        String PM10Flag = (String) valueMap.get("a34002-Flag");
        Object PM10Object = valueMap.get("a34002");
        if ((flag == false || Constants.NORMAL_FLAG.equals(PM10Flag)) && PM10Object != null) {
            Double PM10 = Double.valueOf(PM10Object.toString());
            dto.setA34002(new Double(AmendUtils.sciCal(PM10, 0)).intValue());
        }
        //注入TVOC
        String TVOCFlag = (String) valueMap.get("a99054-Flag");
        Object TVOCObject = valueMap.get("a99054");
        if ((flag == false || Constants.NORMAL_FLAG.equals(TVOCFlag)) && TVOCObject != null) {
            Double TVOC = Double.valueOf(TVOCObject.toString());
            dto.setA99054(AmendUtils.sciCal(TVOC, 2));
        }
        //注入PM2.5
        String PM25Flag = (String) valueMap.get("a34004-Flag");
        Object PM25Object = valueMap.get("a34004");
        if ((flag == false || Constants.NORMAL_FLAG.equals(PM25Flag)) && PM25Object != null) {
            Double PM25 = Double.valueOf(PM25Object.toString());
            dto.setA34004(new Double(AmendUtils.sciCal(PM25, 0)).intValue());
        }
        //注入O3
        String O3Flag = (String) valueMap.get("a05024-Flag");
        Object O3Object = valueMap.get("a05024");
        if ((flag == false || Constants.NORMAL_FLAG.equals(O3Flag)) && O3Object != null) {
            Double O3 = Double.valueOf(O3Object.toString());
            dto.setA05024(new Double(AmendUtils.sciCal(O3, 0)).intValue());
        }
        //注入AQI
        int aqi = AQIUtils.hourlyAqi(valueMap);
        if (aqi != 0)
            dto.setAQI(aqi);
    }


    /**
     * @Description: 注入综合指数数据
     * @Param: [dtos, macDataDailyMap]
     * @return: void
     * @Author: 陈凯裕
     * @Date: 2021/9/29
     */
    private void injectComprehensiveIndex(List<MonitorPointDataDisplayDTO> dtos, Map<String, List<HistoryDaily>> macDataDailyMap) {
        Map<String, MonitorPointDataDisplayDTO> macDtoMap = new HashMap<>();
        dtos.forEach(value -> macDtoMap.put(value.getMac(), value));
        macDataDailyMap.forEach((key, value) -> {
            List<Map<String, Object>> list = new ArrayList<>();
            for (HistoryDaily historyDaily : value) {
                Map valueMap = JSON.parseObject(historyDaily.getValue(), Map.class);
                list.add(valueMap);
            }
            Double comIndex = ComprehensiveIndexUtils.monthData(list);
            MonitorPointDataDisplayDTO dto = macDtoMap.get(key);
            if (dto != null)
                dto.setComIndex(comIndex);
        });
    }

}