com.teambytes.inflatable.Inflatable.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of inflatable_2.10 Show documentation

inflatable

The newest version!

package com.teambytes.inflatable

import akka.actor.{ActorSystem, Props}
import com.teambytes.inflatable.raft.ClusterConfiguration
import com.teambytes.inflatable.raft.cluster.ClusterRaftActor
import com.teambytes.inflatable.raft.protocol._
import com.typesafe.config.Config
import org.slf4j.LoggerFactory

import scala.concurrent.{ExecutionContext, Future}

object Inflatable {

  def startLeaderElection(handler: InflatableLeader)(implicit ec: ExecutionContext): Unit =
    new Inflatable(handler, AkkaConfig.apply())(ec)

  def startLeaderElection(handler: InflatableLeader, defaults: Config)(implicit ec: ExecutionContext): Unit =
    new Inflatable(handler, AkkaConfig(defaults))(ec)

}

class Inflatable(handler: InflatableLeader, akkaConfig: AkkaConfig)(implicit ec: ExecutionContext) {

  import scala.concurrent.duration._

  private val logger = LoggerFactory.getLogger(classOf[Inflatable])

  logger.info("Inflating raft system...")

  private val clusterSystem = ActorSystem("inflatable-raft", akkaConfig.config)

  private val inflatableActor = clusterSystem.actorOf(
    Props(
      classOf[InflatableActor],
      handler
    ),
    name = s"raft-member-${akkaConfig.seedNumberMap(akkaConfig.host)}")

  logger.info(s"Waiting in Init state with ${akkaConfig.seeds.size} members.")

  private val inflatableCluster = clusterSystem.actorOf(
    Props(
      classOf[ClusterRaftActor],
      inflatableActor,
      akkaConfig.seeds.size
    ),
    name = "inflatable-raft-cluster-actor"
  )

  private val memberFutures = Future.sequence(akkaConfig.seeds.map { actorSeed =>
    clusterSystem.actorSelection(s"$actorSeed/user/inflatable-raft-cluster-actor")
  }.map(_.resolveOne(20.seconds)))

  memberFutures.map { members =>
    logger.info("Inflatable raft system fully inflated!")

    val clusterConfiguration = ClusterConfiguration(akkaConfig.singleNodeCluster, members: _*)

    inflatableCluster ! ChangeConfiguration(clusterConfiguration)
  }

}