package com.ld.igds.view;

import com.bstek.dorado.annotation.DataProvider;
import com.bstek.dorado.annotation.Expose;
import com.ld.igds.data.BaseParam;
import com.ld.igds.databoard.service.HGrainAccountService;
import org.apache.commons.lang3.time.DateUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

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

    @Autowired
    private HGrainAccountService grainAccountService;

    /**
     * grainAccountPR#getQuery
     * @return
     */
    @DataProvider
    public Map<String, Object> getQuery() {
        Map<String, Object> result = new HashMap<>();

        result.put("end", new Date());
        Date start = DateUtils.addMonths(new Date(),-1);
        result.put("start", start);

        return result;
    }

    @Expose
    public Map<String,Object> getData(String depotId, Date start, Date end){
        BaseParam param = new BaseParam();
        param.setDepotId(depotId);
        param.setStart(start);
        param.setEnd(end);
        return grainAccountService.getData(param);
    }

}