package com.ruoyi.ems.mapper;

import com.ruoyi.ems.domain.ElecConsumeForecast;
import org.apache.ibatis.annotations.Param;

import java.util.List;
import java.util.Map;

/**
 * 电力消耗预测Mapper接口
 *
 * @author ruoyi
 * @date 2024-08-12
 */
public interface ElecConsumeForecastMapper {

    /**
     * 查询电力消耗预测
     *
     * @param id 电力消耗预测主键
     * @return 电力消耗预测
     */
    ElecConsumeForecast selectForecastById(Long id);

    /**
     * 查询电力消耗预测列表（支持层级汇总）
     * 当选择父节点时，自动包含所有子节点数据
     *
     * @param forecast 电力消耗预测查询条件
     * @return 电力消耗预测集合
     */
    List<ElecConsumeForecast> selectForecastList(ElecConsumeForecast forecast);

    /**
     * 新增电力消耗预测
     *
     * @param forecast 电力消耗预测
     * @return 结果
     */
    int insertForecast(ElecConsumeForecast forecast);

    /**
     * 批量新增电力消耗预测
     *
     * @param list 电力消耗预测列表
     * @return 结果
     */
    int insertBatch(List<ElecConsumeForecast> list);

    /**
     * 修改电力消耗预测
     *
     * @param forecast 电力消耗预测
     * @return 结果
     */
    int updateForecast(ElecConsumeForecast forecast);

    /**
     * 删除电力消耗预测
     *
     * @param id 电力消耗预测主键
     * @return 结果
     */
    int deleteForecastById(Long id);

    /**
     * 按条件删除电力消耗预测
     *
     * @param areaCode  区域代码
     * @param objType   对象类型
     * @param objCode   对象代码
     * @param startTime 开始时间
     * @param endTime   结束时间
     * @return 结果
     */
    int deleteByCondition(@Param("areaCode") String areaCode,
        @Param("objType") int objType,
        @Param("objCode") String objCode,
        @Param("startTime") String startTime,
        @Param("endTime") String endTime);

    /**
     * 批量删除电力消耗预测
     *
     * @param ids 需要删除的数据主键集合
     * @return 结果
     */
    int deleteForecastByIds(Long[] ids);

    /**
     * 查询每日汇总趋势数据（支持层级汇总）
     * 当选择父节点时，自动汇总所有子节点数据
     *
     * @param forecast 查询条件
     * @return 每日汇总数据列表，包含date和elecUseQuantity
     */
    List<ElecConsumeForecast> calcForecastDateRange(ElecConsumeForecast forecast);

    /**
     * 查询预测汇总统计（支持层级汇总）
     * 返回：
     * - totalElec: 总用电量
     * - avgElec: 平均用电量
     * - itemCount: 子项数量（去重）
     * - dayCount: 天数（去重）
     * - startDate: 最小日期
     * - endDate: 最大日期
     *
     * @param forecast 查询条件
     * @return 汇总统计结果Map
     */
    Map<String, Object> selectForecastSummary(ElecConsumeForecast forecast);
}