package com.moral.api.service.impl;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.moral.api.config.mybatis.MybatisPlusConfig;
import com.moral.api.entity.*;
import com.moral.api.mapper.DeviceMapper;
import com.moral.api.mapper.HistoryHourlyMapper;
import com.moral.api.mapper.OrganizationUnitAlarmMapper;
import com.moral.api.mapper.UnitConversionMapper;
import com.moral.api.service.DeviceService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.moral.api.service.SensorService;
import com.moral.api.service.UnitConversionService;
import com.moral.constant.Constants;
import com.moral.constant.RedisConstants;
import com.moral.constant.SeparateTableType;
import com.moral.util.DateUtils;

import com.moral.util.MybatisPLUSUtils;
import org.apache.kafka.streams.state.internals.metrics.Sensors;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.ObjectUtils;

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

/**
 * <p>
 * 设备表 服务实现类
 * </p>
 *
 * @author moral
 * @since 2021-06-28
 */
@Service
public class DeviceServiceImpl extends ServiceImpl<DeviceMapper, Device> implements DeviceService {

    @Autowired
    private DeviceMapper deviceMapper;

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    OrganizationUnitAlarmMapper organizationUnitAlarmMapper;

    @Autowired
    SensorService sensorService;

    @Autowired
    UnitConversionService unitConversionService;

    @Autowired
    HistoryHourlyMapper historyHourlyMapper;

    @Override
    public List<Device> getDevicesByMonitorPointId(Integer monitorPointId) {
        QueryWrapper<Device> wrapper = new QueryWrapper();
        wrapper.eq("monitor_point_id", monitorPointId);
        wrapper.eq("is_delete", Constants.NOT_DELETE);
        wrapper.orderByAsc("dev_num");
        return deviceMapper.selectList(wrapper);
    }

    @Override
    public Map<String, Object> getSensorsByMac(Map<String, Object> params) {
        List<String> sensorCodes = Arrays.asList(Constants.SENSOR_CODE_PM25
                , Constants.SENSOR_CODE_PM10
                , Constants.SENSOR_CODE_SO2
                , Constants.SENSOR_CODE_NO2
                , Constants.SENSOR_CODE_CO
                , Constants.SENSOR_CODE_O3
                , Constants.SENSOR_CODE_TEMP
                , "a01002"
                , Constants.SENSOR_CODE_WIND_SPEED
                , Constants.SENSOR_CODE_WIND_DIR
                , "a01006"
                , "a00e12"
        );

        //设备mac
        List<String> macs = (List<String>) params.remove("macs");
        List<Map<String, Object>> elementLists = new ArrayList<>();

        for (String mac : macs) {
            //从redis中获取设备因子信息
            Device device = (Device) redisTemplate.opsForHash().get(RedisConstants.DEVICE_INFO, mac);
            List<Sensor> sensors = device.getVersion().getSensors();
            Map<String, Object> map = new HashMap<>();
            for (Sensor sensor : sensors) {
                String sensorCode = sensor.getCode();
                String name = sensor.getName();
                map.put(sensorCode, name);
            }
            elementLists.add(map);
        }

        Map<String, Object> map = elementLists.stream()
                .filter(elementList -> elementList.size() != 0)
                .reduce((a, b) -> {
                    a.keySet().retainAll(b.keySet());
                    return a;
                }).orElse(new HashMap<>());
        Map<String, Object> result = new LinkedHashMap<>();
        sensorCodes.forEach(sensorCode -> {
            Object o = map.remove(sensorCode);
            if (o != null) {
                result.put(sensorCode, o);
            }
        });
        result.putAll(map);
        return result;
    }

    @Override
    public List<Map<String, Object>> getTrendChartData(Map<String, Object> params) {
        Object type = params.get("type");
        //设备mac
        List<String> macs = (List<String>) params.remove("macs");

        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
        queryWrapper.select("mac", "name").in("mac", macs);
        List<Device> devices = deviceMapper.selectList(queryWrapper);

        //所选时间
        List<String> times = (List<String>) params.remove("times");
        //因子code
        String sensorCode = params.get("sensorCode").toString();
        String end;
        String timeUnits;
        String dateFormat;
        //返回结果集,time=data
        List<Map<String, Object>> result = new ArrayList<>();

        for (String start : times) {
            if ("hour".equals(type)) {
                end = DateUtils.getDateAddDay(start, 1);
                String yearAndMonth = DateUtils.dateToDateString(DateUtils.getDate(start, DateUtils.yyyy_MM_dd_EN), DateUtils.yyyyMM_EN);
                timeUnits = "hourly_" + yearAndMonth;
                dateFormat = "%Y-%m-%d %H";
            } else if ("day".equals(type)) {
                end = DateUtils.getDateAddMonth(start, 1);
                timeUnits = "daily";
                dateFormat = "%Y-%m-%d";
            } else {
                end = DateUtils.getDateAddYear(start, 1);
                timeUnits = "monthly";
                dateFormat = "%Y-%m";
            }
            params.put("timeUnits", timeUnits);
            params.put("start", start);
            params.put("end", end);
            params.put("macs", macs);
            params.put("dateFormat", dateFormat);
            //获取多设备指定因子数据
            List<Map<String, Object>> list = deviceMapper.getTrendChartData(params);

            for (String s : DateUtils.getTimeLag(start)) {
                Map<String, Object> resultMap = new HashMap<>();
                resultMap.put("time", s);
                List<Map<String, Object>> deviceData = new ArrayList<>();
                for (Device device : devices) {
                    Map<String, Object> valueMap = new HashMap<>();
                    valueMap.put("name", device.getName());
                    valueMap.put("sensorValue", "0.0");
                    for (Map<String, Object> map : list) {
                        Object time = map.get("time");
                        Object sensorValue = map.get(sensorCode);
                        Object mac = map.get("mac");
                        if (s.equals(time) && device.getMac().equals(mac)) {
                            valueMap.put("sensorValue", sensorValue);
                        }
                    }
                    deviceData.add(valueMap);
                }
                resultMap.put("deviceData", deviceData);
                result.add(resultMap);
            }
        }
        return result;
    }

    @Override
    public List<Map<String, Object>> getTrendChartDataV2(Map<String, Object> params) {
        Object type = params.get("type");
        //设备mac
        List<String> macs = (List<String>) params.remove("macs");

        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
        queryWrapper.select("mac", "name").in("mac", macs);
        List<Device> devices = deviceMapper.selectList(queryWrapper);
        Map<String,Object> devicesInfo = new HashMap<>();
        for (Device device:devices) {
            devicesInfo.put(device.getMac(),device.getName());
        }

        //所选时间
        List<String> times = (List<String>) params.remove("times");
        String startTime = times.get(0);
        String endTime = times.get(1);
        //因子code
        String sensorCode = params.get("sensorCode").toString();
        String end;
        String timeUnits;
        String dateFormat;
        //返回结果集,time=data
        List<Map<String, Object>> result = new ArrayList<>();
        List<Map<String, Object>> list = new ArrayList<>();
        if ("hour".equals(type)) {
            Date startDate = DateUtils.getDate(startTime,DateUtils.yyyy_MM_dd_HH_EN);
            Date endDate = DateUtils.getDate(endTime,DateUtils.yyyy_MM_dd_HH_EN);
            List<String> tableNames = MybatisPLUSUtils.getTableNamesByWrapper(startDate, endDate, SeparateTableType.MONTH);
            for (String mac:macs) {
                QueryWrapper<HistoryHourly> wrapper = new QueryWrapper<>();
                wrapper.eq("mac", mac);
                wrapper.between("time", startDate, endDate);
                List<HistoryHourly> historyHourlies = multiTableQuery(wrapper, tableNames);
                historyHourlies = historyHourlies.stream().distinct().collect(Collectors.toList());
                List<HistoryHourly> distinctHistoryHourlies = new ArrayList<>();
                Map<String, Object> disMap = new HashMap<>();
                for (HistoryHourly historyHourly:historyHourlies) {
                    Date time = historyHourly.getTime();
                    String timeStr = DateUtils.dateToDateString(time,DateUtils.yyyy_MM_dd_HH_mm_ss_EN);
                    if (!disMap.containsKey(timeStr)){
                        distinctHistoryHourlies.add(historyHourly);
                        disMap.put(timeStr,true);
                    }
                }
                for (HistoryHourly historyHourly:distinctHistoryHourlies) {
                    Map<String,Object> historyHourlyMap = new HashMap<>();
                    historyHourlyMap.put("mac",mac);
                    JSONObject value = JSONObject.parseObject(historyHourly.getValue());
                    Double sensorValue = Double.parseDouble(value.get(sensorCode).toString());
                    historyHourlyMap.put(sensorCode,sensorValue);
                    Date time = historyHourly.getTime();
                    String timeStr = DateUtils.dateToDateString(time, DateUtils.yyyy_MM_dd_HH_EN);
                    historyHourlyMap.put("time",timeStr);
                    list.add(historyHourlyMap);
                }
            }
            Date middleDate = DateUtils.getDate(startTime,DateUtils.yyyy_MM_dd_HH_EN);
            while (DateUtils.compareDateStr(endTime,DateUtils.dateToDateString(middleDate,DateUtils.yyyy_MM_dd_HH_EN),DateUtils.yyyy_MM_dd_HH_EN)<=0){
                for (String mac:macs) {
                    boolean flag = true;
                    for (Map map:list) {
                        if (map.get("time").equals(DateUtils.dateToDateString(middleDate, DateUtils.yyyy_MM_dd_HH_EN)) && map.get("mac").toString().equals(mac.toString())){
                            flag = false;
                        }
                    }
                    if (flag){
                        Map<String,Object> historyHourlyMap = new HashMap<>();
                        historyHourlyMap.put("mac",mac);
                        historyHourlyMap.put(sensorCode,"0.0");
                        historyHourlyMap.put("time",DateUtils.dateToDateString(middleDate,DateUtils.yyyy_MM_dd_HH_EN));
                        list.add(historyHourlyMap);
                    }
                }
                middleDate = DateUtils.addHours(middleDate,1);
            }
        } else if ("day".equals(type)) {
            end = DateUtils.getDateAddDay(endTime,1);
            timeUnits = "daily";
            dateFormat = "%Y-%m-%d";
            params.put("timeUnits", timeUnits);
            params.put("start", startTime);
            params.put("end", end);
            params.put("macs", macs);
            params.put("dateFormat", dateFormat);
            //获取多设备指定因子数据
            list = deviceMapper.getTrendChartData(params);
            Date middleDate = DateUtils.getDate(startTime,DateUtils.yyyy_MM_dd_EN);
            while (DateUtils.compareDateStr(endTime,DateUtils.dateToDateString(middleDate,DateUtils.yyyy_MM_dd_EN),DateUtils.yyyy_MM_dd_EN)<=0){
                for (String mac:macs) {
                    boolean flag = true;
                    for (Map map:list) {
                        if (map.get("time").equals(DateUtils.dateToDateString(middleDate, DateUtils.yyyy_MM_dd_EN)) && map.get("mac").toString().equals(mac.toString())){
                            flag = false;
                        }
                    }
                    if (flag){
                        Map<String,Object> historyDailyMap = new HashMap<>();
                        historyDailyMap.put("mac",mac);
                        historyDailyMap.put(sensorCode,"0.0");
                        historyDailyMap.put("time",DateUtils.dateToDateString(middleDate,DateUtils.yyyy_MM_dd_EN));
                        list.add(historyDailyMap);
                    }
                }
                middleDate = DateUtils.addDays(middleDate,1);
            }
        } else {
            end = DateUtils.getDateAddMonth(endTime, 1);
            timeUnits = "monthly";
            dateFormat = "%Y-%m";
            params.put("timeUnits", timeUnits);
            params.put("start", startTime);
            params.put("end", end);
            params.put("macs", macs);
            params.put("dateFormat", dateFormat);
            //获取多设备指定因子数据
            list = deviceMapper.getTrendChartData(params);
            Date middleDate = DateUtils.getDate(startTime,DateUtils.yyyy_MM_EN);
            while (DateUtils.compareDateStr(endTime,DateUtils.dateToDateString(middleDate,DateUtils.yyyy_MM_EN),DateUtils.yyyy_MM_EN)<=0){
                for (String mac:macs) {
                    boolean flag = true;
                    for (Map map:list) {
                        if (map.get("time").equals(DateUtils.dateToDateString(middleDate, DateUtils.yyyy_MM_EN)) && map.get("mac").toString().equals(mac.toString())){
                            flag = false;
                        }
                    }
                    if (flag){
                        Map<String,Object> historyMonthlyMap = new HashMap<>();
                        historyMonthlyMap.put("mac",mac);
                        historyMonthlyMap.put(sensorCode,"0.0");
                        historyMonthlyMap.put("time",DateUtils.dateToDateString(middleDate,DateUtils.yyyy_MM_EN));
                        list.add(historyMonthlyMap);
                    }
                }
                middleDate = DateUtils.addMonths(middleDate,1);
            }
        }
        for (Map map:list) {
            String time = map.get("time").toString();
            Map<String,Object> deviceMap = new HashMap<>();
            deviceMap.put("name",devicesInfo.get(map.get("mac")));
            deviceMap.put("sensorValue",map.get(sensorCode));
            if (result.size()>0){
                boolean flag = true;
                for (Map resultMap:result) {
                    if (resultMap.get("time").toString().equals(time)){
                        List<Map<String,Object>> deviceData = new ArrayList<>();
                        deviceData = (List<Map<String,Object>>)resultMap.get("deviceData");
                        deviceData.add(deviceMap);
                        Collections.sort(deviceData, (map1,map2) -> {
                            String name1 = map1.get("name").toString();//name1是从你list里面拿出来的一个
                            String name2 = map2.get("name").toString(); //name1是从你list里面拿出来的第二个name
                            return name1.compareTo(name2);
                        });
                        resultMap.put("deviceData",deviceData);
                        //result.add(resultMap);
                        flag = false;
                        break;
                    }
                }
                if (flag){
                    List<Map<String,Object>> deviceData = new ArrayList<>();
                    deviceData.add(deviceMap);
                    Collections.sort(deviceData, (map1,map2) -> {
                        String name1 = map1.get("name").toString();//name1是从你list里面拿出来的一个
                        String name2 = map2.get("name").toString(); //name1是从你list里面拿出来的第二个name
                        return name1.compareTo(name2);
                    });
                    Map<String,Object> resultMap = new HashMap<>();
                    resultMap.put("deviceData",deviceData);
                    resultMap.put("time",time);
                    result.add(resultMap);
                }
            }else {
                List<Map<String,Object>> deviceData = new ArrayList<>();
                deviceData.add(deviceMap);
                Collections.sort(deviceData, (map1,map2) -> {
                    String name1 = map1.get("name").toString();//name1是从你list里面拿出来的一个
                    String name2 = map2.get("name").toString(); //name1是从你list里面拿出来的第二个name
                    return name1.compareTo(name2);
                });
                Map<String,Object> resultMap = new HashMap<>();
                resultMap.put("deviceData",deviceData);
                resultMap.put("time",time);
                result.add(resultMap);
            }
        }
        Collections.sort(result, new Comparator<Map<String, Object>>() {
            public int compare(Map<String, Object> o1, Map<String, Object> o2) {
                String id1 = (String) o1.get("time");
                String id2 = (String) o2.get("time");
                return id1.compareTo(id2);
            }

        });
        return result;
    }

    @Override
    public Device getDeviceByMac(String mac) {
        Map<String, Object> deviceMap = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.DEVICE, mac);
        Device device = JSON.parseObject(JSON.toJSONString(deviceMap), Device.class);
        //从map获取organizationId和monitorPointId以及versionId
        Map<String, Object> organizationMap = (Map<String, Object>) deviceMap.get("organization");
        Map<String, Object> monitorPointMap = (Map<String, Object>) deviceMap.get("monitorPoint");
        Map<String, Object> versionMap = (Map<String, Object>) deviceMap.get("version");
        device.setDeviceVersionId((Integer) versionMap.get("id"));
        device.setOrganizationId((Integer) organizationMap.get("id"));
        device.setMonitorPointId((Integer) monitorPointMap.get("id"));
        //如果缓存为空则查询数据库
        if (ObjectUtils.isEmpty(device)) {
            return getDeviceByMacFromDB(mac);
        }
        return device;
    }

    @Override
    public List<Map<String, Object>> getDevicesByOrganizationId(Integer orgId) {
        //从数据库获取mac
        List macs = getMacsByOrganizationId(orgId);
        //从redis获取设备详细信息
        List<Map<String, Object>> result = new ArrayList<>();
        for (Object mac : macs) {
            Map<String, Object> map = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.DEVICE, mac.toString());
            result.add(map);
        }
        return result;
    }

    @Override
    public List getMacsByOrganizationId(Integer organizationId) {
        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
        queryWrapper.select("mac")
                .eq("organization_id", organizationId)
                .eq("is_delete", Constants.NOT_DELETE);
        return deviceMapper.selectObjs(queryWrapper);
    }

    @Override
    public List getMacsByOrgIdAndRegionCode(Integer organizationId, Integer regionCode) {
        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
        queryWrapper.select("mac")
                .eq("organization_id", organizationId)
                .eq("is_delete", Constants.NOT_DELETE)
                .eq("town_code", regionCode);
        return deviceMapper.selectObjs(queryWrapper);
    }

    @Override
    public Device getDeviceUnitAlramInforByMac(String mac) {
        Device device = (Device) redisTemplate.opsForHash().get(RedisConstants.DEVICE_INFO,mac);
        if(device==null)
            return getDeviceUnitAlramInforByMacFromDb(mac);
        return device;
    }

    private Device getDeviceUnitAlramInforByMacFromDb(String mac){
        QueryWrapper<Device> wrapper = new QueryWrapper<>();
        wrapper.eq("mac",mac);
        wrapper.eq("is_delete",Constants.NOT_DELETE);
        Device device = deviceMapper.selectOne(wrapper);
        if(device==null)
            return null;
        QueryWrapper<OrganizationUnitAlarm> unitAlarmQueryWrapper = new QueryWrapper<>();
        unitAlarmQueryWrapper.eq("organization_id",device.getOrganizationId());
        unitAlarmQueryWrapper.eq("version_id",device.getDeviceVersionId());
        unitAlarmQueryWrapper.eq("is_delete",Constants.NOT_DELETE);
        List<OrganizationUnitAlarm> organizationUnitAlarms = organizationUnitAlarmMapper.selectList(unitAlarmQueryWrapper);
        Version version = new Version();
        version.setId(device.getDeviceVersionId());
        List<Sensor> sensors = new ArrayList<>();
        for (OrganizationUnitAlarm organizationUnitAlarm : organizationUnitAlarms) {
            Sensor sensor = sensorService.getSensorByCode(organizationUnitAlarm.getSensorCode());
            sensor.setUnit(organizationUnitAlarm.getUnitKey());
            sensor.setShowUnit(organizationUnitAlarm.getShowUnitKey());
            sensor.setShowUnitKey(organizationUnitAlarm.getShowUnitKey());
            sensor.setUnitKey(organizationUnitAlarm.getUnitKey());
            sensor.setAlarmLevel(organizationUnitAlarm.getAlarmLevel());
            String formula = unitConversionService.getFormula(Integer.valueOf(organizationUnitAlarm.getUnitKey()), Integer.valueOf(organizationUnitAlarm.getShowUnitKey()),sensor.getCode());
            sensor.setFormula(formula);
            sensors.add(sensor);
        }
        version.setSensors(sensors);
        device.setVersion(version);
        redisTemplate.opsForHash().put(RedisConstants.DEVICE_INFO,mac,device);
        return device;
    }

    private Device getDeviceByMacFromDB(String mac) {
        QueryWrapper<Device> wrapper = new QueryWrapper<>();
        wrapper.eq("mac", mac);
        wrapper.eq("is_delete", Constants.NOT_DELETE);
        return deviceMapper.selectOne(wrapper);
    }

    /**
     * @Description: 多表查询，传入表名集合，以及条件wrapper，返回数据
     * @Param: [wrapper, tableNames]
     * @return: java.util.List<com.moral.api.entity.HistoryHourly>
     * @Author: 陈凯裕
     * @Date: 2021/9/23
     */
    private List<HistoryHourly> multiTableQuery(QueryWrapper<HistoryHourly> wrapper, List<String> tableNames) {
        List<HistoryHourly> result = new ArrayList<>();
        for (String tableName : tableNames) {
            MybatisPlusConfig.tableName.set(tableName);
            List<HistoryHourly> datas = historyHourlyMapper.selectList(wrapper);
            result.addAll(datas);
        }
        MybatisPlusConfig.tableName.remove();
        return result;
    }


}