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

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.framework.common.exception.ServiceException;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.bi.controller.admin.decision.vo.DecisionAlertRuleImportExcelVO;
import cn.iocoder.yudao.module.bi.controller.admin.decision.vo.DecisionAlertRuleImportReqVO;
import cn.iocoder.yudao.module.bi.controller.admin.decision.vo.DecisionAlertRuleImportRespVO;
import cn.iocoder.yudao.module.bi.controller.admin.decision.vo.DecisionAlertRulePageReqVO;
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.BiAlertRuleMapper;
import cn.iocoder.yudao.module.bi.dal.mysql.decision.BiKpiDefinitionMapper;
import cn.iocoder.yudao.module.bi.enums.ErrorCodeConstants;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.filterList;

/**
 * 预警规则 Service 实现
 *
 * @author 超级管理员
 */
@Service
public class BiAlertRuleServiceImpl implements BiAlertRuleService {

    @Resource
    private BiAlertRuleMapper alertRuleMapper;

    @Resource
    private BiKpiDefinitionMapper kpiMapper;

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Long createRule(BiAlertRuleDO rule) {
        // 校验KPI存在
        if (kpiMapper.selectById(rule.getKpiId()) == null) {
            throw exception(ErrorCodeConstants.ALERT_RULE_KPI_NOT_EXISTS);
        }
        alertRuleMapper.insert(rule);
        return rule.getId();
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateRule(BiAlertRuleDO rule) {
        if (alertRuleMapper.selectById(rule.getId()) == null) {
            throw exception(ErrorCodeConstants.ALERT_RULE_NOT_EXISTS);
        }
        if (rule.getKpiId() != null && kpiMapper.selectById(rule.getKpiId()) == null) {
            throw exception(ErrorCodeConstants.ALERT_RULE_KPI_NOT_EXISTS);
        }
        alertRuleMapper.updateById(rule);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteRule(Long id) {
        if (alertRuleMapper.selectById(id) == null) {
            throw exception(ErrorCodeConstants.ALERT_RULE_NOT_EXISTS);
        }
        alertRuleMapper.deleteById(id);
    }

    @Override
    public BiAlertRuleDO getRule(Long id) {
        return alertRuleMapper.selectById(id);
    }

    @Override
    public PageResult<BiAlertRuleDO> getRulePage(DecisionAlertRulePageReqVO pageReq) {
        return alertRuleMapper.selectPage(pageReq, new LambdaQueryWrapperX<BiAlertRuleDO>()
                .eqIfPresent(BiAlertRuleDO::getKpiId, pageReq.getKpiId())
                .eqIfPresent(BiAlertRuleDO::getEnabled, pageReq.getEnabled())
                .orderByDesc(BiAlertRuleDO::getId));
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public DecisionAlertRuleImportRespVO importAlertRuleList(List<DecisionAlertRuleImportExcelVO> importList,
                                                             DecisionAlertRuleImportReqVO importReqVO) {
        // 1. 校验非空
        importList = filterList(importList, item -> StrUtil.isNotBlank(item.getName()));
        if (CollUtil.isEmpty(importList)) {
            throw exception(ErrorCodeConstants.ALERT_RULE_IMPORT_LIST_IS_EMPTY);
        }
        // 2. 逐条处理
        DecisionAlertRuleImportRespVO respVO = DecisionAlertRuleImportRespVO.builder()
                .createAlertRuleNames(new ArrayList<>())
                .updateAlertRuleNames(new ArrayList<>())
                .failureAlertRuleNames(new LinkedHashMap<>())
                .build();
        importList.forEach(item -> {
            try {
                // 2.1 解析KPI
                BiKpiDefinitionDO kpi = StrUtil.isBlank(item.getKpiCode()) ? null
                        : kpiMapper.selectOne(BiKpiDefinitionDO::getCode, item.getKpiCode());
                if (kpi == null) {
                    throw exception(ErrorCodeConstants.ALERT_RULE_IMPORT_KPI_NOT_EXISTS, item.getKpiCode());
                }
                // 2.2 判断是否存在
                BiAlertRuleDO exist = alertRuleMapper.selectByName(item.getName());
                if (exist == null) {
                    // 2.2.1 新增
                    BiAlertRuleDO rule = BeanUtils.toBean(item, BiAlertRuleDO.class)
                            .setKpiId(kpi.getId());
                    alertRuleMapper.insert(rule);
                    respVO.getCreateAlertRuleNames().add(item.getName());
                    return;
                }
                // 2.2.2 已存在，判断是否允许更新
                if (!importReqVO.getUpdateSupport()) {
                    respVO.getFailureAlertRuleNames().put(item.getName(),
                            StrUtil.format(ErrorCodeConstants.ALERT_RULE_NAME_EXISTS.getMsg(), item.getName()));
                    return;
                }
                BiAlertRuleDO updateObj = BeanUtils.toBean(item, BiAlertRuleDO.class)
                        .setId(exist.getId())
                        .setKpiId(kpi.getId());
                alertRuleMapper.updateById(updateObj);
                respVO.getUpdateAlertRuleNames().add(item.getName());
            } catch (ServiceException ex) {
                respVO.getFailureAlertRuleNames().put(item.getName(), ex.getMessage());
            }
        });
        return respVO;
    }

}