juneau-core/juneau-marshall/src/main/java/org/apache/juneau/http/part/BasicPartIterator.java

// ***************************************************************************************************************************
// * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements.  See the NOTICE file *
// * distributed with this work for additional information regarding copyright ownership.  The ASF licenses this file        *
// * to you under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance            *
// * with the License.  You may obtain a copy of the License at                                                              *
// *                                                                                                                         *
// *  http://www.apache.org/licenses/LICENSE-2.0                                                                             *
// *                                                                                                                         *
// * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an  *
// * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the License for the        *
// * specific language governing permissions and limitations under the License.                                              *
// ***************************************************************************************************************************
package org.apache.juneau.http.part;

import static org.apache.juneau.assertions.Assertions.*;

import java.util.NoSuchElementException;

import org.apache.http.*;
import org.apache.juneau.annotation.*;
import org.apache.juneau.internal.*;

/**
 * Basic implementation of a {@link PartIterator}.
 *
 * {@review}
 */
@NotThreadSafe
public class BasicPartIterator implements PartIterator {

	private final NameValuePair[] entries;
	private final String name;
	private final boolean caseInsensitive;

	private int currentIndex;

	/**
	 * Creates a new part iterator.
	 *
	 * @param parts An array of parts over which to iterate.
	 * @param name The name of the parts over which to iterate, or <jk>null</jk> for all.
	 * @param caseInsensitive Use case-insensitive matching for part name.
	 */
	public BasicPartIterator(NameValuePair[] parts, String name, boolean caseInsensitive) {
		this.entries = assertArgNotNull("parts", parts);
		this.name = name;
		this.caseInsensitive = caseInsensitive;
		this.currentIndex = findNext(-1);
	}

	private int findNext(int pos) {

		int from = pos;

		int to = entries.length - 1;
		boolean found = false;
		while (!found && (from < to)) {
			from++;
			found = filter(from);
		}

		return found ? from : -1;
	}

	private boolean filter(int index) {
		return (name == null) || eq(name, entries[index].getName());
	}

	@Override /* Iterator */
	public boolean hasNext() {
		return (currentIndex >= 0);
	}

	@Override /* Iterator */
	public NameValuePair next() throws NoSuchElementException {

		int current = currentIndex;

		if (current < 0)
			throw new NoSuchElementException("Iteration already finished.");

		currentIndex = findNext(current);

		return entries[current];
	}

	/**
	 * Not supported.
	 */
	@Override /* Iterator */
	public void remove() throws UnsupportedOperationException {
		throw new UnsupportedOperationException("Remove is not supported.");
	}

	private boolean eq(String s1, String s2) {
		return StringUtils.eq(caseInsensitive, s1, s2);
	}
}