溫馨提示×

spark怎么讀取kafka的數據

小億
160
2023-12-28 22:13:39
欄目: 大數據

在Spark中讀取Kafka的數據,可以使用Spark的官方Kafka集成庫,即Spark Streaming Kafka。

首先,你需要在Spark項目中添加Spark Streaming Kafka的依賴。在Maven項目中,可以在pom.xml文件中添加以下依賴:

<dependency>
    <groupId>org.apache.spark</groupId>
    <artifactId>spark-streaming-kafka-0-10_2.12</artifactId>
    <version>3.0.2</version>
</dependency>

然后,你可以使用SparkSession對象創建一個StreamingContext,并指定批處理的時間間隔:

import org.apache.spark.SparkConf;
import org.apache.spark.streaming.Durations;
import org.apache.spark.streaming.api.java.JavaInputDStream;
import org.apache.spark.streaming.api.java.JavaStreamingContext;
import org.apache.spark.streaming.kafka010.KafkaUtils;
import org.apache.kafka.common.serialization.StringDeserializer;

public class KafkaStreamingExample {
    public static void main(String[] args) throws InterruptedException {
        // 創建SparkConf對象
        SparkConf sparkConf = new SparkConf().setAppName("KafkaStreamingExample").setMaster("local[*]");

        // 創建JavaStreamingContext對象
        JavaStreamingContext streamingContext = new JavaStreamingContext(sparkConf, Durations.seconds(1));

        // 設置Kafka參數
        Map<String, Object> kafkaParams = new HashMap<>();
        kafkaParams.put("bootstrap.servers", "localhost:9092");
        kafkaParams.put("key.deserializer", StringDeserializer.class);
        kafkaParams.put("value.deserializer", StringDeserializer.class);
        kafkaParams.put("group.id", "test-group");

        // 創建Kafka主題列表
        Collection<String> topics = Arrays.asList("topic1", "topic2");

        // 創建Kafka輸入流
        JavaInputDStream<ConsumerRecord<String, String>> kafkaStream = KafkaUtils.createDirectStream(
                streamingContext,
                LocationStrategies.PreferConsistent(),
                ConsumerStrategies.<String, String>Subscribe(topics, kafkaParams)
        );

        // 處理Kafka數據
        kafkaStream.foreachRDD(rdd -> {
            // 在這里對每個RDD進行處理
            rdd.foreach(record -> {
                System.out.println("Key: " + record.key() + ", Value: " + record.value());
            });
        });

        // 啟動流處理程序
        streamingContext.start();

        // 等待流處理程序終止
        streamingContext.awaitTermination();
    }
}

在上面的示例中,我們首先創建了一個SparkConf對象和一個JavaStreamingContext對象。然后,我們設置了Kafka的參數,包括Kafka的服務器地址、key和value的反序列化類以及消費者組ID。接下來,我們創建了一個Kafka輸入流,并指定要訂閱的主題和Kafka參數。最后,我們使用foreachRDD方法對每個RDD進行處理,并從中獲取每條記錄的鍵和值。

請注意,上述示例中的createDirectStream方法適用于Kafka 0.10版本及更高版本。如果你使用的是舊版本的Kafka,可以使用createDirectStream方法的另一個重載版本。此外,還可以根據需要調整示例中的其他參數和處理邏輯。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女