package com.ycl.platform.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ycl.platform.domain.entity.CalculateReport;
import com.ycl.platform.domain.entity.CheckIndexCar;
import com.ycl.platform.domain.entity.TMonitor;
import com.ycl.platform.domain.result.HK.SnapshotDataMonitorResult;
import com.ycl.platform.mapper.CalculateReportMapper;
import com.ycl.platform.mapper.CheckIndexCarMapper;
import com.ycl.platform.service.ICheckIndexCarService;
import com.ycl.platform.service.ITMonitorService;
import com.ycl.system.domain.SysConfig;
import com.ycl.system.entity.SysDept;
import com.ycl.system.mapper.SysConfigMapper;
import com.ycl.system.mapper.SysDeptMapper;
import constant.ApiConstants;
import constant.CheckConstants;
import constant.CheckSnapCountConstants;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import utils.DateUtils;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 车辆指标概率数据Service业务层处理
 *
 * @author ruoyi
 * @date 2024-04-29
 */
@Service
@Slf4j
public class CheckIndexCarServiceImpl extends ServiceImpl<CheckIndexCarMapper, CheckIndexCar> implements ICheckIndexCarService {
    @Autowired
    private CheckIndexCarMapper checkIndexCarMapper;
    @Autowired
    private SysConfigMapper sysConfigMapper;
    @Autowired
    private ITMonitorService monitorService;
    @Autowired
    private SysDeptMapper deptMapper;

    /**
     * 查询车辆指标概率数据
     *
     * @param id 车辆指标概率数据主键
     * @return 车辆指标概率数据
     */
    @Override
    public CheckIndexCar selectCheckIndexCarById(Long id) {
        return checkIndexCarMapper.selectCheckIndexCarById(id);
    }

    /**
     * 查询车辆指标概率数据列表
     *
     * @param checkIndexCar 车辆指标概率数据
     * @return 车辆指标概率数据
     */
    @Override
    public List<CheckIndexCar> selectCheckIndexCarList(CheckIndexCar checkIndexCar) {
        return checkIndexCarMapper.selectCheckIndexCarList(checkIndexCar);
    }

    /**
     * 新增车辆指标概率数据
     *
     * @param checkIndexCar 车辆指标概率数据
     * @return 结果
     */
    @Override
    public int insertCheckIndexCar(CheckIndexCar checkIndexCar) {
        checkIndexCar.setCreateTime(DateUtils.getNowDate());
        return checkIndexCarMapper.insertCheckIndexCar(checkIndexCar);
    }

    /**
     * 修改车辆指标概率数据
     *
     * @param checkIndexCar 车辆指标概率数据
     * @return 结果
     */
    @Override
    public int updateCheckIndexCar(CheckIndexCar checkIndexCar) {
        return checkIndexCarMapper.updateCheckIndexCar(checkIndexCar);
    }

    /**
     * 批量删除车辆指标概率数据
     *
     * @param ids 需要删除的车辆指标概率数据主键
     * @return 结果
     */
    @Override
    public int deleteCheckIndexCarByIds(Long[] ids) {
        return checkIndexCarMapper.deleteCheckIndexCarByIds(ids);
    }

    /**
     * 删除车辆指标概率数据信息
     *
     * @param id 车辆指标概率数据主键
     * @return 结果
     */
    @Override
    public int deleteCheckIndexCarById(Long id) {
        return checkIndexCarMapper.deleteCheckIndexCarById(id);
    }

    /**
     * 计算车辆点位在线率、视图库对接稳定性
     * 获取分省厅、区域的map<k,v> k为deptId或者Province_deptId
     * 循环map计算点位在线率
     * 更新或新增
     *
     * @param list
     */
    //TODO：任务顺序需要排在一机一档之后
    @Override
    public void siteOnline(List<SnapshotDataMonitorResult> list) {
        if (CollectionUtils.isEmpty(list)) {
            return;
        }

        Map<String, AreaStats> areaStatsMap = new HashMap<>();
        Map<String, TMonitor> monitorMap = monitorService.list(new QueryWrapper<TMonitor>()
                        .in("serial_number", list.stream().map(SnapshotDataMonitorResult::getExternalIndexCode).collect(Collectors.toList())))
                .stream().collect(Collectors.toMap(TMonitor::getSerialNumber, Function.identity()));

        // TODO: 分省厅市局 需要补充集合数据
        List<String> provinceIds = new ArrayList<>(); // 这里需要根据点位补充

        for (SnapshotDataMonitorResult result : list) {
            TMonitor monitor = monitorMap.get(result.getExternalIndexCode());
            if (monitor == null) continue;

            String deptId = monitor.getDeptId().toString();
            updateAreaStats(areaStatsMap, deptId, result);

            // 处理省厅数据
            if (!CollectionUtils.isEmpty(provinceIds) && provinceIds.contains(monitor.getSerialNumber())) {
                String provinceKey = "Province_" + deptId;
                updateAreaStats(areaStatsMap, provinceKey, result);
            }
        }

        //获取2022同期抓拍平均值 省厅、市局
        BigDecimal cityCountAvg = getAverageCount(CheckSnapCountConstants.City);
        BigDecimal countyCountAvg = getAverageCount(CheckSnapCountConstants.County);

        // 查询今日数据
        List<CheckIndexCar> checkIndexCarList = checkIndexCarMapper.selectToday(DateUtils.getDate());
        List<CheckIndexCar> checkIndexCars = new ArrayList<>();
        areaStatsMap.forEach((deptId, stats) -> {
            if (stats.totalSites > 0) {
                CheckIndexCar checkIndexCar = createOrUpdateCheckIndexCar(deptId, stats, cityCountAvg, countyCountAvg, checkIndexCarList);
                checkIndexCars.add(checkIndexCar);
            }
        });

        saveOrUpdateBatch(checkIndexCars);
    }

    //累计总点位数、离线数、总抓拍量
    private void updateAreaStats(Map<String, AreaStats> areaStatsMap, String key, SnapshotDataMonitorResult result) {
        //返回对象的引用，如果不存在会放入新的key,value
        AreaStats stats = areaStatsMap.computeIfAbsent(key, k -> new AreaStats());
        stats.totalSites++;
        if (ApiConstants.HK_SnapCount_ResultType_Null == result.getResultType()) {
            stats.offlineSites++;
        } else {
            stats.totalDataSum += result.getDataCount();
        }
    }

    //获取2022同期抓拍平均值
    private BigDecimal getAverageCount(String configKey) {
        String count = sysConfigMapper.checkConfigKeyUnique(configKey).getConfigValue();
        return new BigDecimal(count).multiply(new BigDecimal(CheckSnapCountConstants.Multiply))
                .divide(new BigDecimal(CheckSnapCountConstants.CountyNum), 0, RoundingMode.HALF_UP);
    }
    //车辆点位在线率和视图库对接稳定性
    private CheckIndexCar createOrUpdateCheckIndexCar(String key, AreaStats stats, BigDecimal cityCountAvg, BigDecimal countyCountAvg, List<CheckIndexCar> checkIndexCarList) {
        CheckIndexCar checkIndexCar;

        // 检查是否已存在今日数据
        Optional<CheckIndexCar> existingCar = checkIndexCarList.stream()
                .filter(car -> key.equals(car.getDeptId().toString()) &&
                        (key.startsWith("Province_") ? CheckConstants.Examine_Tag_City.equals(car.getExamineTag())
                                : CheckConstants.Examine_Tag_County.equals(car.getExamineTag())))
                .findFirst();

        if (existingCar.isPresent()) {
            checkIndexCar = existingCar.get();
        } else {
            checkIndexCar = new CheckIndexCar();
            checkIndexCar.setDeptId(key.startsWith("Province_") ? Long.parseLong(key.split("_")[1]) : Long.parseLong(key));
            checkIndexCar.setExamineTag(key.startsWith("Province_") ? CheckConstants.Examine_Tag_City : CheckConstants.Examine_Tag_County);
            checkIndexCar.setCreateTime(new Date());
        }

        // 点位在线率
        BigDecimal totalSitesBd = new BigDecimal(stats.totalSites);
        BigDecimal offlineSitesBd = new BigDecimal(stats.offlineSites);
        BigDecimal onlineSitesBd = totalSitesBd.subtract(offlineSitesBd);
        checkIndexCar.setSiteOnline(onlineSitesBd.divide(totalSitesBd, 4, RoundingMode.HALF_UP));
        //视图库对接稳定性
        BigDecimal avgCount = key.startsWith("Province_") ? cityCountAvg : countyCountAvg;
        checkIndexCar.setViewConnectStability(new BigDecimal(stats.totalDataSum).divide(avgCount, 4, RoundingMode.HALF_UP));

        return checkIndexCar;
    }

    //区域车辆点位在线指标的内部类
    private static class AreaStats {
        int totalSites = 0;
        int offlineSites = 0;
        int totalDataSum = 0;
    }

}