利用sparkStreaming接受kafak中的消息,采用的是低层次Api-- direct (kafka 10版本)

package com.spark.streaming

import org.apache.kafka.common.serialization.StringDeserializer
import org.apache.spark.{SparkConf, SparkContext}
import org.apache.spark.streaming.{Seconds, StreamingContext}


//todo:利用sparkStreaming接受kafak中的消息,采用的是低层次Api-- direct
object SparkStreamingKafkaDirectCheckpoint {

    System.setProperty("hadoop.home.dir", "d://software/hadoop-2.9.2")
  def updateFunc(a: Seq[Int], b: Option[Int]): Option[Int] = {
    Some(a.sum + b.getOrElse(0))
  }

  def main(args: Array[String]): Unit = {

    val checkpointPath = "./"

    val ssc = StreamingContext.getOrCreate(checkpointPath, () => {
      createFunc(checkpointPath)
    })
    ssc.start()
    ssc.awaitTermination()
  }
  def createFunc(checkpointPath:String): StreamingContext = {
    //todo:1、创建sparkConf
    val sparkConf: SparkConf = new SparkConf()
      .setAppName("SparkStreamingKafka_direct_checkpoint")
      .setMaster("local[4]")

    //todo:2、创建sparkContext
    val sc = new SparkContext(sparkConf)

    sc.setLogLevel("WARN")
    //todo:3、创建StreamingContext
    val ssc = new StreamingContext(sc, Seconds(5))
    ssc.checkpoint(checkpointPath)
    //todo:4、kafka的参数配置
    val kafkaParams = Map(
      "bootstrap.servers" -> "hadoop-senior.test.com:9092",
      "key.deserializer" -> classOf[StringDeserializer],
      "value.deserializer" -> classOf[StringDeserializer],
      "group.id" -> "kafka-direct01",
      "auto.offset.reset" -> "latest",
      "enable.auto.commit" -> "false"
    )


    //todo:5、定义一个topics ,是一个集合,可以存放多个topic
    val topics = Set("kafka_spark")

    //todo:6、利用KafkaUtils.createDirectStream构建Dstream
    val kafkaTopicDS = KafkaUtils.createDirectStream[String, String](
      ssc,
      PreferConsistent,
      Subscribe[String, String](topics, kafkaParams)
    )

    //todo:7、获取kafka中topic的数据
    val kafkaData = kafkaTopicDS.map(x => x.value)

    //todo:8、切分每一行,每个单词记为1
    val wordAndOne = kafkaData.flatMap(_.split(" ")).map((_,1))

    //todo:9、相同单词出现次数累加
    val result = wordAndOne.updateStateByKey(updateFunc)

    kafkaTopicDS.foreachRDD { rdd =>
      val offsetRanges = rdd.asInstanceOf[HasOffsetRanges].offsetRanges

      // some time later, after outputs have completed
      kafkaTopicDS.asInstanceOf[CanCommitOffsets].commitAsync(offsetRanges)
    }
    //todo:打印
    result.print()
    ssc
  }

}

配置版本:

<dependency>
    <groupId>org.apache.spark</groupId>
    <artifactId>spark-streaming-kafka-0-10_2.11</artifactId>
    <version>2.4.5</version>
</dependency>
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值