Skip to content
Snippets Groups Projects
Commit c1104058 authored by Ankur Dave's avatar Ankur Dave
Browse files

Move shortest path and PageRank to bagel.examples

parent 563c5e71
No related branches found
No related tags found
No related merge requests found
package bagel
package bagel.examples
import spark._
import spark.SparkContext._
import scala.math.min
import bagel._
import bagel.Pregel._
object ShortestPath {
......@@ -80,6 +81,7 @@ object ShortestPath {
}
}
@serializable
object MinCombiner extends Combiner[SPMessage, Int] {
def createCombiner(msg: SPMessage): Int =
msg.value
......
package bagel
package bagel.examples
import spark._
import spark.SparkContext._
import bagel._
import bagel.Pregel._
import scala.collection.mutable.ArrayBuffer
......@@ -75,6 +76,7 @@ object WikipediaPageRank {
}
}
@serializable
object PRCombiner extends Combiner[PRMessage, Double] {
def createCombiner(msg: PRMessage): Double =
msg.value
......@@ -103,6 +105,7 @@ object PRCombiner extends Combiner[PRMessage, Double] {
}
}
@serializable
object PRNoCombiner extends DefaultCombiner[PRMessage] {
def compute(numVertices: Long, epsilon: Double)(self: PRVertex, messages: Option[ArrayBuffer[PRMessage]], superstep: Int): (PRVertex, Iterable[PRMessage]) =
PRCombiner.compute(numVertices, epsilon)(self, messages match {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment