Spark算子执行流程详解之四

17.map

/**  * Return a new RDD by applying a function to all elements of this RDD.  */ def map[U: ClassTag](f: T => U): RDD[U] = withScope {   val cleanF = sc.clean(f)   new MapPartitionsRDD[U, T](this, (context, pid, iter) => iter.map(cleanF)) }

利用映射函数f针对其父RDD的每个元素进行处理，继续往下看MapPartitionsRDD

private[spark] class MapPartitionsRDD[U: ClassTag, T: ClassTag](     prev: RDD[T],     f: (TaskContext, Int, Iterator[T]) => Iterator[U],  // (TaskContext, partition index, iterator)     preservesPartitioning: Boolean = false)   extends RDD[U](prev) {   override val partitioner = if (preservesPartitioning) firstParent[T].partitioner else None   override def getPartitions: Array[Partition] = firstParent[T].partitions   override def compute(split: Partition, context: TaskContext): Iterator[U] =     f(context, split.index, firstParent[T].iterator(split, context)) }

注意关注其compute函数，取得是firstParent