Compare commits

...

4 Commits

Author SHA1 Message Date
Sandro La Bruzzo 4f07dba68f updated index mapping to include orcid 2023-04-27 10:00:34 +02:00
Sandro La Bruzzo ab1842e5dc implemented scholix Flat mapping 2023-04-12 11:52:41 +02:00
Sandro La Bruzzo ef82b8362d Merge remote-tracking branch 'origin/beta' into scholix_flat_indexing 2023-04-04 09:07:53 +02:00
Sandro La Bruzzo 20c395fd86 Added implementation of Scholix Flat 2023-02-14 10:16:40 +01:00
16 changed files with 739 additions and 143 deletions

View File

@ -1,14 +1,23 @@
package eu.dnetlib.dhp.sx.graph.scholix
import eu.dnetlib.dhp.schema.oaf.{Publication, Relation, Result, StructuredProperty}
import eu.dnetlib.dhp.schema.oaf.{
Dataset,
OtherResearchProduct,
Publication,
Relation,
Result,
Software,
StructuredProperty
}
import eu.dnetlib.dhp.schema.sx.scholix._
import eu.dnetlib.dhp.schema.sx.summary.{CollectedFromType, SchemeValue, ScholixSummary, Typology}
import eu.dnetlib.dhp.schema.sx.summary.{AuthorPid, CollectedFromType, SchemeValue, ScholixSummary, Typology}
import eu.dnetlib.dhp.utils.DHPUtils
import org.apache.spark.sql.expressions.Aggregator
import org.apache.spark.sql.{Encoder, Encoders}
import org.json4s
import org.json4s.DefaultFormats
import org.json4s.jackson.JsonMethods.parse
import scala.collection.JavaConverters._
import scala.io.Source
@ -59,6 +68,36 @@ object ScholixUtils extends Serializable {
}
}
def flattenizeScholix(input: Scholix, json: String): ScholixFlat = {
val flat: ScholixFlat = new ScholixFlat
flat.setIdentifier(input.getIdentifier)
flat.setRelationType(input.getRelationship.getName)
if (input.getSource != null && input.getSource.getIdentifier != null) {
flat.setSourceId(input.getSource.getDnetIdentifier)
flat.setSourcePid(input.getSource.getIdentifier.asScala.map(p => p.getIdentifier).distinct.toList.asJava)
flat.setSourcePidType(input.getSource.getIdentifier.asScala.map(p => p.getSchema).distinct.toList.asJava)
flat.setSourceType(input.getSource.getObjectType)
flat.setSourceSubType(input.getSource.getObjectSubType)
} else return null
if (input.getSource.getPublisher != null)
flat.setSourcePublisher(input.getSource.getPublisher.asScala.map(p => p.getName).toList.asJava)
if (input.getTarget != null && input.getTarget.getIdentifier != null) {
flat.setTargetId(input.getTarget.getDnetIdentifier)
flat.setTargetPid(input.getTarget.getIdentifier.asScala.map(p => p.getIdentifier).distinct.toList.asJava)
flat.setTargetPidType(input.getTarget.getIdentifier.asScala.map(p => p.getSchema).distinct.toList.asJava)
flat.setTargetType(input.getTarget.getObjectType)
flat.setTargetSubType(input.getTarget.getObjectSubType)
} else return null
if (input.getTarget.getPublisher != null)
flat.setTargetPublisher(input.getTarget.getPublisher.asScala.map(p => p.getName).distinct.toList.asJava)
flat.setPublicationDate(input.getPublicationDate)
if (input.getLinkprovider != null)
flat.setLinkProviders(input.getLinkprovider.asScala.map(l => l.getName).distinct.toList.asJava)
flat.setBlob(json);
flat
}
def inverseRelationShip(rel: ScholixRelationship): ScholixRelationship = {
new ScholixRelationship(rel.getInverse, rel.getSchema, rel.getName)
@ -232,7 +271,16 @@ object ScholixUtils extends Serializable {
if (summaryObject.getAuthor != null && !summaryObject.getAuthor.isEmpty) {
val l: List[ScholixEntityId] =
summaryObject.getAuthor.asScala.map(a => new ScholixEntityId(a, null)).toList
summaryObject.getAuthor.asScala
.map(a => {
if (a.getORCID != null)
new ScholixEntityId(
a.getFullname,
List(new ScholixIdentifier(a.getORCID, "ORCID", s"https://orcid.org/${a.getORCID}")).asJava
)
else new ScholixEntityId(a.getFullname, null)
})
.toList
if (l.nonEmpty)
r.setCreator(l.asJava)
}
@ -377,10 +425,13 @@ object ScholixUtils extends Serializable {
if (persistentIdentifiers.isEmpty)
return null
s.setLocalIdentifier(persistentIdentifiers.asJava)
if (r.isInstanceOf[Publication])
s.setTypology(Typology.publication)
else
s.setTypology(Typology.dataset)
r match {
case _: Publication => s.setTypology(Typology.publication)
case _: Dataset => s.setTypology(Typology.dataset)
case _: Software => s.setTypology(Typology.software)
case _: OtherResearchProduct => s.setTypology(Typology.otherresearchproduct)
case _ =>
}
s.setSubType(r.getInstance().get(0).getInstancetype.getClassname)
@ -393,7 +444,20 @@ object ScholixUtils extends Serializable {
}
if (r.getAuthor != null && !r.getAuthor.isEmpty) {
val authors: List[String] = r.getAuthor.asScala.map(a => a.getFullname).toList
val authors: List[AuthorPid] = r.getAuthor.asScala
.map(a => {
var ORCID: String = null
if (a.getPid != null) {
val result = a.getPid.asScala.find(p =>
p.getQualifier != null && p.getQualifier.getClassid != null && p.getQualifier.getClassid.toLowerCase
.contains("orcid")
)
if (result.isDefined)
ORCID = result.get.getValue
}
new AuthorPid(a.getFullname, ORCID)
})
.toList
if (authors.nonEmpty)
s.setAuthor(authors.asJava)
}

View File

@ -0,0 +1,43 @@
package eu.dnetlib.dhp.sx.graph.scholix;
import static org.junit.jupiter.api.Assertions.*;
import java.io.*;
import java.util.zip.GZIPInputStream;
import org.junit.jupiter.api.Test;
import com.fasterxml.jackson.databind.ObjectMapper;
import eu.dnetlib.dhp.schema.sx.scholix.Scholix;
import eu.dnetlib.dhp.schema.sx.scholix.ScholixFlat;
public class ScholixFlatTest {
@Test
public void flattenScholixTest() throws IOException {
final ObjectMapper mapper = new ObjectMapper();
InputStream gzipStream = new GZIPInputStream(getClass().getResourceAsStream("scholix_records.gz"));
Reader decoder = new InputStreamReader(gzipStream, "UTF-8");
BufferedReader buffered = new BufferedReader(decoder);
String line;
FileWriter myWriter = new FileWriter("/Users/sandro/Downloads/records");
while ((line = buffered.readLine()) != null) {
final Scholix s = mapper.readValue(line, Scholix.class);
final ScholixFlat flat = ScholixUtils.flattenizeScholix(s, line);
assertNotNull(s);
assertNotNull(flat);
assertEquals(s.getIdentifier(), flat.getIdentifier());
assertEquals(s.getRelationship().getName(), flat.getRelationType());
assertEquals(s.getSource().getObjectType(), flat.getSourceType());
assertEquals(s.getSource().getObjectSubType(), flat.getSourceSubType());
myWriter.write(mapper.writeValueAsString(flat));
myWriter.write("\n");
}
myWriter.close();
}
}

View File

@ -0,0 +1,91 @@
package eu.dnetlib.dhp.sx.graph.dump;
import java.io.Serializable;
import java.util.Optional;
import org.apache.commons.io.IOUtils;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.*;
import eu.dnetlib.dhp.application.ArgumentApplicationParser;
import eu.dnetlib.dhp.common.api.MissingConceptDoiException;
import eu.dnetlib.dhp.common.api.ZenodoAPIClient;
public class SendToZenodoHDFS implements Serializable {
private static final String NEW = "new"; // to be used for a brand new deposition in zenodo
private static final String VERSION = "version"; // to be used to upload a new version of a published deposition
private static final String UPDATE = "update"; // to upload content to an open deposition not published
public static void main(final String[] args) throws Exception, MissingConceptDoiException {
final ArgumentApplicationParser parser = new ArgumentApplicationParser(
IOUtils
.toString(
SendToZenodoHDFS.class
.getResourceAsStream(
"/eu/dnetlib/dhp/sx/graph/dump/upload_zenodo.json")));
parser.parseArgument(args);
final String hdfsPath = parser.get("hdfsPath");
final String hdfsNameNode = parser.get("nameNode");
final String access_token = parser.get("accessToken");
final String connection_url = parser.get("connectionUrl");
final String metadata = parser.get("metadata");
final String depositionType = parser.get("depositionType");
final String concept_rec_id = Optional
.ofNullable(parser.get("conceptRecordId"))
.orElse(null);
final String depositionId = Optional.ofNullable(parser.get("depositionId")).orElse(null);
Configuration conf = new Configuration();
conf.set("fs.defaultFS", hdfsNameNode);
FileSystem fileSystem = FileSystem.get(conf);
RemoteIterator<LocatedFileStatus> fileStatusListIterator = fileSystem
.listFiles(
new Path(hdfsPath), true);
ZenodoAPIClient zenodoApiClient = new ZenodoAPIClient(connection_url, access_token);
switch (depositionType) {
case NEW:
zenodoApiClient.newDeposition();
break;
case VERSION:
if (concept_rec_id == null) {
throw new MissingConceptDoiException("No concept record id has been provided");
}
zenodoApiClient.newVersion(concept_rec_id);
break;
case UPDATE:
if (depositionId == null) {
throw new MissingConceptDoiException("No deposition id has been provided");
}
zenodoApiClient.uploadOpenDeposition(depositionId);
break;
default:
throw new RuntimeException("No available entries");
}
while (fileStatusListIterator.hasNext()) {
LocatedFileStatus fileStatus = fileStatusListIterator.next();
Path p = fileStatus.getPath();
String pString = p.toString();
if (!pString.endsWith("_SUCCESS")) {
String name = pString.substring(pString.lastIndexOf("/") + 1);
FSDataInputStream inputStream = fileSystem.open(p);
// zenodoApiClient.uploadIS(inputStream, name);
}
}
if (!metadata.equals("")) {
zenodoApiClient.sendMretadata(metadata);
}
}
}

View File

@ -0,0 +1,50 @@
[
{
"paramName": "dt",
"paramLongName": "depositionType",
"paramDescription": "the type of the deposition (new, version, update)",
"paramRequired": true
},
{
"paramName": "cri",
"paramLongName": "conceptRecordId",
"paramDescription": "The id of the concept record for a new version",
"paramRequired": false
},
{
"paramName": "di",
"paramLongName": "depositionId",
"paramDescription": "the id of an open deposition which has not been published",
"paramRequired": false
},
{
"paramName": "hdfsp",
"paramLongName": "hdfsPath",
"paramDescription": "the path of the folder tofind files to send to Zenodo",
"paramRequired": true
},
{
"paramName": "nn",
"paramLongName": "nameNode",
"paramDescription": "the name node",
"paramRequired": true
},
{
"paramName": "at",
"paramLongName": "accessToken",
"paramDescription": "the access token for the deposition",
"paramRequired": false
},
{
"paramName": "cu",
"paramLongName": "connectionUrl",
"paramDescription": "the url to connect to deposit",
"paramRequired": false
},
{
"paramName": "m",
"paramLongName": "metadata",
"paramDescription": "metadata associated to the deposition",
"paramRequired": false
}
]

View File

@ -8,148 +8,36 @@
<name>targetPath</name>
<description>the final graph path</description>
</property>
<property>
<name>relationFilter</name>
<description>Filter relation semantic</description>
</property>
<property>
<name>maxNumberOfPid</name>
<description>filter relation with at least #maxNumberOfPid</description>
</property>
<property>
<name>dumpCitations</name>
<value>false</value>
<description>should dump citation relations</description>
</property>
</parameters>
<start to="ImportDatasetEntities"/>
<start to="make_tar"/>
<kill name="Kill">
<message>Action failed, error message[${wf:errorMessage(wf:lastErrorNode())}]</message>
</kill>
<action name="ImportDatasetEntities">
<spark xmlns="uri:oozie:spark-action:0.2">
<master>yarn</master>
<mode>cluster</mode>
<name>Import JSONRDD to Dataset kryo</name>
<class>eu.dnetlib.dhp.sx.graph.SparkConvertRDDtoDataset</class>
<jar>dhp-graph-mapper-${projectVersion}.jar</jar>
<spark-opts>
--executor-memory=${sparkExecutorMemory}
--executor-cores=${sparkExecutorCores}
--driver-memory=${sparkDriverMemory}
--conf spark.extraListeners=${spark2ExtraListeners}
--conf spark.sql.shuffle.partitions=3000
--conf spark.sql.queryExecutionListeners=${spark2SqlQueryExecutionListeners}
--conf spark.yarn.historyServer.address=${spark2YarnHistoryServerAddress}
--conf spark.eventLog.dir=${nameNode}${spark2EventLogDir}
</spark-opts>
<arg>--master</arg><arg>yarn</arg>
<arg>--sourcePath</arg><arg>${sourcePath}</arg>
<arg>--targetPath</arg><arg>${targetPath}</arg>
<arg>--filterRelation</arg><arg>${relationFilter}</arg>
</spark>
<ok to="CreateSummaries"/>
<error to="Kill"/>
</action>
<action name="CreateSummaries">
<spark xmlns="uri:oozie:spark-action:0.2">
<master>yarn</master>
<mode>cluster</mode>
<name>Convert Entities to summaries</name>
<class>eu.dnetlib.dhp.sx.graph.SparkCreateSummaryObject</class>
<jar>dhp-graph-mapper-${projectVersion}.jar</jar>
<spark-opts>
--executor-memory=${sparkExecutorMemory}
--executor-cores=${sparkExecutorCores}
--driver-memory=${sparkDriverMemory}
--conf spark.extraListeners=${spark2ExtraListeners}
--conf spark.sql.shuffle.partitions=20000
--conf spark.sql.queryExecutionListeners=${spark2SqlQueryExecutionListeners}
--conf spark.yarn.historyServer.address=${spark2YarnHistoryServerAddress}
--conf spark.eventLog.dir=${nameNode}${spark2EventLogDir}
</spark-opts>
<arg>--master</arg><arg>yarn</arg>
<arg>--sourcePath</arg><arg>${targetPath}/entities</arg>
<arg>--targetPath</arg><arg>${targetPath}/provision/summaries</arg>
</spark>
<ok to="CreateScholix"/>
<error to="Kill"/>
</action>
<action name="CreateScholix">
<spark xmlns="uri:oozie:spark-action:0.2">
<master>yarn</master>
<mode>cluster</mode>
<name>Generate Scholix Dataset</name>
<class>eu.dnetlib.dhp.sx.graph.SparkCreateScholix</class>
<jar>dhp-graph-mapper-${projectVersion}.jar</jar>
<spark-opts>
--executor-memory=${sparkExecutorMemory}
--executor-cores=${sparkExecutorCores}
--driver-memory=${sparkDriverMemory}
--conf spark.extraListeners=${spark2ExtraListeners}
--conf spark.sql.shuffle.partitions=30000
--conf spark.sql.queryExecutionListeners=${spark2SqlQueryExecutionListeners}
--conf spark.yarn.historyServer.address=${spark2YarnHistoryServerAddress}
--conf spark.eventLog.dir=${nameNode}${spark2EventLogDir}
</spark-opts>
<arg>--master</arg><arg>yarn</arg>
<arg>--summaryPath</arg><arg>${targetPath}/provision/summaries</arg>
<arg>--targetPath</arg><arg>${targetPath}/provision/scholix</arg>
<arg>--relationPath</arg><arg>${targetPath}/relation</arg>
<arg>--dumpCitations</arg><arg>${dumpCitations}</arg>
</spark>
<ok to="DropJSONPath"/>
<error to="Kill"/>
</action>
<action name="DropJSONPath">
<fs>
<delete path='${targetPath}/json'/>
<mkdir path='${targetPath}/json/'/>
</fs>
<ok to="SerializeScholix"/>
<error to="Kill"/>
</action>
<action name="SerializeScholix">
<spark xmlns="uri:oozie:spark-action:0.2">
<master>yarn</master>
<mode>cluster</mode>
<name>Serialize scholix to JSON</name>
<class>eu.dnetlib.dhp.sx.graph.SparkConvertObjectToJson</class>
<jar>dhp-graph-mapper-${projectVersion}.jar</jar>
<spark-opts>
--executor-memory=${sparkExecutorMemory}
--executor-cores=${sparkExecutorCores}
--driver-memory=${sparkDriverMemory}
--conf spark.extraListeners=${spark2ExtraListeners}
--conf spark.sql.shuffle.partitions=6000
--conf spark.sql.queryExecutionListeners=${spark2SqlQueryExecutionListeners}
--conf spark.yarn.historyServer.address=${spark2YarnHistoryServerAddress}
--conf spark.eventLog.dir=${nameNode}${spark2EventLogDir}
</spark-opts>
<arg>--master</arg><arg>yarn</arg>
<arg>--sourcePath</arg><arg>${targetPath}/provision/scholix/scholix</arg>
<arg>--targetPath</arg><arg>${targetPath}/json/scholix_json</arg>
<arg>--objectType</arg><arg>scholix</arg>
<arg>--maxPidNumberFilter</arg><arg>maxNumberOfPid</arg>
</spark>
<ok to="make_tar"/>
<error to="Kill"/>
</action>
<action name="make_tar">
<java>
<main-class>eu.dnetlib.dhp.common.MakeTarArchive</main-class>
<arg>--nameNode</arg><arg>${nameNode}</arg>
<arg>--hdfsPath</arg><arg>${targetPath}</arg>
<arg>--sourcePath</arg><arg>${sourcePath}</arg>
</java>
<ok to="send_zenodo"/>
<error to="Kill"/>
</action>
<action name="send_zenodo">
<java>
<main-class>eu.dnetlib.dhp.sx.graph.dump.SendToZenodoHDFS</main-class>
<arg>--hdfsPath</arg><arg>${targetPath}/tar</arg>
<arg>--sourcePath</arg><arg>${targetPath}/json</arg>
<arg>--nameNode</arg><arg>${nameNode}</arg>
<arg>--accessToken</arg><arg>${accessToken}</arg>
<arg>--connectionUrl</arg><arg>${connectionUrl}</arg>
<arg>--metadata</arg><arg>${metadata}</arg>
<arg>--conceptRecordId</arg><arg>${conceptRecordId}</arg>
<arg>--depositionType</arg><arg>${depositionType}</arg>
<arg>--depositionId</arg><arg>${depositionId}</arg>
</java>
<ok to="End"/>
<error to="Kill"/>

View File

@ -103,7 +103,7 @@ public class DropAndCreateESIndex {
Objects
.requireNonNull(
DropAndCreateESIndex.class
.getResourceAsStream("/eu/dnetlib/dhp/sx/provision/scholix_index.json")));
.getResourceAsStream("/eu/dnetlib/dhp/sx/provision/scholix_index_flat.json")));
log.info("creating Index SCHOLIX");
final HttpPut put = new HttpPut(String.format(url, ip, index, "scholix"));

View File

@ -0,0 +1,190 @@
package eu.dnetlib.dhp.sx.provision;
import java.util.ArrayList;
import java.util.List;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import eu.dnetlib.dhp.schema.sx.scholix.Scholix;
import eu.dnetlib.dhp.schema.sx.scholix.ScholixResource;
public class ScholixFlat {
private static ObjectMapper MAPPER = new ObjectMapper();
private List<String> linkProvider = new ArrayList<>();
private String publicationDate;
private List<String> sourceLinkPublisher = new ArrayList<>();
private List<String> targetLinkPublisher = new ArrayList<>();
private String sourceDnetIdentifier;
private String targetDnetIdentifier;
private List<String> sourcePids = new ArrayList<>();
private List<String> sourcePidTypes = new ArrayList<>();
private List<String> targetPids = new ArrayList<>();
private List<String> targetPidTypes = new ArrayList<>();
private String json;
public void addLinkProvider(final String providerName) {
addStringToList(providerName, this.linkProvider);
}
public void addSourceLinkPublisher(final String linkPublisher) {
addStringToList(linkPublisher, sourceLinkPublisher);
}
public void addTargetLinkPublisher(final String linkPublisher) {
addStringToList(linkPublisher, targetLinkPublisher);
}
public void addSourcePid(final String pid) {
addStringToList(pid, sourcePids);
}
public void addSourcePidType(final String pidType) {
addStringToList(pidType, sourcePidTypes);
}
public void addTargetPidType(final String pidType) {
addStringToList(pidType, targetPidTypes);
}
public void addTargetPid(final String pid) {
addStringToList(pid, targetPids);
}
public void addStringToList(final String s, final List<String> l) {
if (l != null && !l.contains(s))
l.add(s);
}
public String getSourceDnetIdentifier() {
return sourceDnetIdentifier;
}
public void setSourceDnetIdentifier(String sourceDnetIdentifier) {
this.sourceDnetIdentifier = sourceDnetIdentifier;
}
public String getTargetDnetIdentifier() {
return targetDnetIdentifier;
}
public void setTargetDnetIdentifier(String targetDnetIdentifier) {
this.targetDnetIdentifier = targetDnetIdentifier;
}
public List<String> getSourcePids() {
return sourcePids;
}
public void setSourcePids(List<String> sourcePids) {
this.sourcePids = sourcePids;
}
public List<String> getSourcePidTypes() {
return sourcePidTypes;
}
public void setSourcePidTypes(List<String> sourcePidTypes) {
this.sourcePidTypes = sourcePidTypes;
}
public List<String> getTargetPids() {
return targetPids;
}
public void setTargetPids(List<String> targetPids) {
this.targetPids = targetPids;
}
public List<String> getTargetPidTypes() {
return targetPidTypes;
}
public void setTargetPidTypes(List<String> targetPidTypes) {
this.targetPidTypes = targetPidTypes;
}
public List<String> getSourceLinkPublisher() {
return sourceLinkPublisher;
}
public void setSourceLinkPublisher(List<String> sourceLinkPublisher) {
this.sourceLinkPublisher = sourceLinkPublisher;
}
public List<String> getTargetLinkPublisher() {
return targetLinkPublisher;
}
public void setTargetLinkPublisher(List<String> targetLinkPublisher) {
this.targetLinkPublisher = targetLinkPublisher;
}
public List<String> getLinkProvider() {
return linkProvider;
}
public void setLinkProvider(List<String> linkProvider) {
this.linkProvider = linkProvider;
}
public String getPublicationDate() {
return publicationDate;
}
public void setPublicationDate(String publicationDate) {
this.publicationDate = publicationDate;
}
public String getJson() {
return json;
}
public void setJson(String json) {
this.json = json;
}
public static ScholixFlat fromScholix(final Scholix scholix) throws JsonProcessingException {
if (scholix == null || scholix.getSource() == null || scholix.getTarget() == null)
return null;
final ScholixFlat flat = new ScholixFlat();
if (scholix.getLinkprovider() != null)
scholix.getLinkprovider().forEach(l -> flat.addLinkProvider(l.getName()));
flat.setPublicationDate(scholix.getPublicationDate());
final ScholixResource source = scholix.getSource();
flat.setSourceDnetIdentifier(source.getDnetIdentifier());
if (source.getIdentifier() != null) {
source.getIdentifier().forEach(i -> {
flat.addSourcePid(i.getIdentifier());
flat.addSourcePidType(i.getSchema());
});
}
if (source.getPublisher() != null) {
source.getPublisher().forEach(p -> flat.addSourceLinkPublisher(p.getName()));
}
final ScholixResource target = scholix.getSource();
flat.setTargetDnetIdentifier(target.getDnetIdentifier());
if (target.getIdentifier() != null) {
target.getIdentifier().forEach(i -> {
flat.addTargetPid(i.getIdentifier());
flat.addTargetPidType(i.getSchema());
});
}
if (target.getPublisher() != null) {
target.getPublisher().forEach(p -> flat.addTargetLinkPublisher(p.getName()));
}
flat.setJson(MAPPER.writeValueAsString(scholix));
return flat;
}
}

View File

@ -42,7 +42,8 @@ public class SparkIndexCollectionOnES {
.toString(
Objects
.requireNonNull(
DropAndCreateESIndex.class.getResourceAsStream("/eu/dnetlib/dhp/sx/provision/cluster.json")));
SparkIndexCollectionOnES.class
.getResourceAsStream("/eu/dnetlib/dhp/sx/provision/cluster.json")));
@SuppressWarnings("unchecked")
final Map<String, String> clusterMap = new ObjectMapper().readValue(clusterJson, Map.class);

View File

@ -14,7 +14,7 @@
</property>
</parameters>
<start to="DropAndCreateIndex"/>
<start to="indexScholix"/>
<kill name="Kill">
<message>Action failed, error message[${wf:errorMessage(wf:lastErrorNode())}]</message>

View File

@ -0,0 +1,77 @@
{
"mappings": {
"properties": {
"blob": {
"type": "binary",
"index": false
},
"identifier": {
"type": "keyword"
},
"linkProviders": {
"type": "keyword"
},
"publicationDate": {
"type": "date"
},
"relationType": {
"type": "keyword"
},
"sourceId": {
"type": "keyword"
},
"sourcePid": {
"type": "keyword"
},
"sourcePidType": {
"type": "keyword"
},
"sourcePublisher": {
"type": "keyword"
},
"sourceSubType": {
"type": "keyword"
},
"sourceType": {
"type": "keyword"
},
"targetId": {
"type": "keyword"
},
"targetPid": {
"type": "keyword"
},
"targetPidType": {
"type": "keyword"
},
"targetPublisher": {
"type": "keyword"
},
"targetSubType": {
"type": "keyword"
},
"targetType": {
"type": "keyword"
}
}
},
"settings": {
"index": {
"refresh_interval": "600s",
"number_of_shards": "48",
"translog": {
"sync_interval": "15s",
"durability": "ASYNC"
},
"analysis": {
"analyzer": {
"analyzer_keyword": {
"filter": "lowercase",
"tokenizer": "keyword"
}
}
},
"number_of_replicas": "0"
}
}
}

View File

@ -56,7 +56,7 @@ public class XmlRecordFactoryTest {
assertNotNull(doc);
//System.out.println(doc.asXML());
// System.out.println(doc.asXML());
assertEquals("0000-0001-9613-6638", doc.valueOf("//creator[@rank = '1']/@orcid"));
assertEquals("0000-0001-9613-6639", doc.valueOf("//creator[@rank = '1']/@orcid_pending"));

View File

@ -0,0 +1,146 @@
package eu.dnetlib.dhp.sx;
import static eu.dnetlib.dhp.sx.provision.DropAndCreateESIndex.APPLICATION_JSON;
import static eu.dnetlib.dhp.sx.provision.DropAndCreateESIndex.STATUS_CODE_TEXT;
import java.io.*;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Base64;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;
import org.apache.commons.codec.binary.Base64InputStream;
import org.apache.commons.codec.binary.Base64OutputStream;
import org.apache.commons.io.IOUtils;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpDelete;
import org.apache.http.client.methods.HttpPut;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.spark.SparkConf;
import org.apache.spark.api.java.JavaRDD;
import org.apache.spark.api.java.JavaSparkContext;
import org.apache.spark.sql.SparkSession;
import org.elasticsearch.spark.rdd.api.java.JavaEsSpark;
import org.junit.jupiter.api.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.fasterxml.jackson.databind.ObjectMapper;
import eu.dnetlib.dhp.schema.sx.scholix.Scholix;
import eu.dnetlib.dhp.schema.sx.scholix.ScholixFlat;
import eu.dnetlib.dhp.sx.graph.scholix.ScholixUtils;
import eu.dnetlib.dhp.sx.provision.DropAndCreateESIndex;
import eu.dnetlib.dhp.sx.provision.SparkIndexCollectionOnES;
public class FlatIndexTest {
@Test
public void dropAndCreateIndex() {
Logger log = LoggerFactory.getLogger(getClass().getName());
final String url = "http://localhost:9200/dli_scholix";
try (CloseableHttpClient client = HttpClients.createDefault()) {
HttpDelete delete = new HttpDelete(url);
CloseableHttpResponse response = client.execute(delete);
log.info("deleting Index SCHOLIX");
log.info(STATUS_CODE_TEXT, response.getStatusLine());
} catch (IOException e) {
throw new RuntimeException(e);
}
try (CloseableHttpClient client = HttpClients.createDefault()) {
final String scholixConf = IOUtils
.toString(
Objects
.requireNonNull(
DropAndCreateESIndex.class
.getResourceAsStream("/eu/dnetlib/dhp/sx/provision/scholix_index_flat.json")));
log.info("creating Index SCHOLIX");
final HttpPut put = new HttpPut(url);
final StringEntity entity = new StringEntity(scholixConf);
put.setEntity(entity);
put.setHeader("Accept", APPLICATION_JSON);
put.setHeader("Content-type", APPLICATION_JSON);
final CloseableHttpResponse response = client.execute(put);
log.info(STATUS_CODE_TEXT, response.getStatusLine());
} catch (IOException e) {
throw new RuntimeException(e);
}
}
private String compressString(final String input) {
try {
ByteArrayOutputStream os = new ByteArrayOutputStream();
Base64OutputStream b64os = new Base64OutputStream(os);
GZIPOutputStream gzip = new GZIPOutputStream(b64os);
gzip.write(input.getBytes(StandardCharsets.UTF_8));
gzip.close();
b64os.close();
return new String(os.toByteArray(), StandardCharsets.UTF_8);
} catch (Throwable t) {
t.printStackTrace();
return null;
}
}
private String uncompress(final String compressed) throws Exception {
Base64InputStream bis = new Base64InputStream(new ByteArrayInputStream(compressed.getBytes()));
GZIPInputStream gzip = new GZIPInputStream(bis);
return IOUtils.toString(gzip);
}
public void testFeedIndex() throws Exception {
InputStream gzipStream = new GZIPInputStream(
getClass().getResourceAsStream("/eu/dnetlib/dhp/sx/provision/scholix_dump.gz"));
Reader decoder = new InputStreamReader(gzipStream, "UTF-8");
BufferedReader buffered = new BufferedReader(decoder);
final ObjectMapper mapper = new ObjectMapper();
GZIPOutputStream gzip = new GZIPOutputStream(Files.newOutputStream(Paths.get("/tmp/scholix_flat.gz")));
String line = buffered.readLine();
while (line != null) {
final Scholix s = mapper.readValue(line, Scholix.class);
final ScholixFlat flat = ScholixUtils.flattenizeScholix(s, compressString(line));
gzip.write(mapper.writeValueAsString(flat).concat("\n").getBytes(StandardCharsets.UTF_8));
line = buffered.readLine();
}
gzip.close();
SparkConf conf = new SparkConf()
.setAppName(SparkIndexCollectionOnES.class.getSimpleName())
.setMaster("local[*]");
final SparkSession spark = SparkSession.builder().config(conf).getOrCreate();
try (final JavaSparkContext sc = new JavaSparkContext(spark.sparkContext())) {
JavaRDD<String> inputRdd = sc.textFile("/tmp/scholix_flat.gz");
Map<String, String> esCfg = new HashMap<>();
esCfg.put("es.nodes", "localhost");
esCfg.put("es.mapping.id", "identifier");
esCfg.put("es.batch.write.retry.count", "8");
esCfg.put("es.batch.write.retry.wait", "60s");
esCfg.put("es.batch.size.entries", "200");
esCfg.put("es.nodes.wan.only", "true");
JavaEsSpark.saveJsonToEs(inputRdd, "scholix", esCfg);
}
}
}

View File

@ -0,0 +1,46 @@
package eu.dnetlib.dhp.sx.provision
import org.apache.spark.SparkConf
import org.apache.spark.sql.SparkSession
import org.junit.Before
import org.junit.jupiter.api.{Test}
class ScholixFlatTest {
var spark: SparkSession = null
def initSpark(): Unit = {
if (spark != null)
return
println("SONO QUI")
val conf = new SparkConf
conf.setAppName(getClass.getSimpleName)
conf.setMaster("local[*]")
conf.set("spark.driver.host", "localhost")
conf.set("hive.metastore.local", "true")
conf.set("spark.ui.enabled", "false")
spark = SparkSession
.builder()
.appName(getClass.getSimpleName)
.config(conf)
.getOrCreate()
}
def after(): Unit = {
spark.stop()
}
@Test
def testScholixConversion(): Unit = {
initSpark()
val p = getClass.getResource("/eu/dnetlib/dhp/sx/provision/scholix_dump.zip").getPath
val t = spark.read.text(p).count
println(s"total =$t")
}
}

View File

@ -807,7 +807,7 @@
<mockito-core.version>3.3.3</mockito-core.version>
<mongodb.driver.version>3.4.2</mongodb.driver.version>
<vtd.version>[2.12,3.0)</vtd.version>
<dhp-schemas.version>[3.16.0]</dhp-schemas.version>
<dhp-schemas.version>[3.16.1-SNAPSHOT]</dhp-schemas.version>
<dnet-actionmanager-api.version>[4.0.3]</dnet-actionmanager-api.version>
<dnet-actionmanager-common.version>[6.0.5]</dnet-actionmanager-common.version>
<dnet-openaire-broker-common.version>[3.1.6]</dnet-openaire-broker-common.version>