Tom*_*tom 5 scala apache-kafka apache-spark spark-streaming
我正在尝试为Spark流媒体准备应用程序(Spark 2.1,Kafka 0.10)
我需要从Kafka主题“输入”中读取数据,找到正确的数据并将结果写入主题“输出”中
我可以在KafkaUtils.createDirectStream方法的基础上从Kafka读取数据。
我将RDD转换为json并准备了过滤器:
val messages = KafkaUtils.createDirectStream[String, String](
ssc,
PreferConsistent,
Subscribe[String, String](topics, kafkaParams)
)
val elementDstream = messages.map(v => v.value).foreachRDD { rdd =>
val PeopleDf=spark.read.schema(schema1).json(rdd)
import spark.implicits._
PeopleDf.show()
val PeopleDfFilter = PeopleDf.filter(($"value1".rlike("1"))||($"value2" === 2))
PeopleDfFilter.show()
}
Run Code Online (Sandbox Code Playgroud)
我可以从Kafka加载数据并使用KafkaProducer将“原样”写入Kafka:
messages.foreachRDD( rdd => {
rdd.foreachPartition( partition => {
val kafkaTopic = "output"
val props = new HashMap[String, Object]()
props.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, "localhost:9092")
props.put(ProducerConfig.VALUE_SERIALIZER_CLASS_CONFIG,
"org.apache.kafka.common.serialization.StringSerializer")
props.put(ProducerConfig.KEY_SERIALIZER_CLASS_CONFIG,
"org.apache.kafka.common.serialization.StringSerializer")
val producer = new KafkaProducer[String, String](props)
partition.foreach{ record: ConsumerRecord[String, String] => {
System.out.print("########################" + record.value())
val messageResult = new ProducerRecord[String, String](kafkaTopic, record.value())
producer.send(messageResult)
}}
producer.close()
})
})
Run Code Online (Sandbox Code Playgroud)
但是我不能整合这两个动作>在json中找到合适的值并将发现结果写入Kafka:将JSON格式的PeopleDfFilter写入“输出” Kafka主题。
我在Kafka中有很多输入消息,这就是为什么我要使用foreachPartition创建Kafka生产者的原因。
非常感谢您的任何建议。
小智 5
该过程非常简单,那么为什么不完全使用结构化流?
import org.apache.spark.sql.functions.from_json
spark
// Read the data
.readStream
.format("kafka")
.option("kafka.bootstrap.servers", inservers)
.option("subscribe", intopic)
.load()
// Transform / filter
.select(from_json($"value".cast("string"), schema).alias("value"))
.filter(...) // Add the condition
.select(to_json($"value").alias("value")
// Write back
.writeStream
.format("kafka")
.option("kafka.bootstrap.servers", outservers)
.option("subscribe", outtopic)
.start()
Run Code Online (Sandbox Code Playgroud)