1. 程式人生 > >Spark2.10中使用累加器、注意點以及實現自定義累加器

Spark2.10中使用累加器、注意點以及實現自定義累加器

累加器(accumulator)是Spark中提供的一種分散式的變數機制,其原理類似於mapreduce,即分散式的改變,然後聚合這些改變。累加器的一個常見用途是在除錯時對作業執行過程中的事件進行計數。

累加器簡單使用

Spark內建的提供了Long和Double型別的累加器。下面是一個簡單的使用示例,在這個例子中我們在過濾掉RDD中奇數的同時進行計數,最後計算剩下整數的和。

  1. val sparkConf = new SparkConf().setAppName( "Test").setMaster( "local[2]"
    )
  2. val sc = new SparkContext(sparkConf)
  3. val accum = sc.longAccumulator( "longAccum") //統計奇數的個數
  4. val sum = sc.parallelize(Array( 1, 2, 3, 4, 5, 6
    , 7, 8, 9), 2).filter(n=>{
  5. if(n% 2!= 0) accum.add( 1L)
  6. n% 2== 0
  7. }).reduce(_+_)
  8. println( "sum: "
    +sum)
  9. println( "accum: "+accum.value)
  10. sc.stop()

結果為:

sum: 20
accum: 5

這是結果正常的情況,但是在使用累加器的過程中如果對於spark的執行過程理解的不夠深入就會遇到兩類典型的錯誤:少加(或者沒加)、多加。

少加的情況:

對於如下程式碼:

  1. val accum = sc.longAccumulator( "longAccum")
  2. val numberRDD = sc.parallelize(Array( 1, 2, 3, 4, 5, 6, 7, 8, 9), 2).map(n=>{
  3. accum.add( 1L)
  4. n+ 1
  5. })
  6. println( "accum: "+accum.value)


執行完畢,列印的值是多少呢?答案是0,因為累加器不會改變spark的lazy的計算模型,即在列印的時候像map這樣的transformation還沒有真正的執行,從而累加器的值也就不會更新。

多加的情況:

對於如下程式碼:

  1. val accum = sc.longAccumulator( "longAccum")
  2. val numberRDD = sc.parallelize(Array( 1, 2, 3, 4, 5, 6, 7, 8, 9), 2).map(n=>{
  3. accum.add( 1L)
  4. n+ 1
  5. })
  6. numberRDD. count
  7. println ("accum1:"+accum.value)
  8. numberRDD. reduce (_+_)
  9. println ("accum2: "+accum.value)

結果我們得到了:

accum1:9

accum2: 18

我們雖然只在map裡進行了累加器加1的操作,但是兩次得到的累加器的值卻不一樣,這是由於count和reduce都是action型別的操作,觸發了兩次作業的提交,所以map運算元實際上被執行了了兩次,在reduce操作提交作業後累加器又完成了一輪計數,所以最終累加器的值為18。究其原因是因為count雖然促使numberRDD被計出來,但是由於沒有對其進行快取,所以下次再次需要使用numberRDD這個資料集是,還需要從並行化資料集的部分開始執行計算。解釋到這裡,這個問題的解決方法也就很清楚了,就是在count之前呼叫numberRDD的cache方法(或persist),這樣在count後資料集就會被快取下來,reduce操作就會讀取快取的資料集而無需從頭開始計算了。改成如下程式碼即可:

  1. val accum = sc.longAccumulator( "longAccum")
  2. val numberRDD = sc.parallelize(Array( 1, 2, 3, 4, 5, 6, 7, 8, 9), 2).map(n=>{
  3. accum.add( 1L)
  4. n+ 1
  5. })
  6. numberRDD.cache(). count
  7. println ("accum1:"+accum.value)
  8. numberRDD. reduce (_+_)
  9. println ("accum2: "+accum.value)

這次兩次列印的值就會保持一致了。

自定義累加器

自定義累加器型別的功能在1.X版本中就已經提供了,但是使用起來比較麻煩,在2.0版本後,累加器的易用性有了較大的改進,而且官方還提供了一個新的抽象類:AccumulatorV2來提供更加友好的自定義型別累加器的實現方式。官方同時給出了一個實現的示例:CollectionAccumulator類,這個類允許以集合的形式收集spark應用執行過程中的一些資訊。例如,我們可以用這個類收集Spark處理資料時的一些細節,當然,由於累加器的值最終要匯聚到driver端,為了避免 driver端的outofmemory問題,需要對收集的資訊的規模要加以控制,不宜過大。 實現自定義型別累加器需要繼承AccumulatorV2並至少覆寫下例中出現的方法,下面這個累加器可以用於在程式執行過程中收集一些文字類資訊,最終以Set[String]的形式返回。
  1. import java.util
  2. import org.apache.spark.util.AccumulatorV2
  3. class LogAccumulator extends AccumulatorV2[String, java.util.Set[String]] {
  4. private val _logArray: java.util.Set[String] = new java.util.HashSet[String]()
  5. override def isZero: Boolean = {
  6. _logArray.isEmpty
  7. }
  8. override def reset(): Unit = {
  9. _logArray.clear()
  10. }
  11. override def add(v: String): Unit = {
  12. _logArray.add(v)
  13. }
  14. override def merge(other: AccumulatorV2[String, java.util.Set[String]]): Unit = {
  15. other match {
  16. case o: LogAccumulator => _logArray.addAll(o.value)
  17. }
  18. }
  19. override def value: java.util.Set[String] = {
  20. java.util.Collections.unmodifiableSet(_logArray)
  21. }
  22. override def copy(): AccumulatorV2[String, util.Set[String]] = {
  23. val newAcc = new LogAccumulator()
  24. _logArray. synchronized{
  25. newAcc._logArray.addAll(_logArray)
  26. }
  27. newAcc
  28. }
  29. }

測試類:
  1. import scala.collection.JavaConversions._
  2. import org.apache.spark.{SparkConf, SparkContext}
  3. object Main {
  4. def main(args: Array[String]): Unit = {
  5. val sparkConf = new SparkConf().setAppName( "Test").setMaster( "local[2]")
  6. val sc = new SparkContext(sparkConf)
  7. val accum = new LogAccumulator
  8. sc.register(accum, "logAccum")
  9. val sum = sc.parallelize(Array( "1", "2a", "3", "4b", "5", "6", "7cd", "8", "9"), 2).filter(line => {
  10. val pattern = "" "^-?(\d+)" ""
  11. val flag = line.matches(pattern)
  12. if (!flag) {
  13. accum.add(line)
  14. }
  15. flag
  16. }).map(_.toInt).reduce(_ + _)
  17. println( "sum: " + sum)
  18. for (v <- accum.value) print(v + " ")
  19. println()
  20. sc.stop()
  21. }
  22. }

本例中利用自定義的收集器收集過濾操作中被過濾掉的元素,當然這部分的元素的資料量不能太大。執行結果如下: sum; 32
7cd 4b 2a 

來源:https://blog.csdn.net/u013468917/article/details/70617085