50 lines
1.5 KiB
Scala
50 lines
1.5 KiB
Scala
package eu.dnetlib.dhp.collection
|
|
|
|
import eu.dnetlib.dhp.schema.common.ModelSupport
|
|
import eu.dnetlib.dhp.schema.oaf.{Oaf, OafEntity, Relation}
|
|
|
|
object CollectionUtils {
|
|
|
|
/**
|
|
* This method in pipeline to the transformation phase,
|
|
* generates relations in both verse, typically it should be a phase of flatMap
|
|
*
|
|
* @param i input OAF
|
|
* @return
|
|
* If the input OAF is an entity -> List(i)
|
|
* If the input OAF is a relation -> List(relation, inverseRelation)
|
|
*
|
|
*/
|
|
|
|
def fixRelations(i: Oaf): List[Oaf] = {
|
|
if (i.isInstanceOf[OafEntity])
|
|
return List(i)
|
|
else {
|
|
val r: Relation = i.asInstanceOf[Relation]
|
|
val currentRel = ModelSupport.findRelation(r.getRelClass)
|
|
if (currentRel != null) {
|
|
|
|
// Cleaning relation
|
|
r.setRelType(currentRel.getRelType)
|
|
r.setSubRelType(currentRel.getSubReltype)
|
|
r.setRelClass(currentRel.getRelClass)
|
|
val inverse = new Relation
|
|
inverse.setSource(r.getTarget)
|
|
inverse.setTarget(r.getSource)
|
|
inverse.setRelType(currentRel.getRelType)
|
|
inverse.setSubRelType(currentRel.getSubReltype)
|
|
inverse.setRelClass(currentRel.getInverseRelClass)
|
|
inverse.setCollectedfrom(r.getCollectedfrom)
|
|
inverse.setDataInfo(r.getDataInfo)
|
|
inverse.setProperties(r.getProperties)
|
|
inverse.setLastupdatetimestamp(r.getLastupdatetimestamp)
|
|
inverse.setValidated(r.getValidated)
|
|
inverse.setValidationDate(r.getValidationDate)
|
|
return List(r, inverse)
|
|
}
|
|
}
|
|
List()
|
|
}
|
|
|
|
}
|