Spark Streaming源码解读之RDD生成全生命周期彻底研究和思考

RDD的三个问题

1.RDD到底是怎么生成的

2.具体执行的时候,是否和基于Spark Core上的RDD有所不同,runtime级别的

3.运行之后我们对RDD如何处理。会随batch duration不断的产生RDD,内存无法完全容纳这些对象。

每个batch

duration产生的作业执行完RDD之后怎么对以有的RDD进行管理是一个问题。

RDD生成的全生命周期:

ForEachDStream不一定会触发job的执行,会触发job产生,但job真正产生是由timer定时器产生的。

对DStream进行操作其实就是对RDD进行操作,是因为DStream就是一套RDD的模板,后面的DStream与前面的DStream有依赖关系。因为从后往前依赖所以可以推出前面的RDD(回溯)

* DStreams internally is characterized by a few basic properties:

*- A list of other DStreams that the DStream depends on

*  - A time interval at which the DStream generates an RDD

*  - A function that is used to generate an RDD after each time interval

abstract classDStream[T: ClassTag] (

@transientprivate[streaming]varssc: StreamingContext

)extendsSerializablewithLogging {

源码

DStream

/**

* Print the first num elements of each RDD generated in this DStream. This is an output

* operator, so this DStream will be registered as an output stream and there materialized.

*/

defprint(num: Int): Unit = ssc.withScope {

defforeachFunc: (RDD[T], Time) => Unit = {

(rdd: RDD[T], time: Time) => {

valfirstNum =

rdd.take(num +1)

// scalastyle:off println

println("-------------------------------------------")

println("Time: "+ time)

println("-------------------------------------------")

firstNum.take(num).foreach(println)

if(firstNum.length > num)println("...")

println()

// scalastyle:on println}

}

foreachRDD(context.sparkContext.clean(foreachFunc), displayInnerRDDOps =false)

}

private defforeachRDD(

foreachFunc: (RDD[T], Time) => Unit,

displayInnerRDDOps: Boolean): Unit = {

newForEachDStream(this,

context.sparkContext.clean(foreachFunc,false), displayInnerRDDOps).register()

}

/**

* Get the RDD corresponding to the given time; either retrieve it from cache

* or compute-and-cache it.

*/

private[streaming]final

defgetOrCompute(time: Time): Option[RDD[T]] = {

// If RDD was already generated, then retrieve it from HashMap,

// or else compute the RDD

generatedRDDs.get(time).orElse{

// Compute the RDD if time is valid (e.g. correct time in a sliding window)

// of RDD generation, else generate nothing.

if(isTimeValid(time)) {

valrddOption =createRDDWithLocalProperties(time, displayInnerRDDOps =false) {

// Disable checks for existing output directories in jobs launched by the streaming

// scheduler, since we may need to write output to an existing directory during checkpoint

// recovery; see SPARK-4835 for more details. We need to have this call here because

// compute() might cause Spark jobs to be launched.

PairRDDFunctions.disableOutputSpecValidation.withValue(true) {

compute(time)

}

}

rddOption.foreach {casenewRDD =>

// Register the generated RDD for caching and checkpointingif(storageLevel!=

StorageLevel.NONE) {

newRDD.persist(storageLevel)

logDebug(s"Persisting RDD${newRDD.id}for time$timeto$storageLevel")

}

if(checkpointDuration!=null&&

(time -zeroTime).isMultipleOf(checkpointDuration)) {

newRDD.checkpoint()

logInfo(s"Marking RDD${newRDD.id}for time$timefor

checkpointing")

}

generatedRDDs.put(time, newRDD)

}

rddOption

}else{

None

}

}

}

/** Checks whether the 'time' is valid wrt slideDuration for generating RDD */private[streaming]defisTimeValid(time: Time): Boolean = {

if(!isInitialized) {

throw newSparkException (this+" has not been

initialized")

}else if(time <=zeroTime|| !

(time -zeroTime).isMultipleOf(slideDuration)) {

logInfo("Time "+ time +" is

invalid as zeroTime is "+zeroTime+

" and

slideDuration is "+ slideDuration +" and difference is "+ (time -zeroTime))

false}else{

logDebug("Time "+ time +" is

valid")

true}

}

SocketInputDStream继承自ReceiverInputDStream

private[streaming]

classSocketInputDStream[T: ClassTag](

ssc_ : StreamingContext,

host:String,

port: Int,

bytesToObjects: InputStream =>Iterator[T],

storageLevel: StorageLevel

)extendsReceiverInputDStream[T](ssc_) {

ReceiverInputDStream

/**

* Generates RDDs with blocks received by the receiver of this stream. */

override

defcompute(validTime: Time):

Option[RDD[T]] = {

valblockRDD= {

if(validTime <graph.startTime) {

// If this is called for any time before the start time of the context,

// then this returns an empty RDD. This may happen when recovering from a

// driver failure without any write ahead log to recover pre-failure data.

newBlockRDD[T](ssc.sc, Array.empty)

}else{

// Otherwise, ask the tracker for all the blocks that have been allocated to this stream

// for this batch

valreceiverTracker = ssc.scheduler.receiverTrackervalblockInfos = receiverTracker.getBlocksOfBatch(validTime).getOrElse(id,Seq.empty)

// Register the input blocks information into InputInfoTrackervalinputInfo =StreamInputInfo(id, blockInfos.flatMap(_.numRecords).sum)

ssc.scheduler.inputInfoTracker.reportInfo(validTime, inputInfo)

// Create the BlockRDDcreateBlockRDD(validTime, blockInfos)

}

}

Some(blockRDD)

}

private[streaming]defcreateBlockRDD(time: Time,

blockInfos:Seq[ReceivedBlockInfo]): RDD[T] = {

if(blockInfos.nonEmpty) {

valblockIds = blockInfos.map { _.blockId.asInstanceOf[BlockId] }.toArray

// Are WAL record handles present with all the blocksvalareWALRecordHandlesPresent = blockInfos.forall { _.walRecordHandleOption.nonEmpty }

if(areWALRecordHandlesPresent) {

// If all the blocks have WAL record handle, then create a WALBackedBlockRDDvalisBlockIdValid = blockInfos.map { _.isBlockIdValid() }.toArray

valwalRecordHandles = blockInfos.map { _.walRecordHandleOption.get }.toArray

newWriteAheadLogBackedBlockRDD[T](

ssc.sparkContext, blockIds, walRecordHandles, isBlockIdValid)

}else{

// Else, create a BlockRDD. However, if there are some blocks with WAL info but not

// others then that is unexpected and log a warning accordingly.

if(blockInfos.find(_.walRecordHandleOption.nonEmpty).nonEmpty) {

if(WriteAheadLogUtils.enableReceiverLog(ssc.conf)) {

logError("Some blocks

do not have Write Ahead Log information; "+

"this is unexpected and data may not be recoverable after

driver failures")

}else{

logWarning("Some blocks have Write Ahead Log information; this is

unexpected")

}

}

valvalidBlockIds = blockIds.filter { id =>

ssc.sparkContext.env.blockManager.master.contains(id)

}

if(validBlockIds.size != blockIds.size) {

logWarning("Some blocks could not be

recovered as they were not found in memory. "+

"To prevent such data loss, enabled Write Ahead Log (see

programming guide "+

"for more

details.")

}

newBlockRDD[T](ssc.sc, validBlockIds)

}

}else{

// If no block is ready now, creating WriteAheadLogBackedBlockRDD or BlockRDD

// according to the configuration

if(WriteAheadLogUtils.enableReceiverLog(ssc.conf)) {

newWriteAheadLogBackedBlockRDD[T](

ssc.sparkContext, Array.empty, Array.empty, Array.empty)

}else{

newBlockRDD[T](ssc.sc, Array.empty)

}

}

}

MappedDStream

private[streaming]

classMappedDStream[T: ClassTag,U: ClassTag] (

parent: DStream[T],

mapFunc:T=>U

)extendsDStream[U](parent.ssc) {

override defdependencies:List[DStream[_]] =List(parent)

override defslideDuration: Duration = parent.slideDuration

override defcompute(validTime: Time): Option[RDD[U]] = {

parent.getOrCompute(validTime).map(_.map[U](mapFunc))

}

}

ForEachDStream

private[streaming]

classForEachDStream[T: ClassTag] (

parent: DStream[T],

foreachFunc: (RDD[T], Time) => Unit,

displayInnerRDDOps: Boolean

)extendsDStream[Unit](parent.ssc) {

override defdependencies:List[DStream[_]] =List(parent)

override defslideDuration: Duration = parent.slideDuration

override defcompute(validTime: Time): Option[RDD[Unit]] = None

override defgenerateJob(time: Time): Option[Job] = {

parent.getOrCompute(time)match{

caseSome(rdd) =>

valjobFunc = () =>createRDDWithLocalProperties(time, displayInnerRDDOps) {

foreachFunc(rdd, time)

}

Some(newJob(time, jobFunc))

caseNone => None

}

}

}

备注:

资料来源于:DT_大数据梦工厂(Spark发行版本定制)

更多私密内容,请关注微信公众号:DT_Spark

如果您对大数据Spark感兴趣,可以免费听由王家林老师每天晚上20:00开设的Spark永久免费公开课,地址YY房间号:68917580

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,911评论 5 460
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,014评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 142,129评论 0 320
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,283评论 1 264
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,159评论 4 357
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,161评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,565评论 3 382
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,251评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,531评论 1 292
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,619评论 2 310
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,383评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,255评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,624评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,916评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,199评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,553评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,756评论 2 335

推荐阅读更多精彩内容