package ru.bulbasavrteam.cachbot.TradeDataStreams;

import ru.bulbasavrteam.cachbot.DataStreaming.IStream;

import java.util.Date;

public class TradeDateFilterStream implements IStream<TradeData>
{
    private IStream<TradeData> _stream;

    private Date _date1;
    private Date _date2;

    public TradeDateFilterStream(IStream<TradeData> stream)
    {
        _stream = stream;
        _date1 = null;
        _date2 = null;
    }

    public TradeDateFilterStream SetAsset(Date date1,Date date2)
    {
        _date1 = date1;
        _date2 = date2;
        return this;
    }


    @Override
    public boolean IsEmpty()
    {
        return _stream.IsEmpty();
    }

    @Override
    public TradeData Next()
    {
        TradeData data;
        if (_date1 == null) {
            return _stream.Next();
        }

        if (_date2 == null) {
            return _stream.Next();
        }

        while (null != (data = _stream.Next()))
            if (_date1.before(data.date))
            {
                break;
            } else if (_date2.after(data.date)){
                break;
            }


        return data;
    }
}