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

Minor tweak to pregel semantics to give reverse edge illusion on send message.

Browse source
This commit is contained in:
Joseph E. Gonzalez 2013-04-04 17:59:08 -07:00
parent cb99fc193c
commit 0667986c8e
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
graph/src/main/scala/spark/graph/Pregel.scala
View file

@ -15,10 +15,14 @@ object Pregel {
var graph = rawGraph.cache
var i = 0
def reverseGather(vid: Vid, edge: EdgeWithVertices[VD,ED]) =
sendMsg(edge.otherVertex(vid).id, edge)
while (i < numIter) {
val msgs: RDD[(Vid, A)] =
graph.mapReduceNeighborhoodFilter(sendMsg, mergeMsg, EdgeDirection.In)
graph.mapReduceNeighborhoodFilter(reverseGather, mergeMsg, EdgeDirection.In)
def runProg(v: Vertex[VD], msg: Option[A]): VD =
if(msg.isEmpty) v.data else vprog(v, msg.get)