org.apache.pekko.kafka.ConnectionCheckerSettings.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pekko-connectors-kafka_2.13 Show documentation

Apache Pekko Kafka Connector is a Reactive Enterprise Integration library for Java and Scala, based on Reactive Streams and Apache Pekko.

The newest version!

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * license agreements; and to You under the Apache License, version 2.0:
 *
 *   https://www.apache.org/licenses/LICENSE-2.0
 *
 * This file is part of the Apache Pekko project, which was derived from Akka.
 */

/*
 * Copyright (C) 2014 - 2016 Softwaremill 
 * Copyright (C) 2016 - 2020 Lightbend Inc. 
 */

package org.apache.pekko.kafka

import org.apache.pekko.util.JavaDurationConverters._
import com.typesafe.config.Config

import scala.concurrent.duration._

import java.time.{ Duration => JDuration }

class ConnectionCheckerSettings private[kafka] (val enable: Boolean,
    val maxRetries: Int,
    val checkInterval: FiniteDuration,
    val factor: Double) {

  require(factor > 0, "Backoff factor for connection checker must be finite positive number")
  require(maxRetries >= 0, "retries for connection checker must be not negative number")

  private def copy(enable: Boolean = enable,
      maxRetries: Int = maxRetries,
      checkInterval: FiniteDuration = checkInterval,
      factor: Double = factor): ConnectionCheckerSettings =
    new ConnectionCheckerSettings(
      enable,
      maxRetries,
      checkInterval,
      factor)

  def withEnable(enable: Boolean): ConnectionCheckerSettings = copy(enable = enable)
  def withMaxRetries(maxRetries: Int): ConnectionCheckerSettings = copy(maxRetries = maxRetries)
  def withFactor(factor: Double): ConnectionCheckerSettings = copy(factor = factor)

  /** Scala API */
  def withCheckInterval(checkInterval: FiniteDuration): ConnectionCheckerSettings = copy(checkInterval = checkInterval)

  /** Java API */
  def withCheckInterval(checkInterval: JDuration): ConnectionCheckerSettings =
    copy(checkInterval = checkInterval.asScala)

  override def toString: String =
    s"org.apache.pekko.kafka.ConnectionCheckerSettings(" +
    s"enable=$enable," +
    s"maxRetries=$maxRetries," +
    s"checkInterval=${checkInterval.toCoarsest}," +
    s"factor=$factor" +
    ")"
}

object ConnectionCheckerSettings {

  val configPath: String = "connection-checker"
  val fullConfigPath: String = ConsumerSettings.configPath + "." + configPath

  def apply(maxRetries: Int, checkInterval: FiniteDuration, factor: Double): ConnectionCheckerSettings =
    new ConnectionCheckerSettings(true, maxRetries, checkInterval, factor)

  def create(maxRetries: Int, checkInterval: FiniteDuration, factor: Double): ConnectionCheckerSettings =
    apply(maxRetries, checkInterval, factor)

  /**
   * Create settings from a configuration with layout `connection-checker`.
   */
  def apply(config: Config): ConnectionCheckerSettings = {
    val enable = config.getBoolean("enable")
    if (enable) {
      val retries = config.getInt("max-retries")
      val factor = config.getDouble("backoff-factor")
      val checkInterval = config.getDuration("check-interval").asScala
      apply(retries, checkInterval, factor)
    } else Disabled
  }

  /**
   * Java API: Create settings from a configuration with layout `connection-checker`.
   */
  def create(config: Config): ConnectionCheckerSettings = apply(config)

  val Disabled: ConnectionCheckerSettings = new ConnectionCheckerSettings(false, 3, 15.seconds, 2d)

}