47 lines
1.3 KiB
Java
47 lines
1.3 KiB
Java
|
|
package eu.dnetlib.dhp.oa.dedup;
|
|
|
|
import static org.apache.commons.lang3.StringUtils.substringAfter;
|
|
import static org.apache.commons.lang3.StringUtils.substringBefore;
|
|
|
|
import java.io.Serializable;
|
|
import java.util.List;
|
|
|
|
import eu.dnetlib.dhp.oa.dedup.model.Identifier;
|
|
import eu.dnetlib.dhp.schema.oaf.OafEntity;
|
|
import eu.dnetlib.dhp.schema.oaf.utils.PidType;
|
|
|
|
public class IdGenerator implements Serializable {
|
|
|
|
// pick the best pid from the list (consider date and pidtype)
|
|
public static <T extends OafEntity> String generate(List<Identifier<T>> pids, String defaultID) {
|
|
if (pids == null || pids.size() == 0)
|
|
return defaultID;
|
|
|
|
Identifier<T> bp = pids
|
|
.stream()
|
|
.min(Identifier::compareTo)
|
|
.get();
|
|
|
|
String prefix = substringBefore(bp.getOriginalID(), "|");
|
|
String ns = substringBefore(substringAfter(bp.getOriginalID(), "|"), "::");
|
|
String suffix = substringAfter(bp.getOriginalID(), "::");
|
|
|
|
final String pidType = substringBefore(ns, "_");
|
|
if (PidType.isValid(pidType)) {
|
|
return prefix + "|" + dedupify(ns) + "::" + suffix;
|
|
} else {
|
|
return prefix + "|dedup_wf_001::" + suffix;
|
|
}
|
|
}
|
|
|
|
private static String dedupify(String ns) {
|
|
StringBuilder prefix = new StringBuilder(substringBefore(ns, "_")).append("_dedup");
|
|
while (prefix.length() < 12) {
|
|
prefix.append("_");
|
|
}
|
|
return prefix.substring(0, 12);
|
|
}
|
|
|
|
}
|