// This Source Code is in the Public Domain per: http://unlicense.org
package org.litesoft.GWT.client.widgets.datatables;

import java.util.*;

/**
 * A proxy that enables us to limit the number of rows returned to the specified
 * number.
 */
public class LimitingIteratorWrapper<RowType> implements Iterator<RowType> {
    private final Iterator<RowType> mImpl;
    private final int mMaxRows;

    int mCount = 0;

    public LimitingIteratorWrapper( Iterator<RowType> pWrapped, int pMaxRows ) {
        mImpl = pWrapped;
        mMaxRows = pMaxRows;
    }

    @Override
    public boolean hasNext() {
        return mCount < mMaxRows && mImpl.hasNext();
    }

    @Override
    public RowType next() {
        assert mCount < mMaxRows;
        mCount++;
        return mImpl.next();
    }

    @Override
    public void remove() {
        mImpl.remove();
    }

    @Override
    public String toString() {
        return "[maxRows=" + mMaxRows + ",iter=" + mImpl + "]";
    }
}