41 lines
1.3 KiB
Scala
41 lines
1.3 KiB
Scala
package eu.dnetlib.dhp.enrich.orcid
|
|
|
|
import eu.dnetlib.dhp.schema.common.ModelConstants
|
|
import eu.dnetlib.dhp.schema.oaf.{Author, Publication}
|
|
import eu.dnetlib.dhp.schema.sx.OafUtils
|
|
import org.apache.spark.sql.Row
|
|
|
|
import scala.collection.JavaConverters._
|
|
|
|
object AuthorEnricher extends Serializable {
|
|
|
|
def createAuthor(givenName: String, familyName: String, orcid: String): Author = {
|
|
val a = new Author
|
|
a.setName(givenName)
|
|
a.setSurname(familyName)
|
|
a.setFullname(s"$givenName $familyName")
|
|
val pid = OafUtils.createSP(orcid, ModelConstants.ORCID, ModelConstants.ORCID)
|
|
pid.setDataInfo(OafUtils.generateDataInfo())
|
|
pid.getDataInfo.setProvenanceaction(OafUtils.createQualifier("ORCID_ENRICHMENT", "ORCID_ENRICHMENT"))
|
|
a.setPid(List(pid).asJava)
|
|
a
|
|
}
|
|
|
|
def toOAFAuthor(r: Row): java.util.List[Author] = {
|
|
r.getList[Row](1)
|
|
.asScala
|
|
.map(s => createAuthor(s.getAs[String]("givenName"), s.getAs[String]("familyName"), s.getAs[String]("orcid")))
|
|
.toList
|
|
.asJava
|
|
}
|
|
|
|
// def enrichAuthor(p:Publication,r:Row): Unit = {
|
|
// val k:Map[String, OAuthor] =r.getList[Row](1).asScala.map(s => (s.getAs[String]("orcid"), OAuthor(s.getAs[String]("givenName") ,s.getAs[String]("familyName") ))).groupBy(_._1).mapValues(_.map(_._2).head)
|
|
// println(k)
|
|
//
|
|
//
|
|
//
|
|
// }
|
|
|
|
}
|