package org.dromara.demo.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import lombok.RequiredArgsConstructor;
import org.dromara.demo.domain.RsTrafficIndex;
import org.dromara.demo.domain.vo.*;
import org.dromara.demo.mapper.*;
import org.dromara.demo.service.IDataDashboardService;
import org.dromara.demo.util.PeriodDate;
import org.springframework.stereotype.Service;

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

/**
 * @author gonghl
 * @date 2024-3-8
 */
@RequiredArgsConstructor
@Service
public class DataDashboardServiceImpl implements IDataDashboardService {

    private final RsTrafficIndexMapper trafficIndexMapper;
    private final RsPcrDataMapper pcrDataMapper;
    private final RsSceneryOperationDataMapper sceneryOperationDataMapper;
    private final RsDangerInfoMapper dangerInfoMapper;
    private final RsDangerTotalMapper dangerTotalMapper;
    private final RsTrafficPropagandaMapper trafficPropagandaMapper;
    private final RsTrafficAccidentMapper trafficAccidentMapper;
    private final RsGeographyDataMapper geographyDataMapper;

    @Override
    public RsTrafficIndexVo getTrafficIndex() {
        return trafficIndexMapper.getTrafficIndex();
    }

    @Override
    public List<RsPcrDataVo> getPcrData() {
        return pcrDataMapper.getPcrData();
    }

    @Override
    public List<RsSceneryOperationDataVo> getSceneryOperationData() {
        return sceneryOperationDataMapper.getSceneryOperationData();
    }

    @Override
    public List<RsDangerInfoVo> getDangerInfo() {
        return dangerInfoMapper.getDangerInfo();
    }

    @Override
    public RsDangerTotalDashboardVo getDangerTotal(String period) {
        RsDangerTotalDashboardVo dangerTotalDashboardVo = dangerTotalMapper.getDangerTotalNums();
        String periodDate = "quarter".equals(period) ? new PeriodDate().firstdateTimSeason : new PeriodDate().MonthFirst;
        List<RsDangerTotalVo> list = dangerTotalMapper.getDangerTotalInfo(periodDate);
        dangerTotalDashboardVo.setList(list);
        return dangerTotalDashboardVo;
    }

    @Override
    public List<RsTrafficPropagandaVo> getTrafficPropaganda() {
        return trafficPropagandaMapper.getTrafficPropaganda();
    }

    @Override
    public List<RsTrafficAccidentVo> getTrafficAccident() {
        return trafficAccidentMapper.getTrafficAccident();
    }

    @Override
    public List<Map<String, Object>> getGeographyData() {
        return geographyDataMapper.getGeographyData();
    }
}