D-Net

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

import java.util.List;

import javax.annotation.Resource;

import javax.xml.ws.wsaddressing.W3CEndpointReference;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

import com.google.gson.Gson;

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

import eu.dnetlib.msro.workflows.graph.Arc;

import eu.dnetlib.msro.workflows.nodes.SimpleJobNode;

import eu.dnetlib.msro.workflows.procs.Env;

public class FetchMultipleMDStores extends SimpleJobNode {

	private static final Log log = LogFactory.getLog(FetchMultipleMDStores.class); // NOPMD by marko on 11/24/08 5:02 PM

	@Resource

	private UniqueServiceLocator serviceLocator;

	private List<String> mdId;

	private String eprParam;

	/**

	 * The result set factory.

	 */

	@Resource(name = "iterableResultSetFactory")

	private IterableResultSetFactory resultSetFactory;

	/**

	 * The result set client factory.

	 */

	@Autowired

	private ResultSetClientFactory resultSetClientFactory;

	@Override

	protected String execute(final Env env) throws Exception {

		if (getMdId() == null) {

			final String mdIdsJson = env.getAttribute("mdId", String.class);

			@SuppressWarnings("unchecked")

			final List<String> mdIds = new Gson().fromJson(mdIdsJson, List.class);

			log.info(String.format("fetching records for %s mdstores", mdIds.size()));

			if (log.isDebugEnabled()) {

				log.debug(String.format("fetching records for mdstores: %s", mdIdsJson));

			}

			setMdId(mdIds);

		}

		final MultipleMdStoreIterable iter = new MultipleMdStoreIterable(this.serviceLocator, getMdId(), this.resultSetClientFactory);

		final W3CEndpointReference epr = this.resultSetFactory.createIterableResultSet(iter);

		env.setAttribute(getEprParam(), epr.toString());

		return Arc.DEFAULT_ARC;

	}

	public List<String> getMdId() {

		return this.mdId;

	}

	public void setMdId(final List<String> mdId) {

		this.mdId = mdId;

	}

	public String getEprParam() {

		return this.eprParam;

	}

	public void setEprParam(final String eprParam) {

		this.eprParam = eprParam;

	}

}