• Home
• com.github.thanhtien522
• scala-kafka-client_2.11
• 0.1.0
• source code
• ProducerBuilder.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.github.thanhtien522.produce.ProducerBuilder.scala Maven / Gradle / Ivy

package com.github.thanhtien522.produce

import com.typesafe.config.{Config, ConfigFactory}
import org.apache.kafka.common.serialization.Serializer

import scala.collection.JavaConverters._
import scala.collection.mutable

class ProducerBuilder[K, V]() {

  var keySerializer: Serializer[K] = _
  var valueSerializer: Serializer[V] = _
  val config: mutable.Map[String, Any] = mutable.Map(
    "acks" -> "all",
    "retries" -> 0,
    "batch.size" -> 16384,
    "linger.ms" -> 1,
    "buffer.memory" -> 33554432
  )

  def withConfig(config: Config): this.type = {
    config.entrySet().asScala.foreach(entry =>
      this.config.put(entry.getKey, entry.getValue.unwrapped().toString))
    this
  }

  def setBootstrapServers(host: String): this.type = setConfig("bootstrap.servers", host)

  def setSerializer(keySerializer: Serializer[K], valueSerializer: Serializer[V]): this.type = {
    this.keySerializer = keySerializer
    this.valueSerializer = valueSerializer
    this
  }

  def setConfig(key: String, value: String): this.type = {
    this.config(key) = value
    this
  }

  def build(): KafkaProducer[K, V] = {
    new KafkaProducer[K, V](
      ConfigFactory.parseMap(this.config.asJava),
      keySerializer, valueSerializer)
  }
}