SparkStreaming獲取kafka數據的兩種方式：Receiver與Direct的方式

目錄

• Receiver

• Direct

• Direct代碼

 簡單理解爲：Receiver方式是經過zookeeper來鏈接kafka隊列，Direct方式是直接鏈接到kafka的節點上獲取數據

Receiver

使用Kafka的高層次Consumer API來實現。receiver從Kafka中獲取的數據都存儲在Spark Executor的內存中，而後Spark Streaming啓動的job會去處理那些數據。然而，在默認的配置下，這種方式可能會由於底層的失敗而丟失數據。若是要啓用高可靠機制，讓數據零丟失，就必須啓用Spark Streaming的預寫日誌機制（Write Ahead Log，WAL）。該機制會同步地將接收到的Kafka數據寫入分佈式文件系統（好比HDFS）上的預寫日誌中。因此，即便底層節點出現了失敗，也可使用預寫日誌中的數據進行恢復。

注意事項：

一、Kafka中topic的partition與Spark中RDD的partition是沒有關係的，所以，在KafkaUtils.createStream()中，提升partition的數量，只會增長Receiver的數量，也就是讀取Kafka中topic partition的線程數量，不會增長Spark處理數據的並行度。

二、能夠建立多個Kafka輸入DStream，使用不一樣的consumer group和topic，來經過多個receiver並行接收數據。

三、若是基於容錯的文件系統，好比HDFS，啓用了預寫日誌機制，接收到的數據都會被複制一份到預寫日誌中。所以，在KafkaUtils.createStream()中，設置的持久化級別是StorageLevel.MEMORY_AND_DISK_SER。

回到頂部

Direct

Spark1.3中引入Direct方式，用來替代掉使用Receiver接收數據，這種方式會週期性地查詢Kafka，得到每一個topic+partition的最新的offset，從而定義每一個batch的offset的範圍。當處理數據的job啓動時，就會使用Kafka的簡單consumer api來獲取Kafka指定offset範圍的數據。

這種方式有以下優勢：

一、簡化並行讀取：若是要讀取多個partition，不須要建立多個輸入DStream，而後對它們進行union操做。Spark會建立跟Kafka partition同樣多的RDD partition，而且會並行從Kafka中讀取數據。因此在Kafka partition和RDD partition之間，有一個一對一的映射關係。

二、高性能：若是要保證零數據丟失，在基於receiver的方式中，須要開啓WAL機制。這種方式其實效率低下，由於數據實際上被複制了兩份，Kafka本身自己就有高可靠的機制會對數據複製一份，而這裏又會複製一份到WAL中。而基於direct的方式，不依賴Receiver，不須要開啓WAL機制，只要Kafka中做了數據的複製，那麼就能夠經過Kafka的副本進行恢復。

三、一次且僅一次的事務機制：基於receiver的方式，是使用Kafka的高階API來在ZooKeeper中保存消費過的offset的。這是消費Kafka數據的傳統方式。這種方式配合着WAL機制能夠保證數據零丟失的高可靠性，可是卻沒法保證數據被處理一次且僅一次，可能會處理兩次。由於Spark和ZooKeeper之間多是不一樣步的。基於direct的方式，使用kafka的簡單api，Spark Streaming本身就負責追蹤消費的offset，並保存在checkpoint中。Spark本身必定是同步的，所以能夠保證數據是消費一次且僅消費一次。因爲數據消費偏移量是保存在checkpoint中，所以，若是後續想使用kafka高級API消費數據，須要手動的更新zookeeper中的偏移量

回到頂部

Direct代碼

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49

package bigdata.spark   import kafka.serializer.{StringDecoder, Decoder} import org.apache.spark.streaming.kafka.KafkaUtils import org.apache.spark.streaming.{Seconds, StreamingContext} import org.apache.spark.{SparkContext, SparkConf}   import scala.reflect.ClassTag   /**    * Created by Administrator on 2017/4/28.    */ object SparkStreamDemo {    def main(args: Array[String]) {        val conf =  new SparkConf()      conf.setAppName( "spark_streaming" )      conf.setMaster( "local[*]" )        val sc =  new SparkContext(conf)      sc.setCheckpointDir( "D:/checkpoints" )      sc.setLogLevel( "ERROR" )        val ssc =  new StreamingContext(sc, Seconds( 5 ))        // val topics = Map("spark" -> 2)        val kafkaParams = Map[String, String](        "bootstrap.servers" ->  "m1:9092,m2:9092,m3:9092" ,        "group.id" ->  "spark" ,        "auto.offset.reset" ->  "smallest"      )      // 直連方式拉取數據，這種方式不會修改數據的偏移量，須要手動的更新      val lines =  KafkaUtils.createDirectStream[String, String, StringDecoder, StringDecoder](ssc, kafkaParams, Set( "spark" )).map(_._2)      // val lines = KafkaUtils.createStream(ssc, "m1:2181,m2:2181,m3:2181", "spark", topics).map(_._2)        val ds1 = lines.flatMap(_.split( " " )).map((_,  1 ))        val ds2 = ds1.updateStateByKey[Int]((x:Seq[Int], y:Option[Int]) => {        Some(x.sum + y.getOrElse( 0 ))      })        ds2.print()        ssc.start()      ssc.awaitTermination()      } }

　轉自：http://www.javashuo.com/article/p-smzapiqr-cy.html