package com.zhitan.engine.repository;

import com.zhitan.engine.entity.ElectricityDataItem;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.time.LocalDateTime;
import java.util.List;

/**
 * 尖峰平谷数据表数据访问层
 */
@Repository
public interface ElectricityDataItemRepository extends JpaRepository<ElectricityDataItem, String> {

    /**
     * 根据索引ID、时间编码和用电类型查询数据项
     *
     * @param indexId         索引ID
     * @param timeCode        时间编码
     * @param electricityType 用电类型
     * @return 数据项
     */
    @Query("SELECT e FROM ElectricityDataItem e WHERE e.indexId = :indexId AND e.timeCode = :timeCode AND e.electricityType = :electricityType")
    ElectricityDataItem findByIndexIdAndTimeCodeAndElectricityType(
            @Param("indexId") String indexId,
            @Param("timeCode") String timeCode,
            @Param("electricityType") String electricityType);

    /**
     * 根据索引ID和数据时间范围查询数据项
     *
     * @param indexId   索引ID
     * @param startTime 开始时间
     * @param endTime   结束时间
     * @return 数据项列表
     */
    List<ElectricityDataItem> findByIndexIdAndDataTimeBetween(String indexId, LocalDateTime startTime, LocalDateTime endTime);

    /**
     * 根据索引ID和时间类型查询数据项
     *
     * @param indexId  索引ID
     * @param timeType 时间类型
     * @return 数据项列表
     */
    List<ElectricityDataItem> findByIndexIdAndTimeType(String indexId, String timeType);
}