org.archive.util.iterator.BoundedStringIterator Maven / Gradle / Ivy

Go to download

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

The newest version!

package org.archive.util.iterator;

import java.io.IOException;
import java.util.Iterator;

public class BoundedStringIterator extends AbstractPeekableIterator 
	implements CloseableIterator {

	private Iterator inner;
	private String boundary;
	private boolean inclusive;
	private int flip;

	public BoundedStringIterator(Iterator inner, String boundary) {
		this(inner, boundary, false);
	}
	
	public BoundedStringIterator(Iterator inner, String boundary, boolean inclusive) {
		this(inner, boundary, inclusive, false);
	}
	
	public BoundedStringIterator(Iterator inner, String boundary, boolean inclusive, boolean reverse) {
		this.inner = inner;
		this.boundary = boundary;
		this.inclusive = inclusive;
		this.flip = (reverse ? -1 : 1);
	}

	@Override
	public String getNextInner() {
		String tmp = null;
		if(inner.hasNext()) {
			tmp = inner.next();
			if(tmp.compareTo(boundary) * flip >= 0 && (!inclusive || !tmp.startsWith(boundary))) {
				tmp = null;
				try {
					close();
				} catch (IOException e) {
					throw new RuntimeException(e);
				}
			}
		}
		return tmp;
	}

	public void close() throws IOException {
		CloseableIteratorUtil.attemptClose(inner);
	}

}