added dictionary getter

apache · Feb 11, 2015 · 80d9b72 · 80d9b72
1 parent cbc720f
commit 80d9b72
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/graphx/src/main/scala/org/apache/spark/graphx/loaders/RDFLoader.scala b/graphx/src/main/scala/org/apache/spark/graphx/loaders/RDFLoader.scala
@@ -97,6 +97,16 @@ object RDFLoader extends Logging {
     return graph // so far
   } // end of edgeListFile
 
+
+  /**
+   * Gets dictionary for entities (not properties) from the provided graph.
+   */
+  def getdictionary(graph:Graph[String,String]): RDD[(Long,String)] = {
+    val vertices = graph.vertices.map(x => (x._1.toLong, x._2))
+    return vertices
+  }
+
+
   /**
    *   Implements a simple hashing function for Strings
    */