package cn.iocoder.yudao.module.bi.service.decision;

import cn.iocoder.yudao.module.bi.dal.dataobject.decision.BiKpiDefinitionDO;
import cn.iocoder.yudao.module.bi.dal.dataobject.decision.BiKpiValueHistoryDO;
import cn.iocoder.yudao.module.bi.dal.mysql.decision.BiKpiDefinitionMapper;
import cn.iocoder.yudao.module.bi.dal.mysql.decision.BiKpiValueHistoryMapper;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;

/**
 * KPI历史快照聚合任务
 *
 * 定时扫描启用的 KPI 定义，执行查询 SQL 将当前值写入 bi_kpi_value_history，
 * 供趋势研判、同比环比与统计预测使用。同一 KPI 同一整点仅保留最新值。
 *
 * @author 超级管理员
 */
@Slf4j
@Component
public class KpiSnapshotTask {

    @Resource
    private BiKpiDefinitionMapper kpiMapper;

    @Resource
    private BiKpiValueHistoryMapper kpiValueHistoryMapper;

    @Resource
    private JdbcTemplate jdbcTemplate;

    /**
     * 每小时整点触发，默认 cron = 0 0 * * * *，可通过 bi.kpi.snapshot.cron 覆盖
     */
    @Scheduled(cron = "${bi.kpi.snapshot.cron:0 0 * * * *}")
    public void snapshotAllKpis() {
        log.debug("[snapshotAllKpis][开始 KPI 快照聚合]");
        List<BiKpiDefinitionDO> kpis = kpiMapper.selectList(BiKpiDefinitionDO::getStatus, 1);
        if (kpis.isEmpty()) {
            log.debug("[snapshotAllKpis][无启用 KPI]");
            return;
        }

        // 当前整点，保证同一 KPI 同整点幂等
        LocalDateTime snapshotTime = LocalDateTime.now().withMinute(0).withSecond(0).withNano(0);
        for (BiKpiDefinitionDO kpi : kpis) {
            try {
                snapshotKpi(kpi, snapshotTime);
            } catch (Exception e) {
                log.error("[snapshotAllKpis][KPI {} 快照失败]", kpi.getCode(), e);
            }
        }
    }

    private void snapshotKpi(BiKpiDefinitionDO kpi, LocalDateTime snapshotTime) {
        BigDecimal value = executeKpiSql(kpi.getQuerySql());
        if (value == null) {
            log.warn("[snapshotKpi][KPI {} 无值，跳过快照]", kpi.getCode());
            return;
        }

        BiKpiValueHistoryDO exist = kpiValueHistoryMapper
                .selectByKpiCodeAndSnapshotTime(kpi.getCode(), snapshotTime);
        if (exist != null) {
            exist.setKpiValue(value);
            kpiValueHistoryMapper.updateById(exist);
        } else {
            BiKpiValueHistoryDO history = BiKpiValueHistoryDO.builder()
                    .kpiId(kpi.getId())
                    .kpiCode(kpi.getCode())
                    .kpiName(kpi.getName())
                    .category(kpi.getCategory())
                    .kpiValue(value)
                    .snapshotTime(snapshotTime)
                    .periodType("hour")
                    .dimensionValue("all")
                    .build();
            kpiValueHistoryMapper.insert(history);
        }
    }

    private BigDecimal executeKpiSql(String querySql) {
        if (querySql == null || querySql.isBlank()) {
            return null;
        }
        try {
            List<Map<String, Object>> result = jdbcTemplate.queryForList(querySql);
            if (result.isEmpty()) {
                return null;
            }
            Object value = result.get(0).values().iterator().next();
            if (value instanceof Number) {
                return BigDecimal.valueOf(((Number) value).doubleValue());
            }
            return null;
        } catch (Exception e) {
            log.error("[executeKpiSql][SQL执行失败] {}", querySql, e);
            return null;
        }
    }

}