com.happy3w.toolkits.iterator.LimitedIterator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of toolkits Show documentation

Some kits for common use

There is a newer version: 1.2.0

package com.happy3w.toolkits.iterator;

import java.util.Iterator;

public class LimitedIterator implements IEasyIterator {
    protected final Iterator innerIterator;
    protected final long maxSize;
    private long currentIndex;

    /**
     * 限制流最多输出多少条记录
     * @param innerIterator 需要限制的流
     * @param maxSize 最大值，如果最大值小于等于0，则认为不限制
     */
    public LimitedIterator(Iterator innerIterator, long maxSize) {
        this.innerIterator = innerIterator;
        this.maxSize = maxSize;
    }

    @Override
    public boolean hasNext() {
        return (maxSize <= 0 || currentIndex < maxSize) && innerIterator.hasNext();
    }

    @Override
    public T next() {
        currentIndex++;
        return innerIterator.next();
    }
}