Workaround for OrientDB Light edges

Auginte · Jan 13, 2016 · a22024b · a22024b
1 parent b2a8e50
commit a22024b
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/auginte-distribution/src/main/scala/com/auginte/distribution/orientdb/ReferWrapper.scala b/auginte-distribution/src/main/scala/com/auginte/distribution/orientdb/ReferWrapper.scala
@@ -73,9 +73,9 @@ trait ReferWrapper[A, Self <: ReferWrapper[A, Self]] { self: Self =>
         """.stripMargin
         val query = new OCommandSQL(sql.replace("\n", " "))
         val parameters = JMap()
-        persisted.getGraph.command(query).execute[Vertices](parameters).map { row =>
-          val from = row.getProperty[ODocument]("edgeFrom")
-          val to = row.getProperty[ODocument]("edgeTo")
+        persisted.getGraph.command(query).execute[Vertices](parameters).map { row: Vertex =>
+          val from = getEdge(row, "edgeFrom")
+          val to = getEdge(row, "edgeTo")
           val distance = row.getProperty[Int]("$depth")
           cache(to) match {
             case Some(cachedTo) => cache(from) match {
@@ -88,6 +88,13 @@ trait ReferWrapper[A, Self <: ReferWrapper[A, Self]] { self: Self =>
       case _ => EmptyDistantRepresentationIterable
     }
 
+  // Woraround for different light edge configuration
+  private def getEdge(vertex: Vertex, edge: String): ODocument = try {
+    vertex.getProperty[ODocument](edge)
+  } catch {
+    case e: ClassCastException => vertex.getProperty[OrientVertex](edge).getRecord
+  }
+
   def derivedRepresentations(implicit cache: R.Cached = R.defaultCache): Iterable[RepresentationWrapper] =
     storage.persisted match {
       case None => EmptyRepresentationStorageIterable