189 8069 5689

updateStateByKey与mapwithstate怎么实现

这篇文章主要讲解了“updateStateByKey与mapwithstate怎么实现”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“updateStateByKey与mapwithstate怎么实现”吧!

成都创新互联-专业网站定制、快速模板网站建设、高性价比清徐网站开发、企业建站全套包干低至880元,成熟完善的模板库,直接使用。一站式清徐网站制作公司更省心,省钱,快速模板网站建设找我们,业务覆盖清徐地区。费用合理售后完善,十多年实体公司更值得信赖。

updateStateByKey与mapwithstate 这两个方法在Dstream中是找不到的,他们是通过隐式转换来进行实现的

updateStateByKey与mapwithstate怎么实现

由此可以看到,最终是通过PairDStreamFunctions来实现这两个方法的。

updateStateByKey

updateStateByKey与mapwithstate怎么实现

newUpdateFunc 方法是在原有基础上如何进行更新的方法

defaultPartitioner()获得默认的分区数

updateStateByKey与mapwithstate怎么实现

如下代码出现了一个非常关键的地方

new StateDStream(self, ssc.sc.clean(updateFunc), partitioner, rememberPartitioner, None)

updateStateByKey与mapwithstate怎么实现

StateDStream 继承自Dstream。

stateDStream自会持久化到内存中

updateStateByKey与mapwithstate怎么实现

里面有一个很总要的方法:如果存在parent RDD 就将执行computeUsingPreviousRDD方法

updateStateByKey与mapwithstate怎么实现

在该方法中,有一处性能瓶颈的代码

updateStateByKey与mapwithstate怎么实现

每次进行更新的时候都会将原有的parentRDD进行cogroup,这样程序不断的运行这样会导致越来越慢!尽量少用改方法!

updateStateByKey与mapwithstate怎么实现

Mapwithstate

mapWithState方法的返回值是MapWithStateDStream,我们来看看它的实现类

MapWithStateDStreamImpl

updateStateByKey与mapwithstate怎么实现

最终返回InternalMapWithStateDStream

updateStateByKey与mapwithstate怎么实现

跟updateStateByKey一样是持久化在了内存中

persist(StorageLevel.MEMORY_ONLY)

接下来看看每个继承自Dstream的最重要的方法 compute:

updateStateByKey与mapwithstate怎么实现

最终操作的是RDD:MapWithStateRDD

RDD中的partition被MapWithStateRDDRecord代表

updateStateByKey与mapwithstate怎么实现

MapWithStateRDDRecord有伴生对象:中的方法,该方法是对state进行更新操作,不像 updateStateByKey每次都会进cogroup的操作,而是在原有的基础上进行更新,效率得到了提高!

defupdateRecordWithData[K: ClassTag, V: ClassTag, S: ClassTag, E: ClassTag](
    prevRecord: Option[MapWithStateRDDRecord[K, S, E]],
    dataIterator: Iterator[(K, V)],
    mappingFunction: (Time, K, Option[V], State[S]) => Option[E],
    batchTime: Time,
    timeoutThresholdTime: Option[Long],
    removeTimedoutData: Boolean
  ): MapWithStateRDDRecord[K, S, E] = {
    // Create a new state map by cloning the previous one (if it exists) or by creating an empty one
    valnewStateMap = prevRecord.map { _.stateMap.copy() }. getOrElse { newEmptyStateMap[K, S]() }

    valmappedData = newArrayBuffer[E]
    valwrappedState = newStateImpl[S]()

    // Call the mapping function on each record in the data iterator, and accordingly
    // update the states touched, and collect the data returned by the mapping function
    dataIterator.foreach { case(key, value) =>
      wrappedState.wrap(newStateMap.get(key))
      valreturned = mappingFunction(batchTime, key, Some(value), wrappedState)
      if(wrappedState.isRemoved) {
        newStateMap.remove(key)
      } else if(wrappedState.isUpdated
          || (wrappedState.exists && timeoutThresholdTime.isDefined)) {
        newStateMap.put(key, wrappedState.get(), batchTime.milliseconds)
      }
      mappedData ++= returned
    }

    // Get the timed out state records, call the mapping function on each and collect the
    // data returned
    if(removeTimedoutData && timeoutThresholdTime.isDefined) {
      newStateMap.getByTime(timeoutThresholdTime.get).foreach { case(key, state, _) =>
        wrappedState.wrapTimingOutState(state)
        valreturned = mappingFunction(batchTime, key, None, wrappedState)
        mappedData ++= returned
        newStateMap.remove(key)
      }
    }

    MapWithStateRDDRecord(newStateMap, mappedData)
  }
}

感谢各位的阅读,以上就是“updateStateByKey与mapwithstate怎么实现”的内容了,经过本文的学习后,相信大家对updateStateByKey与mapwithstate怎么实现这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是创新互联,小编将为大家推送更多相关知识点的文章,欢迎关注!


网站题目:updateStateByKey与mapwithstate怎么实现
文章出自:http://jkwzsj.com/article/ggheje.html

其他资讯