Avoid overflow in VertexRDD.count()

ankurdave · ankurdave · commit 641f46899b0c · 2014-08-23T16:49:41.000-07:00
diff --git a/graphx/src/main/scala/org/apache/spark/graphx/VertexRDD.scala b/graphx/src/main/scala/org/apache/spark/graphx/VertexRDD.scala
@@ -108,7 +108,7 @@ class VertexRDD[@specialized VD: ClassTag](
 
   /** The number of vertices in the RDD. */
   override def count(): Long = {
-    partitionsRDD.map(_.size).reduce(_ + _)
+    partitionsRDD.map(_.size.toLong).reduce(_ + _)
   }
 
   /**

Original file line number	Diff line number	Diff line change
`@@ -108,7 +108,7 @@ class VertexRDD[@specialized VD: ClassTag](`
`108`	`108`
`109`	`109`	`/** The number of vertices in the RDD. */`
`110`	`110`	`override def count(): Long = {`
`111`		`- partitionsRDD.map(_.size).reduce(_ + _)`
	`111`	`+ partitionsRDD.map(_.size.toLong).reduce(_ + _)`
`112`	`112`	`}`
`113`	`113`
`114`	`114`	`/**`