Project

General

Profile

1 35797 claudio.at
package eu.dnetlib.msro.workflows.hadoop;
2
3
import java.io.IOException;
4
5 37014 claudio.at
import org.apache.commons.lang.StringUtils;
6
import org.apache.commons.logging.Log;
7
import org.apache.commons.logging.LogFactory;
8
import org.springframework.beans.factory.annotation.Autowired;
9 35797 claudio.at
10
import com.googlecode.sarasvati.Arc;
11
import com.googlecode.sarasvati.NodeToken;
12
13 37014 claudio.at
import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpDocumentNotFoundException;
14
import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpException;
15
import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService;
16
import eu.dnetlib.enabling.locators.UniqueServiceLocator;
17
import eu.dnetlib.msro.rmi.MSROException;
18 35797 claudio.at
import eu.dnetlib.msro.workflows.nodes.SimpleJobNode;
19
20
public class PrepareMDStoreImportJobNode extends SimpleJobNode {
21
22 37014 claudio.at
	private static final Log log = LogFactory.getLog(PrepareMDStoreImportJobNode.class);
23
24
	@Autowired
25
	private UniqueServiceLocator serviceLocator;
26
27 35797 claudio.at
	private String hdfsPathParam;
28 37014 claudio.at
29 35797 claudio.at
	private String hdfsPath;
30 37014 claudio.at
31
	private String mappingParam;
32
33
	private String mapping;
34
35 35797 claudio.at
	@Override
36 37014 claudio.at
	protected String execute(final NodeToken token) throws Exception {
37
38 35797 claudio.at
		token.getEnv().setAttribute(getHdfsPathParam(), getHdfsPath());
39 37014 claudio.at
		token.getEnv().setAttribute(getMappingParam(), readXslt(getMapping()));
40
41 35797 claudio.at
		return Arc.DEFAULT_ARC;
42
	}
43
44 37014 claudio.at
	private String readXslt(final String profileId) throws IOException, MSROException, ISLookUpDocumentNotFoundException, ISLookUpException {
45
		if (StringUtils.isBlank(profileId)) throw new MSROException("missing profile id");
46
47
		log.info("loading mapping from profile id: " + profileId);
48
49
		final String xquery =
50
				String.format("/RESOURCE_PROFILE[.//RESOURCE_IDENTIFIER/@value='%s']/BODY/CONFIGURATION/SCRIPT/CODE/*[local-name()='stylesheet']", profileId);
51
		return serviceLocator.getService(ISLookUpService.class).getResourceProfileByQuery(xquery);
52 35797 claudio.at
	}
53
54
	public String getHdfsPathParam() {
55
		return hdfsPathParam;
56
	}
57
58 37014 claudio.at
	public void setHdfsPathParam(final String hdfsPathParam) {
59 35797 claudio.at
		this.hdfsPathParam = hdfsPathParam;
60
	}
61
62
	public String getHdfsPath() {
63
		return hdfsPath;
64
	}
65
66 37014 claudio.at
	public void setHdfsPath(final String hdfsPath) {
67 35797 claudio.at
		this.hdfsPath = hdfsPath;
68
	}
69
70 37014 claudio.at
	public String getMapping() {
71
		return mapping;
72 35797 claudio.at
	}
73
74 37014 claudio.at
	public void setMapping(final String mapping) {
75
		this.mapping = mapping;
76 35797 claudio.at
	}
77
78 37014 claudio.at
	public String getMappingParam() {
79
		return mappingParam;
80 35797 claudio.at
	}
81
82 37014 claudio.at
	public void setMappingParam(final String mappingParam) {
83
		this.mappingParam = mappingParam;
84 35797 claudio.at
	}
85
86
}