package com.zhitan.model.service.impl;

import com.zhitan.common.enums.CalcType;
import com.zhitan.common.enums.TimeType;
import com.zhitan.common.utils.StringUtils;
import com.zhitan.model.domain.*;
import com.zhitan.model.mapper.IndexStorageMapper;
import com.zhitan.model.service.IEnergyIndexService;
import com.zhitan.model.service.IIndexStorageService;
import com.zhitan.model.service.IndexFormulaService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

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

import static org.apache.commons.lang3.StringUtils.equalsIgnoreCase;

/**
 * 指标存储服务实现类
 */
@Service
public class IndexStorageServiceImpl implements IIndexStorageService {

    @Autowired
    private IndexStorageMapper indexStorageMapper;
    @Autowired
    private IndexFormulaService indexFormulaService;
    @Autowired
    private IEnergyIndexService energyIndexService;

    /**
     * 保存指标存储信息
     * @param indexId     指标ID
     * @param indexStorage 指标存储列表
     */
    @Override
    public void saveIndexStorage(String indexId, List<IndexStorage> indexStorage) {
        // 遍历每个存储对象进行存储操作
        indexStorage.forEach(storage -> {
            storage.setIndexId(indexId); // 设置指标ID
            if (StringUtils.isEmpty(storage.getId())) { // 新增记录
                storage.setId(UUID.randomUUID().toString());
                indexStorageMapper.insertIndexStorage(storage);
            } else { // 更新记录
                indexStorageMapper.updateIndexStorage(storage);
            }
            // 保存参数关联关系
            indexStorageMapper.saveParams(storage.getId(), storage.getParamIndex());
        });
    }

    /**
     * 保存公式及存储信息（事务方法）
     * @param indexFormula 公式对象
     * @param indexStorage 存储列表
     * @param indexId      指标ID
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void saveFormulaAndStorage(IndexFormula indexFormula, List<IndexStorage> indexStorage,
                                     String indexId) {
        // 1. 保存公式信息
        indexFormulaService.saveIndexFormula(indexFormula);

        // 2. 处理公式文本参数替换
        String calcText = indexFormula.getFormulaText();
        List<String> paramNames = new ArrayList<>();
        for (IndexFormulaParam param : indexFormula.getIndexFormulaParams()) {
            calcText = calcText
                    .replace("(" + param.getParamName() + ")", "(" + String.format("'%s'", param.getParamValue()) + ")");
            paramNames.add(param.getParamValue()); // 记录参数名称
        }

        // 3. 获取参数指标
        List<EnergyIndex> energyIndexList = energyIndexService.getEnergyIndexByCodes(paramNames);
        List<String> paramIndex = energyIndexList.stream()
                .map(EnergyIndex::getIndexId)
                .collect(Collectors.toList());

        // 4. 补充存储对象信息
        for (IndexStorage storage : indexStorage) {
            storage.setCalcText(calcText);          // 存储计算公式文本
            storage.setIsPvCalc(indexFormula.getIsPvCalc()); // 存储实时计算标识
            storage.getParamIndex().addAll(paramIndex); // 添加参数指标ID
        }

        // 5. 保存存储信息
        saveIndexStorage(indexId, indexStorage);
    }

    /**
     * 根据指标ID获取存储信息
     * @param indexId 指标ID
     * @return 指标存储列表
     */
    @Override
    public List<IndexStorage> getIndexStorage(String indexId) {
        return indexStorageMapper.getIndexStorage(indexId);
    }

    /**
     * 获取所有计算类存储信息
     * @return 包含参数的存储列表
     */
    @Override
    public List<IndexStorage> getAllCalcIndexStorage() {
        // 1. 查询计算类型存储
        List<IndexStorage> indexStorageList = indexStorageMapper.getAllCalcIndexStorage(CalcType.CALC);

        // 2. 查询所有参数信息
        List<IndexStorageParam> params = indexStorageMapper.getAllParameter();

        // 3. 将参数信息合并到存储对象
        indexStorageList.forEach(storage -> {
            List<String> indexIds = params.stream()
                    .filter(f -> equalsIgnoreCase(f.getStorageId(), storage.getId()))
                    .map(IndexStorageParam::getIndexId)
                    .collect(Collectors.toList());
            if (!indexIds.isEmpty()) {
                storage.getParamIndex().addAll(indexIds);
            }
        });
        return indexStorageList;
    }

    /**
     * 根据指标ID和时间类型获取特定存储
     * @param indexId    指标ID
     * @param timeType   时间类型
     * @return 指定时间类型的存储信息
     */
    @Override
    public IndexStorage getIndexStorage(String indexId, TimeType timeType) {
        return indexStorageMapper.getWithTimetype(indexId, timeType);
    }
}