forked from antonis.lempesis/dnet-hadoop
implemented merge scholix
This commit is contained in:
parent
754c86f33e
commit
66efb39634
|
@ -9,7 +9,6 @@ import org.junit.jupiter.api.BeforeAll;
|
||||||
import org.junit.jupiter.api.Disabled;
|
import org.junit.jupiter.api.Disabled;
|
||||||
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
|
||||||
@Disabled
|
@Disabled
|
||||||
public class HttpConnectorTest {
|
public class HttpConnectorTest {
|
||||||
|
|
||||||
|
|
|
@ -6,11 +6,36 @@ import eu.dnetlib.dhp.provision.scholix.summary.ScholixSummary
|
||||||
import eu.dnetlib.dhp.schema.oaf.Relation
|
import eu.dnetlib.dhp.schema.oaf.Relation
|
||||||
import org.apache.commons.io.IOUtils
|
import org.apache.commons.io.IOUtils
|
||||||
import org.apache.spark.SparkConf
|
import org.apache.spark.SparkConf
|
||||||
|
import org.apache.spark.sql.expressions.Aggregator
|
||||||
import org.apache.spark.sql.{Dataset, Encoder, Encoders, SaveMode, SparkSession}
|
import org.apache.spark.sql.{Dataset, Encoder, Encoders, SaveMode, SparkSession}
|
||||||
|
|
||||||
object SparkGenerateScholixIndex {
|
object SparkGenerateScholixIndex {
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def getScholixAggregator(): Aggregator[(String, Scholix), Scholix, Scholix] = new Aggregator[(String, Scholix), Scholix, Scholix]{
|
||||||
|
|
||||||
|
override def zero: Scholix = new Scholix()
|
||||||
|
|
||||||
|
override def reduce(b: Scholix, a: (String, Scholix)): Scholix = {
|
||||||
|
b.mergeFrom(a._2)
|
||||||
|
b
|
||||||
|
}
|
||||||
|
|
||||||
|
override def merge(wx: Scholix, wy: Scholix): Scholix = {
|
||||||
|
wx.mergeFrom(wy)
|
||||||
|
wx
|
||||||
|
}
|
||||||
|
override def finish(reduction: Scholix): Scholix = reduction
|
||||||
|
|
||||||
|
override def bufferEncoder: Encoder[Scholix] =
|
||||||
|
Encoders.kryo(classOf[Scholix])
|
||||||
|
|
||||||
|
override def outputEncoder: Encoder[Scholix] =
|
||||||
|
Encoders.kryo(classOf[Scholix])
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
def main(args: Array[String]): Unit = {
|
def main(args: Array[String]): Unit = {
|
||||||
val parser = new ArgumentApplicationParser(IOUtils.toString(SparkGenerateScholixIndex.getClass.getResourceAsStream("/eu/dnetlib/dhp/provision/input_generate_summary_parameters.json")))
|
val parser = new ArgumentApplicationParser(IOUtils.toString(SparkGenerateScholixIndex.getClass.getResourceAsStream("/eu/dnetlib/dhp/provision/input_generate_summary_parameters.json")))
|
||||||
parser.parseArgument(args)
|
parser.parseArgument(args)
|
||||||
|
@ -40,7 +65,7 @@ object SparkGenerateScholixIndex {
|
||||||
|
|
||||||
(relation.getTarget, Scholix.generateScholixWithSource(summary,relation))
|
(relation.getTarget, Scholix.generateScholixWithSource(summary,relation))
|
||||||
|
|
||||||
}).write.mode(SaveMode.Overwrite).save(s"$workingDirPath/scholix_source")
|
}).repartition(6000).write.mode(SaveMode.Overwrite).save(s"$workingDirPath/scholix_source")
|
||||||
|
|
||||||
val sTarget:Dataset[(String,Scholix)] = spark.read.load(s"$workingDirPath/scholix_source").as[(String, Scholix)]
|
val sTarget:Dataset[(String,Scholix)] = spark.read.load(s"$workingDirPath/scholix_source").as[(String, Scholix)]
|
||||||
|
|
||||||
|
@ -53,9 +78,16 @@ object SparkGenerateScholixIndex {
|
||||||
scholix.generateIdentifier()
|
scholix.generateIdentifier()
|
||||||
scholix.generatelinkPublisher()
|
scholix.generatelinkPublisher()
|
||||||
scholix
|
scholix
|
||||||
}).write.mode(SaveMode.Overwrite).save(s"$workingDirPath/scholix")
|
}).repartition(6000).write.mode(SaveMode.Overwrite).save(s"$workingDirPath/scholix_r")
|
||||||
|
|
||||||
|
|
||||||
|
val finalScholix:Dataset[Scholix] = spark.read.load(s"$workingDirPath/scholix_r").as[Scholix]
|
||||||
|
|
||||||
|
finalScholix.map(d => (d.getIdentifier, d))(Encoders.tuple(Encoders.STRING, scholixEncoder))
|
||||||
|
.groupByKey(_._1)(Encoders.STRING)
|
||||||
|
.agg(getScholixAggregator().toColumn)
|
||||||
|
.map(p => p._2)
|
||||||
|
.write.mode(SaveMode.Overwrite).save(s"$workingDirPath/scholix")
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -5,6 +5,8 @@ import java.io.Serializable;
|
||||||
import java.util.*;
|
import java.util.*;
|
||||||
import java.util.stream.Collectors;
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
|
||||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||||
|
|
||||||
import eu.dnetlib.dhp.provision.scholix.summary.ScholixSummary;
|
import eu.dnetlib.dhp.provision.scholix.summary.ScholixSummary;
|
||||||
|
@ -91,13 +93,91 @@ public class Scholix implements Serializable {
|
||||||
s.setSource(ScholixResource.fromSummary(scholixSummary));
|
s.setSource(ScholixResource.fromSummary(scholixSummary));
|
||||||
|
|
||||||
s.setIdentifier(rel.getTarget());
|
s.setIdentifier(rel.getTarget());
|
||||||
// ScholixResource mockTarget = new ScholixResource();
|
|
||||||
// mockTarget.setDnetIdentifier(rel.getTarget());
|
|
||||||
// s.setTarget(mockTarget);
|
|
||||||
// s.generateIdentifier();
|
|
||||||
return s;
|
return s;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private List<ScholixEntityId> mergeScholixEntityId(final List<ScholixEntityId> a, final List<ScholixEntityId> b) {
|
||||||
|
final List<ScholixEntityId> m = new ArrayList<>(a);
|
||||||
|
if (b != null)
|
||||||
|
b.forEach(s -> {
|
||||||
|
int tt = (int) m.stream().filter(t -> t.getName().equalsIgnoreCase(s.getName())).count();
|
||||||
|
if (tt == 0) {
|
||||||
|
m.add(s);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return m;
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<ScholixIdentifier> mergeScholixIdnetifier(final List<ScholixIdentifier> a,
|
||||||
|
final List<ScholixIdentifier> b) {
|
||||||
|
final List<ScholixIdentifier> m = new ArrayList<>(a);
|
||||||
|
if (b != null)
|
||||||
|
b.forEach(s -> {
|
||||||
|
int tt = (int) m.stream().filter(t -> t.getIdentifier().equalsIgnoreCase(s.getIdentifier())).count();
|
||||||
|
if (tt == 0) {
|
||||||
|
m.add(s);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return m;
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<ScholixCollectedFrom> mergeScholixCollectedFrom(final List<ScholixCollectedFrom> a,
|
||||||
|
final List<ScholixCollectedFrom> b) {
|
||||||
|
final List<ScholixCollectedFrom> m = new ArrayList<>(a);
|
||||||
|
if (b != null)
|
||||||
|
b.forEach(s -> {
|
||||||
|
int tt = (int) m
|
||||||
|
.stream()
|
||||||
|
.filter(t -> t.getProvider().getName().equalsIgnoreCase(s.getProvider().getName()))
|
||||||
|
.count();
|
||||||
|
if (tt == 0) {
|
||||||
|
m.add(s);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return m;
|
||||||
|
}
|
||||||
|
|
||||||
|
private ScholixRelationship mergeRelationships(final ScholixRelationship a, final ScholixRelationship b) {
|
||||||
|
ScholixRelationship result = new ScholixRelationship();
|
||||||
|
result.setName(StringUtils.isEmpty(a.getName()) ? b.getName() : a.getName());
|
||||||
|
result.setInverse(StringUtils.isEmpty(a.getInverse()) ? b.getInverse() : a.getInverse());
|
||||||
|
result.setSchema(StringUtils.isEmpty(a.getSchema()) ? b.getSchema() : a.getSchema());
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
private ScholixResource mergeResource(final ScholixResource a, final ScholixResource b) {
|
||||||
|
|
||||||
|
final ScholixResource result = new ScholixResource();
|
||||||
|
result.setCollectedFrom(mergeScholixCollectedFrom(a.getCollectedFrom(), b.getCollectedFrom()));
|
||||||
|
result.setCreator(mergeScholixEntityId(a.getCreator(), b.getCreator()));
|
||||||
|
result
|
||||||
|
.setDnetIdentifier(
|
||||||
|
StringUtils.isBlank(a.getDnetIdentifier()) ? b.getDnetIdentifier() : a.getDnetIdentifier());
|
||||||
|
result.setIdentifier(mergeScholixIdnetifier(a.getIdentifier(), b.getIdentifier()));
|
||||||
|
result.setObjectType(StringUtils.isNotBlank(a.getObjectType()) ? a.getObjectType() : b.getObjectType());
|
||||||
|
result
|
||||||
|
.setObjectSubType(
|
||||||
|
StringUtils.isNotBlank(a.getObjectSubType()) ? a.getObjectSubType() : b.getObjectSubType());
|
||||||
|
result.setPublisher(mergeScholixEntityId(a.getPublisher(), b.getPublisher()));
|
||||||
|
result
|
||||||
|
.setPublicationDate(
|
||||||
|
StringUtils.isNotBlank(a.getPublicationDate()) ? a.getPublicationDate() : b.getPublicationDate());
|
||||||
|
result.setTitle(StringUtils.isNotBlank(a.getTitle()) ? a.getTitle() : b.getTitle());
|
||||||
|
return result;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
public void mergeFrom(final Scholix other) {
|
||||||
|
linkprovider = mergeScholixEntityId(linkprovider, other.getLinkprovider());
|
||||||
|
publisher = mergeScholixEntityId(publisher, other.getPublisher());
|
||||||
|
if (StringUtils.isEmpty(publicationDate))
|
||||||
|
publicationDate = other.getPublicationDate();
|
||||||
|
relationship = mergeRelationships(relationship, other.getRelationship());
|
||||||
|
source = mergeResource(source, other.getSource());
|
||||||
|
target = mergeResource(target, other.getTarget());
|
||||||
|
generateIdentifier();
|
||||||
|
}
|
||||||
|
|
||||||
public void generatelinkPublisher() {
|
public void generatelinkPublisher() {
|
||||||
Set<String> publisher = new HashSet<>();
|
Set<String> publisher = new HashSet<>();
|
||||||
if (source.getPublisher() != null)
|
if (source.getPublisher() != null)
|
||||||
|
|
|
@ -108,7 +108,7 @@
|
||||||
<arg>-m</arg> <arg>yarn-cluster</arg>
|
<arg>-m</arg> <arg>yarn-cluster</arg>
|
||||||
<arg>--workingPath</arg><arg>${workingDirPath}</arg>
|
<arg>--workingPath</arg><arg>${workingDirPath}</arg>
|
||||||
</spark>
|
</spark>
|
||||||
<ok to="DropAndCreateIndex"/>
|
<ok to="End"/>
|
||||||
<error to="Kill"/>
|
<error to="Kill"/>
|
||||||
</action>
|
</action>
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue