D-Net

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

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

import eu.dnetlib.miscutils.datetime.DateUtils;

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

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

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

import eu.dnetlib.rmi.enabling.ISLookUpService;

import eu.dnetlib.rmi.manager.MSROException;

import org.apache.commons.logging.LogFactory;

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

/**

 * The Class PrepareOaiDataJobNode.

 */

public class PrepareOaiDataJobNode extends SimpleJobNode {

	/** The Constant log. */

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

	/** The service locator. */

	/** Target mongo oai store collection name. **/

	private String oaiStoreCollectionParam;

	/** Target mongo database name. **/

	private String oaiDBName;

	/** Name of the env property where to put the value of oaiDBName. **/

	private String oaiDBNameParam;

	private String formatParam = "oai_format";

	private String layoutParam = "oai_layout";

	private String interpretationParam = "oai_interpretation";

	private boolean skipDuplicates = false;

	private String duplicateXPath;

	/*

	 * (non-Javadoc)

	 *

	 * @see eu.dnetlib.msro.workflows.nodes.SimpleJobNode#execute(com.googlecode.sarasvati.NodeToken)

	 */

	@Override

	protected String execute(final Env env) throws Exception {

		log.info("start preparing job");

		// set the target mongo collection name

		final String format = env.getAttribute(getFormatParam(), String.class);

		final String interpretation = env.getAttribute(getInterpretationParam(), String.class);

		final String layout = env.getAttribute(getLayoutParam(), String.class);

		env.setAttribute(getOaiStoreCollectionParam(), format + "-" + layout + "-" + interpretation);

		final String configurationProfile = getConfigurationProfile();

		env.setAttribute("oaiConfiguration", configurationProfile);

		env.setAttribute("oai.feed.date", DateUtils.now_ISO8601());

		env.setAttribute(getOaiDBNameParam(), getOaiDBName());

		// services.publisher.oai.

		// duplicates

		env.setAttribute("services.publisher.oai.skipDuplicates", this.skipDuplicates);

		env.setAttribute("services.publisher.oai.duplicateXPath", this.duplicateXPath);

		return Arc.DEFAULT_ARC;

	}

	/**

	 * Gets the configuration profile.

	 *

	 * @return the configuration profile

	 * @throws MSROException

	 *             the MSRO exception

	 */

	private String getConfigurationProfile() throws MSROException {

		try {

			return this.serviceLocator.getService(ISLookUpService.class).getResourceProfileByQuery(

					"//RESOURCE_PROFILE[.//RESOURCE_TYPE/@value = 'OAIPublisherConfigurationDSResourceType']");

		} catch (final Exception e) {

			throw new MSROException("Cannot find OAI configuration profile", e);

		}

	}

	/**

	 * Gets the oai store collection param.

	 *

	 * @return the oai store collection param

	 */

	public String getOaiStoreCollectionParam() {

		return this.oaiStoreCollectionParam;

	}

	/**

	 * Sets the oai store collection param.

	 *

	 * @param oaiStoreCollectionParam

	 *            the new oai store collection param

	 */

	public void setOaiStoreCollectionParam(final String oaiStoreCollectionParam) {

		this.oaiStoreCollectionParam = oaiStoreCollectionParam;

	}

	public String getOaiDBName() {

		return this.oaiDBName;

	}

	public void setOaiDBName(final String oaiDBName) {

		this.oaiDBName = oaiDBName;

	}

	public String getOaiDBNameParam() {

		return this.oaiDBNameParam;

	}

	public void setOaiDBNameParam(final String oaiDBNameParam) {

		this.oaiDBNameParam = oaiDBNameParam;

	}

	public String getFormatParam() {

		return this.formatParam;

	}

	public void setFormatParam(final String formatParam) {

		this.formatParam = formatParam;

	}

	public String getLayoutParam() {

		return this.layoutParam;

	}

	public void setLayoutParam(final String layoutParam) {

		this.layoutParam = layoutParam;

	}

	public String getInterpretationParam() {

		return this.interpretationParam;

	}

	public void setInterpretationParam(final String interpretationParam) {

		this.interpretationParam = interpretationParam;

	}

	public boolean isSkipDuplicates() {

		return this.skipDuplicates;

	}

	public void setSkipDuplicates(final boolean skipDuplicates) {

		this.skipDuplicates = skipDuplicates;

	}

	public String getDuplicateXPath() {

		return this.duplicateXPath;

	}

	public void setDuplicateXPath(final String duplicateXPath) {

		this.duplicateXPath = duplicateXPath;

	}

}