net.sf.staccatocommons.collections.stream.internal.algorithms.AppendIterableStream Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of commons-collections Show documentation

Collections library of the Staccato-Commons project, focused on providing new abstractions that mix object oriented and functional programming style for dealing with iterable objects.

The newest version!

/**
 *  Copyright (c) 2011, The Staccato-Commons Team
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation; version 3 of the License.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 */

package net.sf.staccatocommons.collections.stream.internal.algorithms;

import java.util.NoSuchElementException;

import net.sf.staccatocommons.collections.stream.AbstractStream;
import net.sf.staccatocommons.collections.stream.Stream;
import net.sf.staccatocommons.defs.Thunk;
import net.sf.staccatocommons.iterators.thriter.AdvanceThriterator;
import net.sf.staccatocommons.iterators.thriter.Thriter;
import net.sf.staccatocommons.iterators.thriter.Thriterator;
import net.sf.staccatocommons.iterators.thriter.Thriterators;
import net.sf.staccatocommons.restrictions.check.NonNull;

/**
 * @author flbulgarelli
 * 
 */
public final class AppendIterableStream extends AbstractStream {
  private final Stream stream;
  private final Iterable other;

  /**
   * Creates a new {@link AppendIterableStream}
   */
  public AppendIterableStream(@NonNull Stream stream, @NonNull Iterable other) {
    this.stream = stream;
    this.other = other;
  }

  @Override
  public Thriterator iterator() { // FIXME bad impl
    return new AdvanceThriterator() {
      private Thriter iter = stream.iterator();
      private boolean second = false;

      public boolean hasNext() {
        if (iter.hasNext())
          return true;

        if (second)
          return false;

        iter = Thriterators.from(other.iterator());
        second = true;
        return iter.hasNext();
      }

      public void advanceNext() throws NoSuchElementException {
        iter.advanceNext();
      }

      public A current() throws NoSuchElementException {
        return iter.current();
      }

      @Override
      public Thunk delayedCurrent() {
        return iter.delayedCurrent();
      }
    };
  }
}