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

import cn.iocoder.yudao.module.bi.dal.dataobject.decision.BiAlertRecordDO;
import cn.iocoder.yudao.module.bi.dal.dataobject.decision.BiAlertRuleDO;
import cn.iocoder.yudao.module.bi.dal.dataobject.decision.BiKpiDefinitionDO;
import cn.iocoder.yudao.module.bi.dal.mysql.decision.BiAlertRecordMapper;
import cn.iocoder.yudao.module.bi.dal.mysql.decision.BiAlertRuleMapper;
import cn.iocoder.yudao.module.bi.dal.mysql.decision.BiKpiDefinitionMapper;
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.time.LocalDateTime;
import java.util.List;
import java.util.Map;

/**
 * 预警定时评估任务
 *
 * 定期扫描所有活跃预警规则，评估阈值并生成预警记录
 *
 * @author 超级管理员
 */
@Slf4j
@Component
public class AlertEvaluationTask {

    @Resource
    private BiAlertRuleMapper alertRuleMapper;

    @Resource
    private BiKpiDefinitionMapper kpiMapper;

    @Resource
    private BiAlertRecordMapper alertRecordMapper;

    @Resource
    private JdbcTemplate jdbcTemplate;

    /**
     * 每5分钟评估一次所有活跃规则
     */
    @Scheduled(fixedDelayString = "${bi.alert.evaluation.interval:300000}")
    public void evaluateAllAlertRules() {
        log.debug("[evaluateAllAlertRules][开始评估预警规则]");
        List<BiAlertRuleDO> rules = alertRuleMapper.selectList(
                BiAlertRuleDO::getEnabled, 1);
        if (rules.isEmpty()) {
            return;
        }

        for (BiAlertRuleDO rule : rules) {
            try {
                evaluateRule(rule);
            } catch (Exception e) {
                log.error("[evaluateAllAlertRules][规则 {} 评估异常]", rule.getId(), e);
            }
        }
    }

    private void evaluateRule(BiAlertRuleDO rule) {
        BiKpiDefinitionDO kpi = kpiMapper.selectById(rule.getKpiId());
        if (kpi == null || kpi.getStatus() == 0) {
            return;
        }

        // 执行 KPI 查询
        BigDecimal currentValue = executeQuery(kpi.getQuerySql());
        if (currentValue == null) {
            return;
        }

        // 比较阈值
        int cmp = currentValue.compareTo(rule.getThresholdValue());
        boolean triggered = switch (rule.getComparisonOperator()) {
            case ">" -> cmp > 0;
            case "<" -> cmp < 0;
            case ">=" -> cmp >= 0;
            case "<=" -> cmp <= 0;
            default -> false;
        };

        if (!triggered) {
            return;
        }

        // 检查是否已有未处理的同KPI预警（去重）
        Long existingCount = alertRecordMapper.selectCountByStatus(0);
        if (existingCount > 0) {
            List<BiAlertRecordDO> existing = alertRecordMapper.selectListByStatus(0);
            boolean dup = existing.stream().anyMatch(r -> r.getKpiId().equals(rule.getKpiId()));
            if (dup) {
                return;
            }
        }

        // 生成预警记录
        BiAlertRecordDO record = BiAlertRecordDO.builder()
                .ruleId(rule.getId())
                .kpiId(kpi.getId())
                .kpiCode(kpi.getCode())
                .kpiName(kpi.getName())
                .actualValue(currentValue)
                .thresholdValue(rule.getThresholdValue())
                .severity(rule.getSeverity())
                .alertMessage(String.format("%s 当前值 %s 超出阈值 %s",
                        kpi.getName(), currentValue, rule.getThresholdValue()))
                .status(0)
                .triggeredTime(LocalDateTime.now())
                .build();
        alertRecordMapper.insert(record);
        log.warn("[evaluateRule][触发预警] KPI={}, 当前值={}, 阈值={}",
                kpi.getName(), currentValue, rule.getThresholdValue());
    }

    private BigDecimal executeQuery(String sql) {
        if (sql == null || sql.isBlank()) {
            return null;
        }
        try {
            List<Map<String, Object>> result = jdbcTemplate.queryForList(sql);
            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("[executeQuery][SQL执行失败] {}", sql, e);
            return null;
        }
    }

}