fs2.kafka.producer.MkProducer.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.producer

import cats.effect.Sync
import fs2.kafka.{KafkaByteProducer, ProducerSettings}
import fs2.kafka.internal.converters.collection.*

import org.apache.kafka.common.serialization.ByteArraySerializer

/**
  * A capability trait representing the ability to instantiate the Java `Producer` that underlies
  * the fs2-kafka `KafkaProducer`. This is needed in order to instantiate
  * [[fs2.kafka.KafkaProducer]] and [[fs2.kafka.TransactionalKafkaProducer]].


  *
  * By default, the instance provided by [[MkProducer.mkProducerForSync]] will be used. However,
  * this behaviour can be overridden, e.g. for testing purposes, by placing an alternative implicit
  * instance in lexical scope.
  */
trait MkProducer[F[_]] {
  def apply[G[_]](settings: ProducerSettings[G, ?, ?]): F[KafkaByteProducer]
}

object MkProducer {

  implicit def mkProducerForSync[F[_]](implicit F: Sync[F]): MkProducer[F] =
    new MkProducer[F] {

      def apply[G[_]](settings: ProducerSettings[G, ?, ?]): F[KafkaByteProducer] = F.delay {
        val byteArraySerializer = new ByteArraySerializer
        new org.apache.kafka.clients.producer.KafkaProducer(
          (settings.properties: Map[String, AnyRef]).asJava,
          byteArraySerializer,
          byteArraySerializer
        )
      }

    }

}