diff options
author | Ankur Dave <ankurdave@gmail.com> | 2011-04-15 13:43:29 -0700 |
---|---|---|
committer | Ankur Dave <ankurdave@gmail.com> | 2011-05-03 15:40:31 -0700 |
commit | 1c8ca0ebe1537c8f424722294794a66ff123f132 (patch) | |
tree | f4375a4cbe417dc440d7bd0a2ecea458f8bddea0 /bagel/src/main | |
parent | c5b3ea755ff8a69aa39dd6e46d57cbe9d5bcbcae (diff) | |
download | spark-1c8ca0ebe1537c8f424722294794a66ff123f132.tar.gz spark-1c8ca0ebe1537c8f424722294794a66ff123f132.tar.bz2 spark-1c8ca0ebe1537c8f424722294794a66ff123f132.zip |
Add Bagel test suite
Note: This test suite currently fails for the same reason that the
Spark Core test suite fails: Spark currently seems to have a bug where
any test after the first one fails.
Diffstat (limited to 'bagel/src/main')
-rw-r--r-- | bagel/src/main/scala/bagel/Pregel.scala | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/bagel/src/main/scala/bagel/Pregel.scala b/bagel/src/main/scala/bagel/Pregel.scala index 5ef398d783..67bc582fd1 100644 --- a/bagel/src/main/scala/bagel/Pregel.scala +++ b/bagel/src/main/scala/bagel/Pregel.scala @@ -75,6 +75,14 @@ object Pregel extends Logging { run(sc, newVerts, newMsgs, createCombiner, mergeMsg, mergeCombiners, numSplits, superstep + 1)(compute) } } + + def defaultCreateCombiner[M <: Message](msg: M): ArrayBuffer[M] = ArrayBuffer(msg) + def defaultMergeMsg[M <: Message](combiner: ArrayBuffer[M], msg: M): ArrayBuffer[M] = + combiner += msg + def defaultMergeCombiners[M <: Message](a: ArrayBuffer[M], b: ArrayBuffer[M]): ArrayBuffer[M] = + a ++= b + def defaultCompute[V <: Vertex, M <: Message](self: V, msgs: Option[ArrayBuffer[M]], superstep: Int): (V, Iterable[M]) = + (self, List()) } /** |