kafka-0-8结合sparkstreaming直连方式

添加依赖,这里的 ${spark.version}是你当前spark版本,2.11是scala版本

        <!-- sparkSteaming跟Kafka整合的依赖 -->
        <dependency>
            <groupId>org.apache.spark</groupId>
            <artifactId>spark-streaming-kafka-0-8_2.11</artifactId>
            <version>${spark.version}</version>
        </dependency>

参考代码:

package sparkStreaming

import kafka.common.TopicAndPartition
import kafka.message.MessageAndMetadata
import kafka.serializer.StringDecoder
import kafka.utils.{ZKGroupTopicDirs, ZkUtils}
import org.I0Itec.zkclient.ZkClient
import org.apache.spark.SparkConf
import org.apache.spark.streaming.dstream.{DStream, InputDStream}
import org.apache.spark.streaming.kafka.{HasOffsetRanges, KafkaUtils, OffsetRange}
import org.apache.spark.streaming.{Duration, StreamingContext}

object KafkaDirectWordCount {
  def main(args: Array[String]): Unit = {
    val group = "g001"
    val conf = new SparkConf().setAppName("KafkaDirectWordCount").setMaster("local[4]")
    val ssc = new StreamingContext(conf,Duration(5000))
    val topic = "first"
    //指定kafka的broker地址(sparkStream的Task直连到kafka的分区上,用更加底层的API消费,效率更高)
    val brokerList ="hdp-1:9092,hdp-2:9092,hdp-3:9092"
    //指定zk的地址,后期更新消费的偏移量时使用(以后可以使用Redis、MySQL来记录偏移量)
    val zkQuorum = "hdp-1:2181,hdp-2:2181,hdp-3:2181"
    //创建 stream 时使用的 topic 名字集合,SparkStreaming可同时消费多个topic
    val topics:Set[String] = Set(topic)
    //创建一个 ZKGroupTopicDirs 对象,其实是指定往zk中写入数据的目录,用于保存偏移量
    val topicDirs = new ZKGroupTopicDirs(group,topic)
    //获取zookeeper中的路径
    val zkTopicPath = s"${topicDirs.consumerOffsetDir}"

    //准备kafka的参数
    val kafkaParams = Map(
      "metadata.broker.list" -> brokerList,
      "group.id" -> group,
      //从头开始读取数据
      "auth.offset.reset" -> kafka.api.OffsetRequest.SmallestTimeString
    )
    //zookeeper的host和ip,创建一个client,用于跟新偏移量
    //zookepper的客户端,可以从zk中读取偏移量数据,并跟新偏移量
    val zkClient = new ZkClient(zkQuorum)
    val children = zkClient.countChildren(zkTopicPath)
    var kafkaStream:InputDStream[(String,String )] = null

    //如果zookeeper中有保存offset,我们会利用这个offset作为kafkaStream的
    //起始位置
    var fromOffsets: Map[TopicAndPartition, Long] = Map()
    //如果保存过offset
    if (children >0){
      for (i <- 0 until children){
        val partitionOffset = zkClient.readData[String](s"$zkTopicPath/${i}")
        val tp = TopicAndPartition(topic,i)
        //将不同partition对应的offset增加到fromOffsets中
        fromOffsets += (tp ->partitionOffset.toLong)
      }
      //Key: kafka的key   values: "hello tom hello jerry"
      //这个会将 kafka 的消息进行 transform,最终 kafak 的数据都会变成 (kafka的key, message) 这样的 tuple
      val messageHandler = (mmd: MessageAndMetadata[String,String]) =>(mmd.key(),mmd.message())
      //通过KafkaUtils创建直连的DStream(fromOffsets参数的作用是:按照前面计算好了的偏移量继续消费数据)
      //[String, String, StringDecoder, StringDecoder,     (String, String)]
      //  key    value    key的解码方式   value的解码方式
      kafkaStream = KafkaUtils.createDirectStream[String, String, StringDecoder, StringDecoder, (String, String)](ssc, kafkaParams, fromOffsets, messageHandler)
    } else {
      //如果未保存,根据 kafkaParam 的配置使用最新(largest)或者最旧的(smallest) offset
      kafkaStream = KafkaUtils.createDirectStream[String, String, StringDecoder, StringDecoder](ssc, kafkaParams, topics)
    }
    //偏移量的范围
    var offsetRanges = Array[OffsetRange]()
    //从kafka读取的消息,DStream的Transform方法可以将当前批次的RDD获取出来
    //该transform方法计算获取到当前批次RDD,然后将RDD的偏移量取出来,然后在将RDD返回到DStream
    val transform:DStream[(String,String)] = kafkaStream.transform{ rdd =>
      //得到该 rdd 对应 kafka 的消息的 offset
      //该RDD是一个KafkaRDD,可以获得偏移量的范围
      offsetRanges = rdd.asInstanceOf[HasOffsetRanges].offsetRanges
      rdd
    }

    val messages: DStream[String] = transform.map(_._2)
    //依次迭代DStream中的RDD
    messages.foreachRDD{rdd =>
      //对RDD进行操作,触发Action
      rdd.foreachPartition(partition =>
        partition.foreach(x =>{
          println(x)
        }))
      for (o <- offsetRanges){
        val zkPath = s"${topicDirs.consumerOffsetDir}/${o.partition}"
        //将该 partition 的 offset 保存到 zookeeper
        ZkUtils.updatePersistentPath(zkClient,zkPath,o.untilOffset.toString)
      }
    }
    ssc.start()
    ssc.awaitTermination()
  }

}

如果将程序断开,继续往kafka生产者中输入数据,等下次再开启程序会自动从上次断开处打印结果

发布了85 篇原创文章 · 获赞 3 · 访问量 2522

猜你喜欢

转载自blog.csdn.net/Romantic_sir/article/details/103383666
今日推荐