95 lines
2.6 KiB
Java
95 lines
2.6 KiB
Java
|
|
package eu.dnetlib.dhp.oa.provision.model;
|
|
|
|
import java.io.Serializable;
|
|
import java.util.Map;
|
|
import java.util.Optional;
|
|
|
|
import com.google.common.base.Objects;
|
|
import com.google.common.collect.ComparisonChain;
|
|
import com.google.common.collect.Maps;
|
|
|
|
import eu.dnetlib.dhp.schema.common.ModelConstants;
|
|
import eu.dnetlib.dhp.schema.oaf.Relation;
|
|
|
|
public class SortableRelationKey implements Comparable<SortableRelationKey>, Serializable {
|
|
|
|
private static final Map<Relation.SUBRELTYPE, Integer> weights = Maps.newHashMap();
|
|
|
|
static {
|
|
//TODO Claudio check why we need to have SUBRELTYPE AND RELTYPE
|
|
weights.put(Relation.SUBRELTYPE.participation, 0);
|
|
weights.put(Relation.SUBRELTYPE.outcome, 1);
|
|
weights.put(Relation.SUBRELTYPE.affiliation, 2);
|
|
weights.put(Relation.SUBRELTYPE.dedup, 3);
|
|
//TODO COMMENTED BUT SHOULD BE REPLACED WITH RELATIONSHIPS??
|
|
// weights.put(ModelConstants.PUBLICATION_DATASET, 4);
|
|
weights.put(Relation.SUBRELTYPE.supplement, 5);
|
|
weights.put(Relation.SUBRELTYPE.review, 6);
|
|
weights.put(Relation.SUBRELTYPE.relationship, 7);
|
|
weights.put(Relation.SUBRELTYPE.part, 8);
|
|
weights.put(Relation.SUBRELTYPE.provision, 9);
|
|
weights.put(Relation.SUBRELTYPE.version, 10);
|
|
weights.put(Relation.SUBRELTYPE.similarity, 11);
|
|
weights.put(Relation.SUBRELTYPE.citation, 12);
|
|
}
|
|
|
|
private static final long serialVersionUID = 3232323;
|
|
|
|
private String groupingKey;
|
|
|
|
private Relation.SUBRELTYPE subRelType;
|
|
|
|
public static SortableRelationKey create(Relation r, String groupingKey) {
|
|
SortableRelationKey sr = new SortableRelationKey();
|
|
sr.setGroupingKey(groupingKey);
|
|
sr.setSubRelType(r.getSubRelType());
|
|
return sr;
|
|
}
|
|
|
|
@Override
|
|
public boolean equals(Object o) {
|
|
if (this == o)
|
|
return true;
|
|
if (o == null || getClass() != o.getClass())
|
|
return false;
|
|
SortableRelationKey that = (SortableRelationKey) o;
|
|
return getGroupingKey().equals(that.getGroupingKey());
|
|
}
|
|
|
|
@Override
|
|
public int hashCode() {
|
|
return Objects.hashCode(getGroupingKey());
|
|
}
|
|
|
|
@Override
|
|
public int compareTo(SortableRelationKey o) {
|
|
return ComparisonChain
|
|
.start()
|
|
.compare(getGroupingKey(), o.getGroupingKey())
|
|
.compare(getWeight(this), getWeight(o))
|
|
.result();
|
|
}
|
|
|
|
private Integer getWeight(SortableRelationKey o) {
|
|
return Optional.ofNullable(weights.get(o.getSubRelType())).orElse(Integer.MAX_VALUE);
|
|
}
|
|
|
|
public Relation.SUBRELTYPE getSubRelType() {
|
|
return subRelType;
|
|
}
|
|
|
|
public void setSubRelType(Relation.SUBRELTYPE subRelType) {
|
|
this.subRelType = subRelType;
|
|
}
|
|
|
|
public String getGroupingKey() {
|
|
return groupingKey;
|
|
}
|
|
|
|
public void setGroupingKey(String groupingKey) {
|
|
this.groupingKey = groupingKey;
|
|
}
|
|
|
|
}
|