com.ibm.wala.util.collections.IteratorPlusOne Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of com.ibm.wala.util Show documentation

T. J. Watson Libraries for Analysis

There is a newer version: 1.6.6

/*
 * Copyright (c) 2002 - 2006 IBM Corporation.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     IBM Corporation - initial API and implementation
 */
package com.ibm.wala.util.collections;

import java.util.Iterator;
import java.util.NoSuchElementException;
import org.jspecify.annotations.Nullable;

/** A utility to efficiently compose an iterator and a singleton */
public class IteratorPlusOne implements Iterator {
  public static  IteratorPlusOne make(Iterator it, T xtra) {
    if (it == null) {
      throw new IllegalArgumentException("null it");
    }
    return new IteratorPlusOne<>(it, xtra);
  }

  private final Iterator it;

  // the following field will be nulled out after visiting xtra.
  private @Nullable T xtra;

  private IteratorPlusOne(Iterator it, T xtra) {
    this.it = it;
    this.xtra = xtra;
  }

  @Override
  public boolean hasNext() {
    return it.hasNext() || (xtra != null);
  }

  @Override
  public T next() {
    if (it.hasNext()) {
      return it.next();
    } else {
      T result = xtra;
      if (result != null) {
        xtra = null;
        return result;
      } else {
        throw new NoSuchElementException();
      }
    }
  }

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