package com.qianwen.smartman.modules.mdc.service.impl;

import cn.hutool.core.date.LocalDateTimeUtil;
import java.time.LocalDate;
import java.util.List;
import com.qianwen.smartman.common.constant.DateConstant;
import com.qianwen.smartman.modules.mdc.entity.SuperAggregateOutput;
import com.qianwen.smartman.modules.mdc.enums.StatisticalMethodEnum;
import com.qianwen.smartman.modules.mdc.mapper.SuperAggregateOutputMapper;
import com.qianwen.smartman.modules.mdc.service.ISuperAggregateOutputService;
import org.springframework.stereotype.Service;

@Service
public class SuperAggregateOutputServiceImpl implements ISuperAggregateOutputService {
    private final SuperAggregateOutputMapper baseMapper;

    public SuperAggregateOutputServiceImpl(final SuperAggregateOutputMapper baseMapper) {
        this.baseMapper = baseMapper;
    }

    @Override
    public List<SuperAggregateOutput> getOutputData(List<Long> workstationIds, StatisticalMethodEnum statisticalMethodEnum, LocalDate startDate, LocalDate endDate) {
        List<SuperAggregateOutput> superAggregateOutputList;
        if (StatisticalMethodEnum.SHIFT.equals(statisticalMethodEnum) || StatisticalMethodEnum.DAY.equals(statisticalMethodEnum) || StatisticalMethodEnum.WEEK.equals(statisticalMethodEnum) || StatisticalMethodEnum.MONTH.equals(statisticalMethodEnum)) {
            superAggregateOutputList = this.baseMapper.getOutputDataByFactoryDate(workstationIds, LocalDateTimeUtil.format(startDate, "yyyyMMdd"), LocalDateTimeUtil.format(endDate, "yyyyMMdd"));
        } else {
            superAggregateOutputList = this.baseMapper.getOutputDataByNaturalDate(workstationIds, LocalDateTimeUtil.format(startDate, DateConstant.PATTERN_DATE_TIME), LocalDateTimeUtil.format(endDate.plusDays(1L), DateConstant.PATTERN_DATE_TIME));
        }
        return superAggregateOutputList;
    }

    @Override
    public List<SuperAggregateOutput> queryPerfByDay(Long workstationId, String startTime, String endTime, Long employeeId) {
        return this.baseMapper.queryPerfByDay(workstationId, startTime, endTime, employeeId);
    }

    @Override
    public List<SuperAggregateOutput> queryPerfByWeek(Long workstationId, List<Integer> weeks, Long employeeId) {
        return this.baseMapper.queryPerfByWeek(workstationId, weeks, employeeId);
    }

    @Override
    public List<SuperAggregateOutput> queryPerfByMonth(Long workstationId, List<Integer> months, Long employeeId) {
        return this.baseMapper.queryPerfByMonth(workstationId, months, employeeId);
    }

    @Override
    public List<SuperAggregateOutput> queryPerfByEtDay(Long employeeId, String time) {
        return this.baseMapper.queryPerfByEtDay(employeeId, time);
    }

    @Override
    public List<SuperAggregateOutput> queryPerfByEtWeek(Long employeeId, Integer week) {
        return this.baseMapper.queryPerfByEtWeek(employeeId, week);
    }

    @Override
    public List<SuperAggregateOutput> queryPerfByEtMonth(Long employeeId, Integer month) {
        return this.baseMapper.queryPerfByEtMonth(employeeId, month);
    }

    @Override
    public List<SuperAggregateOutput> queryPerfByWtDay(Long workstationId, String time) {
        return this.baseMapper.queryPerfByWtDay(workstationId, time);
    }

    @Override
    public List<SuperAggregateOutput> queryPerfByWtWeek(Long workstationId, Integer week) {
        return this.baseMapper.queryPerfByWtWeek(workstationId, week);
    }

    @Override
    public List<SuperAggregateOutput> queryPerfByWtMonth(Long workstationId, Integer month) {
        return this.baseMapper.queryPerfByWtMonth(workstationId, month);
    }
}