fs2.kafka.internal.WithConsumer.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fs2-kafka_2.12 Show documentation

fs2-kafka

The newest version!

/*
 * Copyright 2018-2024 OVO Energy Limited
 *
 * SPDX-License-Identifier: Apache-2.0
 */

package fs2.kafka.internal

import cats.effect.{Async, Resource}
import cats.syntax.all.*
import fs2.kafka.{ConsumerSettings, KafkaByteConsumer}
import fs2.kafka.consumer.MkConsumer
import fs2.kafka.internal.syntax.*

sealed abstract private[kafka] class WithConsumer[F[_]] {
  def blocking[A](f: KafkaByteConsumer => A): F[A]
}

private[kafka] object WithConsumer {

  def apply[F[_]: Async, K, V](
    mk: MkConsumer[F],
    settings: ConsumerSettings[F, K, V]
  ): Resource[F, WithConsumer[F]] = {
    val blocking: Resource[F, Blocking[F]] = settings.customBlockingContext match {
      case None     => Blocking.singleThreaded[F]("fs2-kafka-consumer")
      case Some(ec) => Resource.pure(Blocking.fromExecutionContext(ec))
    }

    blocking.flatMap { b =>
      Resource.make {
        mk(settings).map { consumer =>
          new WithConsumer[F] {

            override def blocking[A](f: KafkaByteConsumer => A): F[A] =
              b(f(consumer))

          }
        }
      }(_.blocking(_.close(settings.closeTimeout.toJava)))
    }
  }

}