dnet-hadoop/dhp-schemas/src/main/java/eu/dnetlib/dhp/schema/oaf/Relation.java

package eu.dnetlib.dhp.schema.oaf;

import java.util.*;
import java.util.stream.Collectors;

public class Relation extends Oaf {

    private String relType;

    private String subRelType;

    private String relClass;

    private String source;

    private String target;

    private List<KeyValue> collectedFrom;

    public String getRelType() {
        return relType;
    }

    public void setRelType(String relType) {
        this.relType = relType;
    }

    public String getSubRelType() {
        return subRelType;
    }

    public void setSubRelType(String subRelType) {
        this.subRelType = subRelType;
    }

    public String getRelClass() {
        return relClass;
    }

    public void setRelClass(String relClass) {
        this.relClass = relClass;
    }

    public String getSource() {
        return source;
    }

    public void setSource(String source) {
        this.source = source;
    }

    public String getTarget() {
        return target;
    }

    public void setTarget(String target) {
        this.target = target;
    }

    public List<KeyValue> getCollectedFrom() {
        return collectedFrom;
    }

    public void setCollectedFrom(List<KeyValue> collectedFrom) {
        this.collectedFrom = collectedFrom;
    }

    public void mergeFrom(Relation other) {
        this.mergeOAFDataInfo(other);
        if (other.getCollectedFrom() == null || other.getCollectedFrom().size() == 0)
            return;
        if (collectedFrom == null && other.getCollectedFrom() != null) {
            collectedFrom = other.getCollectedFrom();
            return;
        }
        if (other.getCollectedFrom() != null) {
            collectedFrom.addAll(other.getCollectedFrom());

            collectedFrom = new ArrayList<>(collectedFrom
                    .stream()
                    .collect(Collectors.toMap(KeyValue::toComparableString, x -> x, (x1, x2) -> x1))
                    .values());
        }
    }
}