D-Net

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

import javax.annotation.Resource;

import javax.xml.ws.wsaddressing.W3CEndpointReference;

import eu.dnetlib.data.transformation.service.rmi.TransformationService;

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

import eu.dnetlib.enabling.resultset.client.utils.EPRUtils;

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

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

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

public class TransformJobNode extends SimpleJobNode {

	private String inputEprParam;

	private String outputEprParam;

	private String ruleId;

	@Resource

	private UniqueServiceLocator serviceLocator;

	@Override

	protected String execute(final Env env) throws Exception {

		final W3CEndpointReference inputEpr = new EPRUtils().getEpr(env.getAttribute(inputEprParam, String.class));

		final W3CEndpointReference outputEpr = serviceLocator.getService(TransformationService.class).transform(ruleId, inputEpr);

		env.setAttribute(outputEprParam, outputEpr.toString());

		return Arc.DEFAULT_ARC;

	}

	public String getInputEprParam() {

		return inputEprParam;

	}

	public void setInputEprParam(final String inputEprParam) {

		this.inputEprParam = inputEprParam;

	}

	public String getOutputEprParam() {

		return outputEprParam;

	}

	public void setOutputEprParam(final String outputEprParam) {

		this.outputEprParam = outputEprParam;

	}

	public String getRuleId() {

		return ruleId;

	}

	public void setRuleId(final String ruleId) {

		this.ruleId = ruleId;

	}

}