com.rockset.jdbc.LengthLimitedIterator Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of rockset-java Show documentation

The official Rockset Java client library

There is a newer version: 0.13.0

package com.rockset.jdbc;

import static com.google.common.base.Preconditions.checkArgument;
import static java.util.Objects.requireNonNull;

import java.util.Iterator;
import java.util.NoSuchElementException;

/**
 * This {@code Iterator} is like Guava's {@code Iterators.limit()} but
 * uses a {@code long} limit instead of {@code int}.
 */
final class LengthLimitedIterator implements Iterator {
  private final Iterator iterator;
  private final long limit;
  private long count;

  public LengthLimitedIterator(Iterator iterator, long limit) {
    checkArgument(limit >= 0, "limit is negative");
    this.iterator = requireNonNull(iterator, "iterator is null");
    this.limit = limit;
  }

  @Override
  public boolean hasNext() {
    return count < limit && iterator.hasNext();
  }

  @Override
  public T next() {
    if (!hasNext()) {
      throw new NoSuchElementException();
    }
    count++;
    return iterator.next();
  }
}