zio.kafka.consumer.RebalanceConsumer.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zio-kafka_3 Show documentation

zio-kafka

There is a newer version: 2.9.0

package zio.kafka.consumer

import org.apache.kafka.clients.consumer.{ Consumer => JConsumer, OffsetAndMetadata }
import org.apache.kafka.common.TopicPartition
import zio.{ Task, ZIO }

import scala.jdk.CollectionConverters._

/**
 * A subset of Consumer methods available during rebalances.
 */
trait RebalanceConsumer {
  def commit(offsets: Map[TopicPartition, OffsetAndMetadata]): Task[Unit]
}

object RebalanceConsumer {
  final case class Live(consumer: JConsumer[Array[Byte], Array[Byte]]) extends RebalanceConsumer {
    def commit(offsets: Map[TopicPartition, OffsetAndMetadata]): Task[Unit] =
      ZIO.attemptBlocking(consumer.commitSync(offsets.asJava))
  }
}