refactor(zhitan):调整模型代码和优化能源统计功能

baoshiwei

2025-05-06 8dd7af51db4cf768fbf92002b5b62f7fb9d1c929

 zhitan-system/src/main/java/com/zhitan/model/service/impl/IndexStorageServiceImpl.java

@@ -12,89 +12,127 @@
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
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;
    @Autowired
    private IndexStorageMapper indexStorageMapper;
    @Autowired
    private IndexFormulaService indexFormulaService;
    @Autowired
    private IEnergyIndexService energyIndexService;

  @Override
  public void saveIndexStorage(String indexId, List<IndexStorage> indexStorage) {
    indexStorage.forEach(storage -> {
      storage.setIndexId(indexId);
      if (StringUtils.isEmpty(storage.getId())) {
        storage.setId(UUID.randomUUID().toString());
        indexStorageMapper.insertIndexStorage(storage);
      } else {
        indexStorageMapper.updateIndexStorage(storage);
      }

      indexStorageMapper.saveParams(storage.getId(), storage.getParamIndex());
    });
  }

  @Override
  @Transactional(rollbackFor = Exception.class)
  public void saveFormulaAndStorage(IndexFormula indexFormula, List<IndexStorage> indexStorage,
                                    String indexId) {
    indexFormulaService.saveIndexFormula(indexFormula);
    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());
    /**
     * 保存指标存储信息
     * @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());
        });
    }

    List<EnergyIndex> energyIndexList = energyIndexService.getEnergyIndexByCodes(paramNames);
    List<String> paramIndex = energyIndexList.stream()
        .map(EnergyIndex::getIndexId)
        .collect(Collectors.toList());
    for (IndexStorage storage : indexStorage) {
      storage.setCalcText(calcText);
      storage.setIsPvCalc(indexFormula.getIsPvCalc());
      storage.getParamIndex().addAll(paramIndex);
    /**
     * 保存公式及存储信息（事务方法）
     * @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);
    }

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

  @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);

  @Override
  public List<IndexStorage> getAllCalcIndexStorage() {
    List<IndexStorage> indexStorageList = indexStorageMapper.getAllCalcIndexStorage(CalcType.CALC);
    List<IndexStorageParam> params = indexStorageMapper.getAllParameter();
    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);
      }
    });
        // 2. 查询所有参数信息
        List<IndexStorageParam> params = indexStorageMapper.getAllParameter();

    return indexStorageList;
  }
        // 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;
    }

  @Override
  public IndexStorage getIndexStorage(String indexId, TimeType timeType) {
    return indexStorageMapper.getWithTimetype(indexId, timeType);
  }

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