ODIn/spark-instrumented-optimizer
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Added a vertices method to Graph.

Browse source
This commit is contained in:
Reynold Xin 2013-04-02 01:26:20 +08:00
parent d7011b0f78
commit 25c71b185d
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
graph/src/main/scala/spark/graph/Graph.scala
View file

@ -63,6 +63,8 @@ class Graph[VD: Manifest, ED: Manifest](
private val vTable: RDD[(Vid, (VD, Array[Pid]))] = Graph.createVTable( private val vTable: RDD[(Vid, (VD, Array[Pid]))] = Graph.createVTable(
_vertices, eTable, numVertexPartitions) _vertices, eTable, numVertexPartitions)
def vertices: RDD[Vertex[VD]] = vTable.map { case(vid, (data, pids)) => new Vertex(vid, data) }
def edges: RDD[Edge[ED]] = eTable.mapPartitions { iter => iter.next._2.iterator } def edges: RDD[Edge[ED]] = eTable.mapPartitions { iter => iter.next._2.iterator }
def edgesWithVertices: RDD[EdgeWithVertices[VD, ED]] = new EdgeWithVerticesRDD(vTable, eTable) def edgesWithVertices: RDD[EdgeWithVertices[VD, ED]] = new EdgeWithVerticesRDD(vTable, eTable)