D-Net

package eu.dnetlib.msro.workflows.nodes.mdstore;

import java.util.Iterator;

import java.util.List;

import javax.xml.ws.wsaddressing.W3CEndpointReference;

import org.springframework.beans.factory.annotation.Autowired;

import eu.dnetlib.data.mdstore.MDStoreService;

import eu.dnetlib.data.mdstore.MDStoreServiceException;

import eu.dnetlib.enabling.resultset.client.ResultSetClientFactory;

import eu.dnetlib.enabling.tools.ServiceLocator;

// TODO: Auto-generated Javadoc

/**

 * The Class MultipleMdStoreIterator.

 */

public class MultipleMdStoreIterator implements Iterable<String>, Iterator<String> {

	/** The mdstore locator. */

	private ServiceLocator<MDStoreService> mdstoreLocator;

	/** The md i ds. */

	private List<String> mdIDs;

	/** The current id. */

	private String currentId = null;

	/** The current iterator. */

	private Iterator<String> currentIterator;

	/** The result set client factory. */

	@Autowired

	private ResultSetClientFactory resultSetClientFactory;

	/**

	 * Instantiates a new multiple md store iterator.

	 *

	 * @param mdstoreLocator

	 *            the mdstore locator

	 * @param mdIds

	 *            the md ids

	 */

	public MultipleMdStoreIterator(final ServiceLocator<MDStoreService> mdstoreLocator, final List<String> mdIds,

			final ResultSetClientFactory resultSetClientFactory) {

		this.mdstoreLocator = mdstoreLocator;

		this.mdIDs = mdIds;

		this.resultSetClientFactory = resultSetClientFactory;

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see java.lang.Iterable#iterator()

	 */

	@Override

	public Iterator<String> iterator() {

		getNextMDStoreRecords();

		return this;

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see java.util.Iterator#hasNext()

	 */

	@Override

	public boolean hasNext() {

		if ((currentId == null) || (currentIterator == null)) return false;

		if (!currentIterator.hasNext()) {

			getNextMDStoreRecords();

		}

		return currentIterator.hasNext();

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see java.util.Iterator#next()

	 */

	@Override

	public String next() {

		return currentIterator.next();

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see java.util.Iterator#remove()

	 */

	@Override

	public void remove() {

		currentIterator.remove();

	}

	/**

	 * Gets the next md store records.

	 *

	 * @return the next md store records

	 */

	private void getNextMDStoreRecords() {

		if (mdIDs.size() > 0) {

			currentId = mdIDs.remove(0);

			currentIterator = getIterableResultset(currentId);

		}

	}

	/**

	 * Gets the iterable resultset.

	 *

	 * @param id

	 *            the id

	 * @return the iterable resultset

	 */

	private Iterator<String> getIterableResultset(final String id) {

		try {

			W3CEndpointReference epr = mdstoreLocator.getService().deliverMDRecords(id, "", "", "");

			Iterable<String> input = resultSetClientFactory.getClient(epr);

			return input.iterator();

		} catch (MDStoreServiceException e) {

			return null;

		}

	}

}