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.springframework.beans.factory.annotation.Autowired;

import com.google.gson.Gson;

import com.googlecode.sarasvati.Arc;

import com.googlecode.sarasvati.NodeToken;

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

import eu.dnetlib.enabling.resultset.IterableResultSetFactory;

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

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

public class FetchMultipleMDStores extends SimpleJobNode {

	@Resource

	private UniqueServiceLocator serviceLocator;

	private List<String> mdId;

	private String mdFormat;

	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 NodeToken token) throws Exception {

		if (getMdId() == null) {

			@SuppressWarnings("unchecked")

			List<String> mdIds = new Gson().fromJson(token.getEnv().getAttribute("mdId"), List.class);

			setMdId(mdIds);

		}

		if (getMdFormat() == null) {

			setMdFormat(token.getEnv().getAttribute("mdFormat"));

		}

		MultipleMdStoreIterable iter = new MultipleMdStoreIterable(serviceLocator, getMdId(), resultSetClientFactory);

		W3CEndpointReference eprUrls = resultSetFactory.createIterableResultSet(iter);

		token.getEnv().setAttribute(getEprParam(), eprUrls.toString());

		return Arc.DEFAULT_ARC;

	}

	public List<String> getMdId() {

		return mdId;

	}

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

		this.mdId = mdId;

	}

	public String getMdFormat() {

		return mdFormat;

	}

	public void setMdFormat(final String mdFormat) {

		this.mdFormat = mdFormat;

	}

	public String getEprParam() {

		return eprParam;

	}

	public void setEprParam(final String eprParam) {

		this.eprParam = eprParam;

	}

}