forked from D-Net/dnet-hadoop
191 lines
6.0 KiB
Java
191 lines
6.0 KiB
Java
|
package eu.dnetlib.dhp.migration;
|
||
|
|
||
|
import java.io.Closeable;
|
||
|
import java.io.IOException;
|
||
|
import java.io.StringReader;
|
||
|
import java.util.ArrayList;
|
||
|
import java.util.List;
|
||
|
import java.util.Map.Entry;
|
||
|
|
||
|
import org.apache.commons.io.IOUtils;
|
||
|
import org.apache.commons.logging.Log;
|
||
|
import org.apache.commons.logging.LogFactory;
|
||
|
import org.dom4j.Document;
|
||
|
import org.dom4j.DocumentException;
|
||
|
import org.dom4j.Node;
|
||
|
import org.dom4j.io.SAXReader;
|
||
|
|
||
|
import eu.dnetlib.dhp.application.ArgumentApplicationParser;
|
||
|
import eu.dnetlib.dhp.schema.oaf.Dataset;
|
||
|
import eu.dnetlib.dhp.schema.oaf.Oaf;
|
||
|
import eu.dnetlib.dhp.schema.oaf.OtherResearchProduct;
|
||
|
import eu.dnetlib.dhp.schema.oaf.Publication;
|
||
|
import eu.dnetlib.dhp.schema.oaf.Relation;
|
||
|
import eu.dnetlib.dhp.schema.oaf.Result;
|
||
|
import eu.dnetlib.dhp.schema.oaf.Software;
|
||
|
|
||
|
public class MigrateMongoMdstoresApplication extends AbstractMigrateApplication implements Closeable {
|
||
|
|
||
|
private static final Log log = LogFactory.getLog(MigrateMongoMdstoresApplication.class);
|
||
|
|
||
|
private final MdstoreClient mdstoreClient;
|
||
|
|
||
|
public static void main(final String[] args) throws Exception {
|
||
|
final ArgumentApplicationParser parser = new ArgumentApplicationParser(
|
||
|
IOUtils.toString(MigrateMongoMdstoresApplication.class.getResourceAsStream("/eu/dnetlib/dhp/migration/migrate_mongo_mstores_parameters.json")));
|
||
|
parser.parseArgument(args);
|
||
|
|
||
|
final String mongoBaseUrl = parser.get("mongoBaseUrl");
|
||
|
final String mongoDb = parser.get("mongoDb");
|
||
|
|
||
|
final String mdFormat = parser.get("mdFormat");
|
||
|
final String mdLayout = parser.get("mdLayout");
|
||
|
final String mdInterpretation = parser.get("mdInterpretation");
|
||
|
|
||
|
final String hdfsPath = parser.get("hdfsPath");
|
||
|
final String hdfsNameNode = parser.get("namenode");
|
||
|
final String hdfsUser = parser.get("hdfsUser");
|
||
|
|
||
|
try (final MigrateMongoMdstoresApplication mig = new MigrateMongoMdstoresApplication(hdfsPath, hdfsNameNode, hdfsUser, mongoBaseUrl, mongoDb)) {
|
||
|
mig.processMdRecords(mdFormat, mdLayout, mdInterpretation);
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
public MigrateMongoMdstoresApplication(final String hdfsPath, final String hdfsNameNode, final String hdfsUser, final String mongoBaseUrl,
|
||
|
final String mongoDb) throws Exception {
|
||
|
super(hdfsPath, hdfsNameNode, hdfsUser);
|
||
|
this.mdstoreClient = new MdstoreClient(mongoBaseUrl, mongoDb);
|
||
|
|
||
|
}
|
||
|
|
||
|
public void processMdRecords(final String mdFormat, final String mdLayout, final String mdInterpretation) throws DocumentException {
|
||
|
|
||
|
for (final Entry<String, String> entry : mdstoreClient.validCollections(mdFormat, mdLayout, mdInterpretation).entrySet()) {
|
||
|
// final String mdId = entry.getKey();
|
||
|
final String currentColl = entry.getValue();
|
||
|
|
||
|
for (final String xml : mdstoreClient.listRecords(currentColl)) {
|
||
|
for (final Oaf oaf : createOafs(xml)) {
|
||
|
emitOaf(oaf);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private List<Oaf> createOafs(final String xml) throws DocumentException {
|
||
|
final SAXReader reader = new SAXReader();
|
||
|
final Document doc = reader.read(new StringReader(xml));
|
||
|
|
||
|
final String type = doc.valueOf(""); // TODO
|
||
|
|
||
|
final List<Oaf> oafs = new ArrayList<>();
|
||
|
|
||
|
switch (type.toLowerCase()) {
|
||
|
case "publication":
|
||
|
final Publication p = new Publication();
|
||
|
populateResultFields(p, doc);
|
||
|
p.setJournal(null); // TODO
|
||
|
oafs.add(p);
|
||
|
break;
|
||
|
case "dataset":
|
||
|
final Dataset d = new Dataset();
|
||
|
populateResultFields(d, doc);
|
||
|
d.setStoragedate(null); // TODO
|
||
|
d.setDevice(null); // TODO
|
||
|
d.setSize(null); // TODO
|
||
|
d.setVersion(null); // TODO
|
||
|
d.setLastmetadataupdate(null); // TODO
|
||
|
d.setMetadataversionnumber(null); // TODO
|
||
|
d.setGeolocation(null); // TODO
|
||
|
oafs.add(d);
|
||
|
break;
|
||
|
case "otherresearchproducts":
|
||
|
final OtherResearchProduct o = new OtherResearchProduct();
|
||
|
populateResultFields(o, doc);
|
||
|
o.setContactperson(null); // TODO
|
||
|
o.setContactgroup(null); // TODO
|
||
|
o.setTool(null); // TODO
|
||
|
oafs.add(o);
|
||
|
break;
|
||
|
case "software":
|
||
|
final Software s = new Software();
|
||
|
populateResultFields(s, doc);
|
||
|
s.setDocumentationUrl(null); // TODO
|
||
|
s.setLicense(null); // TODO
|
||
|
s.setCodeRepositoryUrl(null); // TODO
|
||
|
s.setProgrammingLanguage(null); // TODO
|
||
|
oafs.add(s);
|
||
|
break;
|
||
|
default:
|
||
|
log.error("Inavlid type: " + type);
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
if (!oafs.isEmpty()) {
|
||
|
addRelations(oafs, doc, "//*", "TYPE");
|
||
|
addRelations(oafs, doc, "//*", "TYPE");
|
||
|
addRelations(oafs, doc, "//*", "TYPE");
|
||
|
}
|
||
|
|
||
|
return oafs;
|
||
|
}
|
||
|
|
||
|
private void addRelations(final List<Oaf> oafs, final Document doc, final String xpath, final String type) {
|
||
|
for (final Object o : doc.selectNodes(xpath)) {
|
||
|
final Node n = (Node) o;
|
||
|
final Relation r = new Relation();
|
||
|
r.setRelType(null); // TODO
|
||
|
r.setSubRelType(null); // TODO
|
||
|
r.setRelClass(null); // TODO
|
||
|
r.setSource(null); // TODO
|
||
|
r.setTarget(null); // TODO
|
||
|
r.setCollectedFrom(null); // TODO
|
||
|
oafs.add(r);
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
private void populateResultFields(final Result r, final Document doc) {
|
||
|
r.setDataInfo(null); // TODO
|
||
|
r.setLastupdatetimestamp(null); // TODO
|
||
|
r.setId(null); // TODO
|
||
|
r.setOriginalId(null); // TODO
|
||
|
r.setCollectedfrom(null); // TODO
|
||
|
r.setPid(null); // TODO
|
||
|
r.setDateofcollection(null); // TODO
|
||
|
r.setDateoftransformation(null); // TODO
|
||
|
r.setExtraInfo(null); // TODO
|
||
|
r.setOaiprovenance(null); // TODO
|
||
|
r.setAuthor(null); // TODO
|
||
|
r.setResulttype(null); // TODO
|
||
|
r.setLanguage(null); // TODO
|
||
|
r.setCountry(null); // TODO
|
||
|
r.setSubject(null); // TODO
|
||
|
r.setTitle(null); // TODO
|
||
|
r.setRelevantdate(null); // TODO
|
||
|
r.setDescription(null); // TODO
|
||
|
r.setDateofacceptance(null); // TODO
|
||
|
r.setPublisher(null); // TODO
|
||
|
r.setEmbargoenddate(null); // TODO
|
||
|
r.setSource(null); // TODO
|
||
|
r.setFulltext(null); // TODO
|
||
|
r.setFormat(null); // TODO
|
||
|
r.setContributor(null); // TODO
|
||
|
r.setResourcetype(null); // TODO
|
||
|
r.setCoverage(null); // TODO
|
||
|
r.setRefereed(null); // TODO
|
||
|
r.setContext(null); // TODO
|
||
|
r.setExternalReference(null); // TODO
|
||
|
r.setInstance(null); // TODO
|
||
|
r.setProcessingchargeamount(null); // TODO
|
||
|
r.setProcessingchargecurrency(null); // TODO
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
public void close() throws IOException {
|
||
|
super.close();
|
||
|
mdstoreClient.close();
|
||
|
}
|
||
|
}
|