D-Net

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

import java.util.Map;

import eu.dnetlib.data.mdstore.MDStoreService;

import eu.dnetlib.enabling.resultset.rmi.ResultSetException;

import eu.dnetlib.enabling.tools.blackboard.BlackboardJob;

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

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

import eu.dnetlib.msro.workflows.nodes.blackboard.BlackboardWorkflowJobListener;

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

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

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

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

import eu.dnetlib.msro.workflows.resultset.ProcessCountingResultSetFactory;

import eu.dnetlib.msro.workflows.util.ProgressProvider;

import eu.dnetlib.msro.workflows.util.ResultsetProgressProvider;

import eu.dnetlib.msro.workflows.util.WorkflowsConstants;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

public class StoreMDStoreRecordsJobNode extends BlackboardJobNode implements ProgressJobNode, ProcessAware {

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

	private String eprParam;

	private String mdId;

	private String storingType;

	private ProcessCountingResultSetFactory processCountingResultSetFactory;

	private ResultsetProgressProvider progressProvider;

	private WorkflowProcess process;

	@Override

	protected String obtainServiceId(final Env env) {

		return getServiceLocator().getServiceId(MDStoreService.class, mdId);

	}

	@Override

	protected void prepareJob(final BlackboardJob job, final Env env) throws ResultSetException {

		job.setAction("FEED");

		final String eprS = env.getAttribute(getEprParam(), String.class);

		this.progressProvider = processCountingResultSetFactory.createProgressProvider(process, eprS);

		job.getParameters().put("epr", progressProvider.getEpr().toString());

		job.getParameters().put("storingType", getStoringType());

		job.getParameters().put("mdId", getMdId());

	}

	@Override

	protected BlackboardWorkflowJobListener generateBlackboardListener(final Token token) {

		return new BlackboardWorkflowJobListener(token) {

			@Override

			protected void responseToEnv(final Env env, final Map<String, String> responseParams) {

				log.info("Number of stored records: " + responseParams.get("total"));

				env.setAttribute(WorkflowsConstants.MAIN_LOG_PREFIX + "total", responseParams.get("total"));

			}

		};

	}

	public String getEprParam() {

		return eprParam;

	}

	public void setEprParam(final String eprParam) {

		this.eprParam = eprParam;

	}

	public String getMdId() {

		return mdId;

	}

	public void setMdId(final String mdId) {

		this.mdId = mdId;

	}

	public String getStoringType() {

		return storingType;

	}

	public void setStoringType(final String storingType) {

		this.storingType = storingType;

	}

	public ProcessCountingResultSetFactory getProcessCountingResultSetFactory() {

		return processCountingResultSetFactory;

	}

	@Required

	public void setProcessCountingResultSetFactory(final ProcessCountingResultSetFactory processCountingResultSetFactory) {

		this.processCountingResultSetFactory = processCountingResultSetFactory;

	}

	@Override

	public ProgressProvider getProgressProvider() {

		return progressProvider;

	}

	@Override

	public void setProcess(final WorkflowProcess process) {

		this.process = process;

	}

}