Huzhenyu
Huzhenyu

Reputation: 97

Why Spark Streaming with kafka always poll(0) before seektoEnd

In class DirectKafkaInputDStream, I am a little confused about that why should invoke paraniodPoll before seekToEnd?

protected def latestOffsets(): Map[TopicPartition, Long] = {
    val c = consumer
    paranoidPoll(c)
    val parts = c.assignment().asScala

    // make sure new partitions are reflected in currentOffsets
    val newPartitions = parts.diff(currentOffsets.keySet)

    // Check if there's any partition been revoked because of consumer rebalance.
    val revokedPartitions = currentOffsets.keySet.diff(parts)
    if (revokedPartitions.nonEmpty) {
      throw new IllegalStateException(s"Previously tracked partitions " +
        s"${revokedPartitions.mkString("[", ",", "]")} been revoked by Kafka because of consumer " +
        s"rebalance. This is mostly due to another stream with same group id joined, " +
        s"please check if there're different streaming application misconfigure to use same " +
        s"group id. Fundamentally different stream should use different group id")
    }

    // position for new partitions determined by auto.offset.reset if no commit
    currentOffsets = currentOffsets ++ newPartitions.map(tp => tp -> c.position(tp)).toMap

    // find latest available offsets
    c.seekToEnd(currentOffsets.keySet.asJava)
    parts.map(tp => tp -> c.position(tp)).toMap
  }

Upvotes: 0

Views: 358

Answers (1)

QingSun
QingSun

Reputation: 26

Without poll(0), assignment() may return empty set, poll ensures client connects to the Kafka coordinator node.

But poll(0) has been deprecated by kafka-client, check alternate API from Spark.

Also see: KafkaConsumer assignment() returns empty

Upvotes: 1

Related Questions