package com.spark.streaming
import kafka.serializer.StringDecoder
import org.apache.spark.streaming.dstream.{DStream, InputDStream}
import org.apache.spark.streaming.kafka.KafkaUtils
import org.apache.spark.streaming.{Seconds, StreamingContext}
import org.apache.spark.{SparkConf, SparkContext}
//todo:利用sparkStreaming接受kafak中的消息,采用的是低层次Api-- direct
object SparkStreamingKafka10DirectCheckpoint {
System.setProperty("hadoop.home.dir", "d://soft/hadoop/hadoop-2.7.3")
def updateFunc(a: Seq[Int], b: Option[Int]): Option[Int] = {
Some(a.sum + b.getOrElse(0))
}
def main(args: Array[String]): Unit = {
val checkpointPath = "./kafka-direct"
val ssc = StreamingContext.getOrCreate(checkpointPath, () => {
createFunc(checkpointPath)
})
ssc.start()
ssc.awaitTermination()
}
def createFunc(checkpointPath:String): StreamingContext = {
//todo:1、创建sparkConf
val 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("metadata.broker.list" ->"star.com:9092"
,"group.id" -> "kafka-direct01")
//todo:5、定义一个topics ,是一个集合,可以存放多个topic
val topics = Set("kafka_spark")
//todo:6、利用KafkaUtils.createDirectStream构建Dstream
val kafkaTopicDS = KafkaUtils.createDirectStream[String,String,StringDecoder,StringDecoder](ssc,kafkaParams,topics)
//todo:7、获取kafka中topic的数据
val kafkaData = kafkaTopicDS.map(x=>x._2)
//todo:8、切分每一行,每个单词记为1
val wordAndOne = kafkaData.flatMap(_.split(" ")).map((_,1))
//todo:9、相同单词出现次数累加
val result = wordAndOne.updateStateByKey(updateFunc)
//todo:打印
result.print()
ssc
}
}
配置版本:
<dependency>
<groupId>org.apache.spark</groupId>
<artifactId>spark-streaming-kafka-0-8_2.11</artifactId>
<version>2.4.5</version>
</dependency>