package com.its.api.its.service.ifsc;

import com.its.api.its.model.dto.analysis.TbTraf15mStatCountDto;
import com.its.api.its.model.dto.analysis.TbTraf15mStatCountInf;
import com.its.api.its.model.dto.analysis.TbTraf15mStatListDto;
import com.its.api.its.model.dto.analysis.TbTraf15mStatListInf;
import com.its.api.its.model.dto.ifsc.TbIfsc15mStatDto;
import com.its.api.its.model.entity.ifsc.TbIfsc15mStat;
import com.its.api.its.model.entity.ifsc.TbIfsc15mStatKey;
import com.its.api.its.repository.ifsc.TbIfsc15mStatRepository;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;

@Slf4j
@RequiredArgsConstructor
@Service
public class TbIfsc15mStatService {

    private final TbIfsc15mStatRepository repo;

    // 데이터 1건 조회, 없으면 exception
    private TbIfsc15mStat requireOne(TbIfsc15mStatKey id) {
        return this.repo.findById(id)
                .orElseThrow(() -> new NoSuchElementException("데이터가 존재하지 않습니다: " + id));
    }

    // 전체 데이터 조회
    @Transactional(readOnly = true)
    public List<TbIfsc15mStatDto> findAll() {
        List<TbIfsc15mStatDto> result = new ArrayList<>();
        List<TbIfsc15mStat> data = this.repo.findAll();
        for (TbIfsc15mStat entity : data) {
            result.add(entity.toDto());
        }
        return result;
    }

    // 데이터 1건 조회(기존 데이터가 반드시 존재해야 함)
    @Transactional(readOnly = true)
    public TbIfsc15mStatDto findById(TbIfsc15mStatKey id) {
        TbIfsc15mStat entity = requireOne(id);
        return entity.toDto();
    }

    /**
     * 통계/분석: 소통정보 이력분석-데이터 건수 조회
     * @param fromDt
     * @param toDt
     * @return
     */
    public List<TbTraf15mStatCountDto> findAllCountByDateRange(String fromDt, String toDt) {

        List<TbTraf15mStatCountDto> result = new ArrayList<>();
        List<TbTraf15mStatCountInf> data = this.repo.findAllCountByDateRange(fromDt, toDt);
        if (data != null) {
            data.forEach(obj -> {
                result.add(TbTraf15mStatCountDto.builder()
                        .statDt(obj.getStatDt())
                        .dataCnt(obj.getDataCnt().intValue())
                        .build());
            });
        }
        return result;
    }

    /**
     * 통계/분석: 소통정보 이력분석-이력 데이터 조회
     * @param statDt
     * @return
     */
    public List<TbTraf15mStatListDto> findAllDataByDateRange(String statDt) {

        List<TbTraf15mStatListDto> result = new ArrayList<>();
        List<TbTraf15mStatListInf> data = this.repo.findAllDataByDateRange(statDt);
        if (data != null) {
            data.forEach(obj -> {
                result.add(TbTraf15mStatListDto.builder()
                        .id(obj.getId().longValue())
                        .cmtrGradCd(obj.getCmtrGradCd())
                        .sped(obj.getSped().intValue())
                        .build());
            });
        }
        return result;
    }
}
