fs2.kafka.internal.WithAdminClient.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, Sync}
import cats.syntax.all.*
import fs2.kafka.admin.MkAdminClient
import fs2.kafka.internal.syntax.*
import fs2.kafka.AdminClientSettings

import org.apache.kafka.clients.admin.AdminClient
import org.apache.kafka.common.KafkaFuture

sealed abstract private[kafka] class WithAdminClient[F[_]] {
  def apply[A](f: AdminClient => KafkaFuture[A]): F[A]
}

private[kafka] object WithAdminClient {

  def apply[F[_], G[_]](
    mk: MkAdminClient[F],
    settings: AdminClientSettings
  )(implicit F: Sync[F], G: Async[G]): Resource[F, WithAdminClient[G]] =
    Resource {
      mk(settings).map { adminClient =>
        val withAdminClient =
          new WithAdminClient[G] {

            override def apply[A](f: AdminClient => KafkaFuture[A]): G[A] =
              G.delay(f(adminClient)).cancelable_

          }

        val close =
          F.blocking(adminClient.close(settings.closeTimeout.toJava))

        (withAdminClient, close)
      }
    }

}