Merge pull request #97 from dcrankshaw/fix-rddtop

Added BoundedPriorityQueue kryo registrator. Fixes top issue.
This commit is contained in:
Reynold Xin 2013-12-11 22:30:54 -08:00
commit ce6ca4ea61

View file

@ -5,6 +5,7 @@ import com.esotericsoftware.kryo.Kryo
import org.apache.spark.graph.impl._
import org.apache.spark.serializer.KryoRegistrator
import org.apache.spark.util.collection.BitSet
import org.apache.spark.util.BoundedPriorityQueue
class GraphKryoRegistrator extends KryoRegistrator {
@ -19,6 +20,7 @@ class GraphKryoRegistrator extends KryoRegistrator {
kryo.register(classOf[VertexIdToIndexMap])
kryo.register(classOf[VertexAttributeBlock[Object]])
kryo.register(classOf[PartitionStrategy])
kryo.register(classOf[BoundedPriorityQueue[Object]])
// This avoids a large number of hash table lookups.
kryo.setReferences(false)