joinediterator.java

来自「人力资源管理系统主要包括：人员管理、招聘管理、培训管理、奖惩管理和薪金管理五大管」· Java 代码 · 共 79 行

//$Id: JoinedIterator.java,v 1.3.2.2 2004/01/29 02:19:33 oneovthafew Exp $
package net.sf.hibernate.util;

import java.util.Iterator;

import org.apache.commons.collections.IteratorUtils;

/**
 * An JoinedIterator is an Iterator that wraps a number of Iterators. 
 * 
 * This class makes multiple iterators look like one to the caller. 
 * When any method from the Iterator interface is called, the JoinedIterator
 * will delegate to a single underlying Iterator. The JoinedIterator will 
 * invoke the Iterators in sequence until all Iterators are exhausted. 
 * 
 */
public class JoinedIterator implements Iterator {
    // wrapped iterators
	private Iterator[] iterators;
	
	// index of current iterator in the wrapped iterators array
	private int currentIteratorIndex;
	
	// the current iterator
	private Iterator currentIterator;
	
	// the last used iterator
	private Iterator lastUsedIterator;
	

	public JoinedIterator(Iterator[] iterators) {
	    if( iterators==null )
	        throw new NullPointerException("Unexpected NULL iterators argument");
		this.iterators = iterators;
	}
	
	public JoinedIterator(Iterator first, Iterator second) {
		this( new Iterator[] { first, second } );
	}
	
	public boolean hasNext() {
	    updateCurrentIterator();
	    return currentIterator.hasNext();
	}
	
	public Object next() {
	    updateCurrentIterator();
	    return currentIterator.next();
	}
	
	public void remove() {
	    updateCurrentIterator();
		lastUsedIterator.remove();
	}
	
	
	// call this before any Iterator method to make sure that the current Iterator
	// is not exhausted
	protected void updateCurrentIterator() {
	    
	    if (currentIterator == null) {
	        if( iterators.length==0  ) {
	            currentIterator = IteratorUtils.EMPTY_ITERATOR;
	        } 
	        else {
	            currentIterator = iterators[0];
	        }
	        // set last used iterator here, in case the user calls remove
	        // before calling hasNext() or next() (although they shouldn't)
	        lastUsedIterator = currentIterator;
	    }

	    while (currentIterator.hasNext() == false && currentIteratorIndex < iterators.length - 1) {
	        currentIteratorIndex++;
	        currentIterator = iterators[currentIteratorIndex];
	    }
	}
}