package com.ld.igds.databoard.service;

import com.ld.igds.data.BaseParam;
import com.ld.igds.es.dto.EsData;
import com.ld.igds.es.dto.EsParam;
import com.ld.igds.es.dto.EsSummary;
import com.ld.igds.es.mapper.EsServiceMapper;
import com.ld.igds.es.service.CoreEsService;
import com.ld.igds.gas.dto.GasData;
import com.ld.igds.grain.dto.GrainData;
import com.ld.igds.m.mapper.ManagerMapper;
import com.ld.igds.pest.dto.PestData;
import com.ld.igds.util.ContextUtil;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author he
 */
@Component
public class HGrainAccountService {

    @Autowired
    private ManagerMapper managerMapper;

    @Autowired
    private EsServiceMapper esMapper;

    @Autowired
    private CoreEsService coreEsService;

    public Map<String,Object> getData(BaseParam param){
        if(null==param){
            return null;
        }
        if(StringUtils.isEmpty(param.getDepotId())){
            return null;
        }
        if(StringUtils.isEmpty(param.getCompanyId())){
            param.setCompanyId(ContextUtil.getCompanyId());
        }

        Map<String, Object> info = new HashMap<>();

        info.put("grainList",getGrain(param));
        info.put("esList",getEnergy(param));
        info.put("esSummary",getEsSummary(param));
        info.put("pestList",getPest(param));
        info.put("gasList",getGas(param));

        return info;
    }

    public List<GrainData> getGrain(BaseParam param){
        List<GrainData> grainList = managerMapper.getGrainByDate(param);
        List<GrainData> list = new ArrayList<>();
        if(grainList!=null && grainList.size()>0){
            list.add(grainList.get(grainList.size()-1));
            list.add(grainList.get(0));
        }
        return list;
    }

    public List<EsData> getEnergy(BaseParam param){
        List<EsData> esList = esMapper.getEnergyByDate(param);
        List<EsData> list = new ArrayList<>();
        if(esList!=null && esList.size()>0){
            list.add(esList.get(esList.size()-1));
            list.add(esList.get(0));
        }
        return list;
    }

    public EsSummary getEsSummary(BaseParam param){
        EsParam esParam = new EsParam();
        esParam.setDepotId(param.getDepotId());
        esParam.setCompanyId(param.getCompanyId());
        esParam.setStart(param.getStart());
        esParam.setEnd(param.getEnd());

        EsSummary sum = coreEsService.getSummaryByParam(esParam);

        return sum;
    }

    public List<PestData> getPest(BaseParam param){
        List<PestData> pestList = managerMapper.getPestByDate(param);
        List<PestData> list = new ArrayList<>();
        if(pestList!=null && pestList.size()>0){
            list.add(pestList.get(pestList.size()-1));
            list.add(pestList.get(0));
        }
        return list;
    }

    public List<GasData> getGas(BaseParam param){
        List<GasData> gasList = managerMapper.getGasByDate(param);
        List<GasData> list = new ArrayList<>();
        if(gasList!=null && gasList.size()>0){
            list.add(gasList.get(gasList.size()-1));
            list.add(gasList.get(0));
        }
        return list;
    }
}