• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• zio-kafka-master
  • src
    • main
      • scala
        • zio
          • kafka
            • admin
              • AdminClient.scala
              • AdminClientSettings.scala
            • consumer
              • diagnostics
                • DiagnosticEvent.scala
                • Diagnostics.scala
              • package.scala
              • ConsumerSettings.scala
              • internal
                • Runloop.scala
                • ConsumerAccess.scala
                • RequestBuffer.scala
                • RebalanceListener.scala
              • CommittableRecord.scala
              • OffsetBatch.scala
              • Subscription.scala
              • Offset.scala
              • SubscribedConsumer.scala
            • serde
              • Serializer.scala
              • Serdes.scala
              • Serde.scala
              • Deserializer.scala
            • producer
              • package.scala
              • ProducerSettings.scala
    • test
      • resources
        • logback.xml
      • scala
        • zio
          • kafka
            • Benchmarks.scala
            • ConsumerSpec.scala
            • ProducerSpec.scala
            • KafkaTestUtils.scala
            • AdminSpec.scala
            • serde
              • SerializerSpec.scala
              • SerdeSpec.scala
              • DeserializerSpec.scala
            • embedded
              • package.scala
            • KafkaFutureSpec.scala
  • .circleci
    • config.yml
  • LICENSE
  • project
    • build.properties
    • plugins.sbt
  • CLA.md
  • README.md
  • .scalafmt.conf
  • build.sbt
  • .gitignore
  • docs
    • annotatedTests.md

package zio.kafka.consumer

import org.apache.kafka.clients.consumer.RetriableCommitFailedException
import org.apache.kafka.common.TopicPartition
import zio.{ Schedule, Task, ZIO }

sealed trait Offset {
  def topicPartition: TopicPartition
  def offset: Long
  def commit: Task[Unit]
  def batch: OffsetBatch

  /**
   * Attempts to commit and retries according to the given policy when the commit fails
   * with a RetriableCommitFailedException
   */
  def commitOrRetry[R](policy: Schedule[R, Throwable, Any]): ZIO[R, Throwable, Unit] =
    Offset.commitOrRetry(commit, policy)
}

object Offset {
  private[consumer] def commitOrRetry[R, B](
    commit: Task[Unit],
    policy: Schedule[R, Throwable, B]
  ): ZIO[R, Throwable, Unit] =
    commit.retry(
      Schedule.doWhile[Throwable] {
        case _: RetriableCommitFailedException => true
        case _                                 => false
      } && policy
    )
}

private final case class OffsetImpl(
  topicPartition: TopicPartition,
  offset: Long,
  commitHandle: Map[TopicPartition, Long] => Task[Unit]
) extends Offset {
  def commit: Task[Unit] = commitHandle(Map(topicPartition    -> offset))
  def batch: OffsetBatch = OffsetBatchImpl(Map(topicPartition -> offset), commitHandle)
}