cucumber.runtime.io.FlatteningIterator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cucumber-core Show documentation

There is a newer version: 1.2.6

package cucumber.runtime.io;

import java.util.ArrayDeque;
import java.util.Deque;
import java.util.Iterator;
import java.util.NoSuchElementException;

public class FlatteningIterator implements Iterator {
    private final Deque> iterators = new ArrayDeque>();

    private Object next;
    private boolean nextBlank = true;

    public void push(Iterator iterator) {
        iterators.addFirst(iterator);
    }

    @Override
    public void remove() {
        throw new UnsupportedOperationException();
    }

    private void moveToNext() {
        if (nextBlank && !this.iterators.isEmpty()) {
            if (!iterators.peek().hasNext()) {
                iterators.removeFirst();
                moveToNext();
            } else {
                final Object next = iterators.peekFirst().next();
                if (next instanceof Iterator) {
                    push((Iterator) next);
                    moveToNext();
                } else {
                    this.next = next;
                    nextBlank = false;
                }
            }
        }
    }

    @Override
    public Object next() {
        moveToNext();

        if (nextBlank) {
            throw new NoSuchElementException();
        } else {
            Object next = this.next;
            this.next = null;
            nextBlank = true;
            return next;
        }
    }

    @Override
    public boolean hasNext() {
        moveToNext();
        return !nextBlank;
    }
}