318 lines
11 KiB
Java
318 lines
11 KiB
Java
|
|
package eu.dnetlib.dhp.schema.oaf.utils;
|
|
|
|
import static com.google.common.base.Preconditions.checkArgument;
|
|
import static eu.dnetlib.dhp.schema.common.ModelConstants.*;
|
|
|
|
import java.io.Serializable;
|
|
import java.nio.charset.StandardCharsets;
|
|
import java.security.MessageDigest;
|
|
import java.util.*;
|
|
import java.util.function.Function;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.Stream;
|
|
|
|
import org.apache.commons.codec.binary.Hex;
|
|
import org.apache.commons.lang3.StringUtils;
|
|
|
|
import com.google.common.collect.HashBiMap;
|
|
import com.google.common.collect.Maps;
|
|
|
|
import eu.dnetlib.dhp.schema.oaf.*;
|
|
import eu.dnetlib.dhp.schema.oaf.common.ModelSupport;
|
|
|
|
/**
|
|
* Factory class for OpenAIRE identifiers in the Graph
|
|
*/
|
|
public class IdentifierFactory implements Serializable {
|
|
|
|
public static final String ID_SEPARATOR = "::";
|
|
public static final String ID_PREFIX_SEPARATOR = "|";
|
|
|
|
public static final int ID_PREFIX_LEN = 12;
|
|
|
|
/**
|
|
* Declares the associations PID_TYPE -> [DATASOURCE ID, NAME] considered authoritative for that PID_TYPE.
|
|
* The id of the record (source_::id) will be rewritten as pidType_::id)
|
|
*/
|
|
public static final Map<PidType, HashBiMap<String, String>> PID_AUTHORITY = Maps.newHashMap();
|
|
|
|
static {
|
|
PID_AUTHORITY.put(PidType.doi, HashBiMap.create());
|
|
PID_AUTHORITY.get(PidType.doi).put(CROSSREF_ID, "Crossref");
|
|
PID_AUTHORITY.get(PidType.doi).put(DATACITE_ID, "Datacite");
|
|
PID_AUTHORITY.get(PidType.doi).put(ZENODO_OD_ID, "ZENODO");
|
|
PID_AUTHORITY.get(PidType.doi).put(ZENODO_R3_ID, "Zenodo");
|
|
|
|
PID_AUTHORITY.put(PidType.pmc, HashBiMap.create());
|
|
PID_AUTHORITY.get(PidType.pmc).put(EUROPE_PUBMED_CENTRAL_ID, "Europe PubMed Central");
|
|
PID_AUTHORITY.get(PidType.pmc).put(PUBMED_CENTRAL_ID, "PubMed Central");
|
|
|
|
PID_AUTHORITY.put(PidType.pmid, HashBiMap.create());
|
|
PID_AUTHORITY.get(PidType.pmid).put(EUROPE_PUBMED_CENTRAL_ID, "Europe PubMed Central");
|
|
PID_AUTHORITY.get(PidType.pmid).put(PUBMED_CENTRAL_ID, "PubMed Central");
|
|
|
|
PID_AUTHORITY.put(PidType.arXiv, HashBiMap.create());
|
|
PID_AUTHORITY.get(PidType.arXiv).put(ARXIV_ID, "arXiv.org e-Print Archive");
|
|
|
|
PID_AUTHORITY.put(PidType.w3id, HashBiMap.create());
|
|
PID_AUTHORITY.get(PidType.w3id).put(ROHUB_ID, "ROHub");
|
|
}
|
|
|
|
/**
|
|
* Declares the associations PID_TYPE -> [DATASOURCE ID, PID SUBSTRING] considered as delegated authority for that
|
|
* PID_TYPE. Example, Zenodo is delegated to forge DOIs that contain the 'zenodo' word.
|
|
*
|
|
* If a record with the same id (same pid) comes from 2 data sources, the one coming from a delegated source wins. E.g. Zenodo records win over those from Datacite.
|
|
* See also https://code-repo.d4science.org/D-Net/dnet-hadoop/pulls/187 and the class dhp-common/src/main/java/eu/dnetlib/dhp/schema/oaf/utils/OafMapperUtils.java
|
|
*/
|
|
public static final Map<PidType, Map<String, String>> DELEGATED_PID_AUTHORITY = Maps.newHashMap();
|
|
|
|
static {
|
|
DELEGATED_PID_AUTHORITY.put(PidType.doi, new HashMap<>());
|
|
DELEGATED_PID_AUTHORITY.get(PidType.doi).put(ZENODO_OD_ID, "zenodo");
|
|
DELEGATED_PID_AUTHORITY.get(PidType.doi).put(ZENODO_R3_ID, "zenodo");
|
|
DELEGATED_PID_AUTHORITY.put(PidType.w3id, new HashMap<>());
|
|
DELEGATED_PID_AUTHORITY.get(PidType.w3id).put(ROHUB_ID, "ro-id");
|
|
}
|
|
|
|
/**
|
|
* Declares the associations PID_TYPE -> [DATASOURCE ID, NAME] whose records are considered enrichment for the graph.
|
|
* Their OpenAIRE ID is built from the declared PID type. Are merged with their corresponding record, identified by
|
|
* the same OpenAIRE id.
|
|
*/
|
|
public static final Map<PidType, HashBiMap<String, String>> ENRICHMENT_PROVIDER = Maps.newHashMap();
|
|
|
|
static {
|
|
ENRICHMENT_PROVIDER.put(PidType.doi, HashBiMap.create());
|
|
ENRICHMENT_PROVIDER.get(PidType.doi).put(OPEN_APC_ID, OPEN_APC_NAME);
|
|
}
|
|
|
|
public static Set<String> delegatedAuthorityDatasourceIds() {
|
|
return DELEGATED_PID_AUTHORITY
|
|
.values()
|
|
.stream()
|
|
.flatMap(m -> m.keySet().stream())
|
|
.collect(Collectors.toCollection(HashSet::new));
|
|
}
|
|
|
|
public static List<StructuredProperty> getPids(List<StructuredProperty> pid, KeyValue collectedFrom) {
|
|
return pidFromInstance(pid, collectedFrom, true).distinct().collect(Collectors.toList());
|
|
}
|
|
|
|
public static <T extends Result> String createDOIBoostIdentifier(T entity) {
|
|
if (entity == null)
|
|
return null;
|
|
|
|
StructuredProperty pid = null;
|
|
if (entity.getPid() != null) {
|
|
pid = entity
|
|
.getPid()
|
|
.stream()
|
|
.filter(Objects::nonNull)
|
|
.filter(s -> s.getQualifier() != null && "doi".equalsIgnoreCase(s.getQualifier().getClassid()))
|
|
.filter(CleaningFunctions::pidFilter)
|
|
.findAny()
|
|
.orElse(null);
|
|
} else {
|
|
if (entity.getInstance() != null) {
|
|
pid = entity
|
|
.getInstance()
|
|
.stream()
|
|
.filter(i -> i.getPid() != null)
|
|
.flatMap(i -> i.getPid().stream())
|
|
.filter(CleaningFunctions::pidFilter)
|
|
.findAny()
|
|
.orElse(null);
|
|
}
|
|
}
|
|
if (pid != null)
|
|
return idFromPid(entity, pid, true);
|
|
return null;
|
|
}
|
|
|
|
/**
|
|
* Creates an identifier from the most relevant PID (if available) provided by a known PID authority in the given
|
|
* entity T. Returns entity.id when none of the PIDs meet the selection criteria is available.
|
|
*
|
|
* @param entity the entity providing PIDs and a default ID.
|
|
* @param <T> the specific entity type. Currently Organization and Result subclasses are supported.
|
|
* @param md5 indicates whether should hash the PID value or not.
|
|
* @return an identifier from the most relevant PID, entity.id otherwise
|
|
*/
|
|
public static <T extends Entity> String createIdentifier(T entity, boolean md5) {
|
|
|
|
checkArgument(StringUtils.isNoneBlank(entity.getId()), "missing entity identifier");
|
|
|
|
final Map<String, Set<StructuredProperty>> pids = extractPids(entity);
|
|
|
|
return pids
|
|
.values()
|
|
.stream()
|
|
.flatMap(Set::stream)
|
|
.min(new PidComparator<>(entity))
|
|
.map(
|
|
min -> Optional
|
|
.ofNullable(pids.get(min.getQualifier().getClassid()))
|
|
.map(
|
|
p -> p
|
|
.stream()
|
|
.sorted(new PidValueComparator())
|
|
.findFirst()
|
|
.map(s -> idFromPid(entity, s, md5))
|
|
.orElseGet(entity::getId))
|
|
.orElseGet(entity::getId))
|
|
.orElseGet(entity::getId);
|
|
}
|
|
|
|
private static <T extends Entity> Map<String, Set<StructuredProperty>> extractPids(T entity) {
|
|
if (entity instanceof Result) {
|
|
return Optional
|
|
.ofNullable(((Result) entity).getInstance())
|
|
.map(IdentifierFactory::mapPids)
|
|
.orElse(new HashMap<>());
|
|
} else {
|
|
return entity
|
|
.getPid()
|
|
.stream()
|
|
.map(CleaningFunctions::normalizePidValue)
|
|
.filter(CleaningFunctions::pidFilter)
|
|
.collect(
|
|
Collectors
|
|
.groupingBy(
|
|
p -> p.getQualifier().getClassid(),
|
|
Collectors.mapping(p -> p, Collectors.toCollection(HashSet::new))));
|
|
}
|
|
}
|
|
|
|
private static Map<String, Set<StructuredProperty>> mapPids(List<Instance> instance) {
|
|
return instance
|
|
.stream()
|
|
.map(i -> pidFromInstance(i.getPid(), i.getCollectedfrom(), false))
|
|
.flatMap(Function.identity())
|
|
.collect(
|
|
Collectors
|
|
.groupingBy(
|
|
p -> p.getQualifier().getClassid(),
|
|
Collectors.mapping(p -> p, Collectors.toCollection(HashSet::new))));
|
|
}
|
|
|
|
private static Stream<StructuredProperty> pidFromInstance(List<StructuredProperty> pid, KeyValue collectedFrom,
|
|
boolean mapHandles) {
|
|
return Optional
|
|
.ofNullable(pid)
|
|
.map(
|
|
pp -> pp
|
|
.stream()
|
|
// filter away PIDs provided by a DS that is not considered an authority for the
|
|
// given PID Type
|
|
.filter(p -> shouldFilterPidByCriteria(collectedFrom, p, mapHandles))
|
|
.map(CleaningFunctions::normalizePidValue)
|
|
.filter(p -> isNotFromDelegatedAuthority(collectedFrom, p))
|
|
.filter(CleaningFunctions::pidFilter))
|
|
.orElse(Stream.empty());
|
|
}
|
|
|
|
private static boolean shouldFilterPidByCriteria(KeyValue collectedFrom, StructuredProperty p, boolean mapHandles) {
|
|
final PidType pType = PidType.tryValueOf(p.getQualifier().getClassid());
|
|
|
|
if (Objects.isNull(collectedFrom)) {
|
|
return false;
|
|
}
|
|
|
|
boolean isEnrich = Optional
|
|
.ofNullable(ENRICHMENT_PROVIDER.get(pType))
|
|
.map(
|
|
enrich -> enrich.containsKey(collectedFrom.getKey())
|
|
|| enrich.containsValue(collectedFrom.getValue()))
|
|
.orElse(false);
|
|
|
|
boolean isAuthority = Optional
|
|
.ofNullable(PID_AUTHORITY.get(pType))
|
|
.map(
|
|
authorities -> authorities.containsKey(collectedFrom.getKey())
|
|
|| authorities.containsValue(collectedFrom.getValue()))
|
|
.orElse(false);
|
|
|
|
return (mapHandles && pType.equals(PidType.handle)) || isEnrich || isAuthority;
|
|
}
|
|
|
|
private static boolean isNotFromDelegatedAuthority(KeyValue collectedFrom, StructuredProperty p) {
|
|
final PidType pType = PidType.tryValueOf(p.getQualifier().getClassid());
|
|
|
|
final Map<String, String> da = DELEGATED_PID_AUTHORITY.get(pType);
|
|
if (Objects.isNull(da)) {
|
|
return true;
|
|
}
|
|
if (!da.containsKey(collectedFrom.getKey())) {
|
|
return true;
|
|
}
|
|
return StringUtils.contains(p.getValue(), da.get(collectedFrom.getKey()));
|
|
}
|
|
|
|
/**
|
|
* @see {@link IdentifierFactory#createIdentifier(Entity, boolean)}
|
|
*/
|
|
public static <T extends Entity> String createIdentifier(T entity) {
|
|
|
|
return createIdentifier(entity, true);
|
|
}
|
|
|
|
private static <T extends Entity> String idFromPid(T entity, StructuredProperty s, boolean md5) {
|
|
return idFromPid(ModelSupport.getIdPrefix(entity.getClass()), s.getQualifier().getClassid(), s.getValue(), md5);
|
|
}
|
|
|
|
public static String idFromPid(String numericPrefix, String pidType, String pidValue, boolean md5) {
|
|
return new StringBuilder()
|
|
.append(numericPrefix)
|
|
.append(ID_PREFIX_SEPARATOR)
|
|
.append(createPrefix(pidType))
|
|
.append(ID_SEPARATOR)
|
|
.append(md5 ? ModelSupport.md5(pidValue) : pidValue)
|
|
.toString();
|
|
}
|
|
|
|
// create the prefix (length = 12)
|
|
private static String createPrefix(String pidType) {
|
|
StringBuilder prefix = new StringBuilder(StringUtils.left(pidType, ID_PREFIX_LEN));
|
|
while (prefix.length() < ID_PREFIX_LEN) {
|
|
prefix.append("_");
|
|
}
|
|
return prefix.substring(0, ID_PREFIX_LEN);
|
|
}
|
|
|
|
public static String createOpenaireId(
|
|
final int prefix,
|
|
final String originalId,
|
|
final boolean to_md5) {
|
|
if (StringUtils.isBlank(originalId)) {
|
|
return null;
|
|
} else if (to_md5) {
|
|
final String nsPrefix = StringUtils.substringBefore(originalId, "::");
|
|
final String rest = StringUtils.substringAfter(originalId, "::");
|
|
return String.format("%s|%s::%s", prefix, nsPrefix, ModelSupport.md5(rest));
|
|
} else {
|
|
return String.format("%s|%s", prefix, originalId);
|
|
}
|
|
}
|
|
|
|
public static String createOpenaireId(
|
|
final String type,
|
|
final String originalId,
|
|
final boolean to_md5) {
|
|
switch (type) {
|
|
case "datasource":
|
|
return createOpenaireId(10, originalId, to_md5);
|
|
case "organization":
|
|
return createOpenaireId(20, originalId, to_md5);
|
|
case "person":
|
|
return createOpenaireId(30, originalId, to_md5);
|
|
case "project":
|
|
return createOpenaireId(40, originalId, to_md5);
|
|
default:
|
|
return createOpenaireId(50, originalId, to_md5);
|
|
}
|
|
}
|
|
|
|
}
|