Skip to content
Snippets Groups Projects
Commit 7c9bbf17 authored by uncleGen's avatar uncleGen Committed by Ankur Dave
Browse files

[SPARK-3123][GraphX]: override the "setName" function to set EdgeRDD's name...

[SPARK-3123][GraphX]: override the "setName" function to set EdgeRDD's name manually just as VertexRDD does.

Author: uncleGen <hustyugm@gmail.com>

Closes #2033 from uncleGen/master_origin and squashes the following commits:

801994b [uncleGen] Update EdgeRDD.scala
parent 7c92b49d
No related branches found
No related tags found
No related merge requests found
...@@ -37,7 +37,15 @@ class EdgeRDD[@specialized ED: ClassTag, VD: ClassTag]( ...@@ -37,7 +37,15 @@ class EdgeRDD[@specialized ED: ClassTag, VD: ClassTag](
val targetStorageLevel: StorageLevel = StorageLevel.MEMORY_ONLY) val targetStorageLevel: StorageLevel = StorageLevel.MEMORY_ONLY)
extends RDD[Edge[ED]](partitionsRDD.context, List(new OneToOneDependency(partitionsRDD))) { extends RDD[Edge[ED]](partitionsRDD.context, List(new OneToOneDependency(partitionsRDD))) {
partitionsRDD.setName("EdgeRDD") override def setName(_name: String): this.type = {
if (partitionsRDD.name != null) {
partitionsRDD.setName(partitionsRDD.name + ", " + _name)
} else {
partitionsRDD.setName(_name)
}
this
}
setName("EdgeRDD")
override protected def getPartitions: Array[Partition] = partitionsRDD.partitions override protected def getPartitions: Array[Partition] = partitionsRDD.partitions
......
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