D-Net

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

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import com.googlecode.sarasvati.NodeToken;

import com.googlecode.sarasvati.env.Env;

import eu.dnetlib.msro.worker.WorkflowConstants;

import eu.dnetlib.msro.worker.nodes.ProgressBlackboardJobNode;

import eu.dnetlib.rmi.blackboard.FeedMDStoreMessage;

import eu.dnetlib.rmi.objects.resultSet.ResultSet;

import eu.dnetlib.rmi.soap.MDStoreService;

public class StoreMDStoreRecordsJobNode extends ProgressBlackboardJobNode<FeedMDStoreMessage> {

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

	private String resultSetParam;

	private String mdId;

	private String storingType;

	@Override

	protected String obtainServiceId(final NodeToken token) {

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

	}

	@Override

	protected FeedMDStoreMessage prepareMessage(final NodeToken token) {

		@SuppressWarnings("unchecked")

		final ResultSet<String> rs = (ResultSet<String>) token.getEnv().getTransientAttribute(resultSetParam);

		return new FeedMDStoreMessage(mdId, storingType, rs);

	}

	@Override

	protected void saveResponseInEnv(final FeedMDStoreMessage m, final Env env) {

		log.info("Number of stored records: " + m.getSize());

		env.setAttribute(WorkflowConstants.MAIN_LOG_PREFIX + "total", m.getSize());

	}

	public String getResultSetParam() {

		return resultSetParam;

	}

	public void setResultSetParam(final String resultSetParam) {

		this.resultSetParam = resultSetParam;

	}

	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;

	}

}