• Home
• com.crobox
• clickhouse-scala-client_2.11
• 0.2.1
• source code
• MultiHostBalancer.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.crobox.clickhouse.balancing.MultiHostBalancer.scala Maven / Gradle / Ivy

package com.crobox.clickhouse.balancing

import akka.actor.{ActorRef, ActorSystem}
import akka.http.scaladsl.model.Uri
import akka.pattern.ask
import akka.util.Timeout.durationToTimeout
import com.crobox.clickhouse.balancing.discovery.ConnectionManagerActor
import com.crobox.clickhouse.balancing.discovery.ConnectionManagerActor.GetConnection
import com.crobox.clickhouse.internal.ClickhouseHostBuilder

import scala.concurrent.Future
import scala.concurrent.duration._

case class MultiHostBalancer(hosts: Set[Uri], manager: ActorRef)(implicit system: ActorSystem)
    extends HostBalancer
    with ClickhouseHostBuilder {

  private implicit val timeout = durationToTimeout(5 seconds)

  manager ! ConnectionManagerActor.Connections(hosts)

  override def nextHost: Future[Uri] = (manager ? GetConnection()).mapTo[Uri]
}