apache-kafka - 如何从 Spark 结构化流中的特定 Kafka 分区中读取数据

Question

我的 Kafka 主题有三个分区，我想知道我是否可以从三个分区中的一个分区中读取。我的消费者是 spark 结构化流应用程序。

下面是我在 spark 中现有的 kafka 设置。

  val inputDf = spark.readStream
  .format("kafka")
  .option("kafka.bootstrap.servers", brokers)
  .option("subscribe", topic)
  .option("startingOffsets", "latest")
  .load()

Answer 1

这是从特定分区读取的方法。

 val inputDf = spark.readStream
  .format("kafka")
  .option("kafka.bootstrap.servers", brokers)
  .option("assign", """{"topic":[0]}""") 
  .option("startingOffsets", "latest")
  .load()

PS：从多个分区而不是 1--> """{"topic":[0,1,2..n]}""" 读取

Answer 2

同样，您如何写入特定分区。我试过这个，但它不起作用。

        someDF
          .selectExpr("key", "value")
          .writeStream
          .format("kafka")
          .option("kafka.bootstrap.servers", kafkaServers)
          .option("topic", "someTopic")
          .option("partition", partIdx)
          .start()