2011-10-26 08:48:16 +02:00
/* *
2012-01-19 18:21:06 +01:00
* Copyright ( C ) 2009 - 2012 Typesafe Inc . < http : //www.typesafe.com>
2011-10-26 08:48:16 +02:00
*/
2012-01-31 13:33:04 +01:00
package akka.cluster
2011-10-26 08:48:16 +02:00
import akka.actor._
import akka.actor.Status._
2012-01-31 13:33:04 +01:00
import akka.remote._
2012-02-20 15:26:12 +01:00
import akka.routing._
2011-10-27 12:46:10 +02:00
import akka.event.Logging
2012-01-24 12:09:32 +01:00
import akka.dispatch.Await
2012-01-31 13:33:04 +01:00
import akka.pattern.ask
import akka.util._
2012-03-02 16:20:30 +01:00
import akka.util.duration._
2011-11-25 12:02:25 +01:00
import akka.config.ConfigurationException
2012-01-30 11:41:41 +01:00
import java.util.concurrent.atomic. { AtomicReference , AtomicBoolean }
2012-01-24 12:09:32 +01:00
import java.util.concurrent.TimeUnit._
import java.util.concurrent.TimeoutException
2011-10-27 15:14:15 +02:00
import java.security.SecureRandom
2011-11-25 12:02:25 +01:00
2012-01-30 11:41:41 +01:00
import scala.collection.immutable. { Map , SortedSet }
2011-10-26 08:48:16 +02:00
import scala.annotation.tailrec
2011-11-25 12:02:25 +01:00
2012-01-24 12:09:32 +01:00
import com.google.protobuf.ByteString
2011-10-26 08:48:16 +02:00
/* *
2012-02-09 15:59:10 +01:00
* Interface for membership change listener .
2011-10-26 08:48:16 +02:00
*/
2012-02-18 22:14:53 +01:00
trait MembershipChangeListener {
2012-02-09 15:59:10 +01:00
def notify ( members : SortedSet [ Member ] ) : Unit
}
/* *
* Interface for meta data change listener .
*/
trait MetaDataChangeListener { // FIXME add management and notification for MetaDataChangeListener
def notify ( meta : Map [ String , Array [ Byte ] ] ) : Unit
2011-10-26 08:48:16 +02:00
}
2012-02-07 16:53:49 +01:00
// FIXME create Protobuf messages out of all the Gossip stuff - but wait until the prototol is fully stablized.
2011-10-26 08:48:16 +02:00
/* *
2012-01-30 11:41:41 +01:00
* Base trait for all cluster messages . All ClusterMessage 's are serializable .
2011-10-26 08:48:16 +02:00
*/
2012-01-30 19:40:28 +01:00
sealed trait ClusterMessage extends Serializable
2012-01-30 11:41:41 +01:00
/* *
2012-02-28 17:04:48 +01:00
* Cluster commands sent by the USER .
2012-01-30 11:41:41 +01:00
*/
2012-02-29 10:02:00 +01:00
object ClusterAction {
2011-10-26 08:48:16 +02:00
2012-02-28 17:04:48 +01:00
/* *
* Command to join the cluster . Sent when a node ( reprsesented by 'address' )
* wants to join another node ( the receiver ) .
*/
case class Join ( address : Address ) extends ClusterMessage
2012-02-09 13:36:39 +01:00
2012-02-29 10:02:00 +01:00
/* *
* Command to set a node to Up ( from Joining ) .
*/
case object Up extends ClusterMessage
2012-02-28 17:04:48 +01:00
/* *
* Command to leave the cluster .
*/
case object Leave extends ClusterMessage
/* *
* Command to mark node as temporary down .
*/
case object Down extends ClusterMessage
/* *
* Command to mark a node to be removed from the cluster immediately .
*/
case object Exit extends ClusterMessage
/* *
* Command to remove a node from the cluster immediately .
*/
case object Remove extends ClusterMessage
}
2011-10-26 08:48:16 +02:00
2012-01-30 11:41:41 +01:00
/* *
* Represents the address and the current status of a cluster member node .
*/
case class Member ( address : Address , status : MemberStatus ) extends ClusterMessage
2011-10-26 08:48:16 +02:00
2012-02-09 15:59:10 +01:00
/* *
* Envelope adding a sender address to the gossip .
*/
case class GossipEnvelope ( sender : Member , gossip : Gossip ) extends ClusterMessage
2012-01-24 12:09:32 +01:00
/* *
2012-01-30 11:41:41 +01:00
* Defines the current status of a cluster member node
*
* Can be one of : Joining , Up , Leaving , Exiting and Down .
2012-01-24 12:09:32 +01:00
*/
2012-02-07 16:53:49 +01:00
sealed trait MemberStatus extends ClusterMessage
2012-01-30 11:41:41 +01:00
object MemberStatus {
2012-02-07 16:53:49 +01:00
case object Joining extends MemberStatus
case object Up extends MemberStatus
case object Leaving extends MemberStatus
case object Exiting extends MemberStatus
case object Down extends MemberStatus
2012-02-28 17:04:48 +01:00
case object Removed extends MemberStatus
2012-01-24 12:09:32 +01:00
}
2012-02-09 13:36:39 +01:00
/* *
2012-02-14 20:50:12 +01:00
* Represents the overview of the cluster , holds the cluster convergence table and set with unreachable nodes .
2012-02-09 13:36:39 +01:00
*/
case class GossipOverview (
2012-02-09 15:59:10 +01:00
seen : Map [ Address , VectorClock ] = Map . empty [ Address , VectorClock ] ,
2012-02-14 20:50:12 +01:00
unreachable : Set [ Address ] = Set . empty [ Address ] ) {
override def toString =
"GossipOverview(seen = [" + seen . mkString ( ", " ) +
"], unreachable = [" + unreachable . mkString ( ", " ) +
"])"
}
2012-02-09 13:36:39 +01:00
/* *
* Represents the state of the cluster ; cluster ring membership , ring convergence , meta data - all versioned by a vector clock .
*/
case class Gossip (
overview : GossipOverview = GossipOverview ( ) ,
members : SortedSet [ Member ] , // sorted set of members with their status, sorted by name
meta : Map [ String , Array [ Byte ] ] = Map . empty [ String , Array [ Byte ] ] ,
version : VectorClock = VectorClock ( ) ) // vector clock version
extends ClusterMessage // is a serializable cluster message
2012-02-14 20:50:12 +01:00
with Versioned [ Gossip ] {
/* *
* Increments the version for this 'Node' .
*/
def + ( node : VectorClock . Node ) : Gossip = copy ( version = version + node )
2012-02-09 15:59:10 +01:00
2012-02-14 20:50:12 +01:00
def + ( member : Member ) : Gossip = {
2012-02-09 15:59:10 +01:00
if ( members contains member ) this
else this copy ( members = members + member )
}
/* *
* Marks the gossip as seen by this node ( remoteAddress ) by updating the address entry in the 'gossip. overview . seen '
* Map with the VectorClock for the new gossip .
*/
2012-03-02 09:55:54 +01:00
def seen ( address : Address ) : Gossip = {
if ( overview . seen . contains ( address ) && overview . seen ( address ) == version ) this
else this copy ( overview = overview copy ( seen = overview . seen + ( address -> version ) ) )
}
2012-02-09 15:59:10 +01:00
2012-02-14 20:50:12 +01:00
override def toString =
"Gossip(" +
"overview = " + overview +
", members = [" + members . mkString ( ", " ) +
"], meta = [" + meta . mkString ( ", " ) +
"], version = " + version +
")"
2012-02-09 15:59:10 +01:00
}
2012-01-24 12:09:32 +01:00
2012-02-20 15:26:12 +01:00
/* *
2012-02-28 17:04:48 +01:00
* FSM actor managing the different cluster nodes states .
* Single instance - e . g . serialized access to Node - message after message .
2012-02-20 15:26:12 +01:00
*/
2012-02-28 17:04:48 +01:00
final class ClusterCommandDaemon ( system : ActorSystem , node : Node ) extends Actor with FSM [ MemberStatus , Unit ] {
2012-02-20 15:26:12 +01:00
2012-02-28 17:04:48 +01:00
// start in JOINING
startWith ( MemberStatus . Joining , Unit )
// ========================
// === IN JOINING ===
when ( MemberStatus . Joining ) {
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Up , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . up ( )
goto ( MemberStatus . Up )
}
// ========================
// === IN UP ===
when ( MemberStatus . Up ) {
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Down , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . downing ( )
goto ( MemberStatus . Down )
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Leave , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . leaving ( )
goto ( MemberStatus . Leaving )
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Exit , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . exiting ( )
goto ( MemberStatus . Exiting )
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Remove , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . removing ( )
goto ( MemberStatus . Removed )
}
// ========================
// === IN LEAVING ===
when ( MemberStatus . Leaving ) {
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Down , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . downing ( )
goto ( MemberStatus . Down )
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Remove , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . removing ( )
goto ( MemberStatus . Removed )
}
// ========================
// === IN EXITING ===
when ( MemberStatus . Exiting ) {
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Remove , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . removing ( )
goto ( MemberStatus . Removed )
}
// ========================
// === IN DOWN ===
when ( MemberStatus . Down ) {
// FIXME How to transition from DOWN => JOINING when node comes back online. Can't just listen to Gossip message since it is received be another actor. How to fix this?
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Remove , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . removing ( )
goto ( MemberStatus . Removed )
}
// ========================
// === IN REMOVED ===
when ( MemberStatus . Removed ) {
case command ⇒
2012-02-29 10:02:00 +01:00
log . warning ( "Removed node [{}] received cluster command [{}]" , system . name , command )
2012-02-28 17:04:48 +01:00
stay
}
// ========================
// === GENERIC AND UNHANDLED COMMANDS ===
whenUnhandled {
// should be able to handle Join in any state
2012-02-29 10:02:00 +01:00
case Event ( ClusterAction . Join ( address ) , _ ) ⇒
2012-02-28 17:04:48 +01:00
node . joining ( address )
stay
case Event ( command , _ ) ⇒ {
log . warning ( "Unhandled command [{}] in state [{}]" , command , stateName )
stay
}
2012-02-20 15:26:12 +01:00
}
}
/* *
* Pooled and routed wit N number of configurable instances .
2012-02-20 15:45:50 +01:00
* Concurrent access to Node .
2012-02-20 15:26:12 +01:00
*/
2012-02-20 15:45:50 +01:00
final class ClusterGossipDaemon ( system : ActorSystem , node : Node ) extends Actor {
2012-02-20 15:26:12 +01:00
val log = Logging ( system , "ClusterGossipDaemon" )
2012-01-24 12:09:32 +01:00
def receive = {
2012-02-20 15:45:50 +01:00
case GossipEnvelope ( sender , gossip ) ⇒ node . receive ( sender , gossip )
2012-01-24 12:09:32 +01:00
}
2012-03-02 09:55:54 +01:00
2012-03-02 16:20:30 +01:00
override def unhandled ( unknown : Any ) = log . error ( "Unknown message sent to cluster daemon [{}]" , unknown )
}
final class ClusterDaemonSupervisor ( system : ActorSystem , node : Node ) extends Actor {
val log = Logging ( system , "ClusterDaemonSupervisor" )
override val supervisorStrategy = SupervisorStrategy . defaultStrategy
private val commands = context . actorOf ( Props ( new ClusterCommandDaemon ( system , node ) ) , "commands" )
private val gossip = context . actorOf ( Props ( new ClusterGossipDaemon ( system , node ) )
. withRouter ( RoundRobinRouter ( node . clusterSettings . NrOfGossipDaemons ) ) , "gossip" )
def receive = {
case unknown ⇒ log . error ( "/system/cluster can not respond to messages - received [{}]" , unknown )
}
2012-01-24 12:09:32 +01:00
}
2012-02-22 18:40:16 +01:00
/* *
* Node Extension Id and factory for creating Node extension .
* Example :
* { { {
2012-03-02 09:55:54 +01:00
* val node = Node ( system )
2012-02-29 10:02:00 +01:00
*
* if ( node . isLeader ) { . . . }
* } } }
2012-02-22 18:40:16 +01:00
*/
2012-03-02 09:55:54 +01:00
object Node extends ExtensionId [ Node ] with ExtensionIdProvider {
2012-02-22 18:40:16 +01:00
override def get ( system : ActorSystem ) : Node = super . get ( system )
2012-03-02 09:55:54 +01:00
override def lookup = Node
2012-02-22 18:40:16 +01:00
2012-03-02 09:55:54 +01:00
override def createExtension ( system : ExtendedActorSystem ) : Node = new Node ( system )
2012-02-22 18:40:16 +01:00
}
2012-02-14 20:50:12 +01:00
2011-10-26 08:48:16 +02:00
/* *
* This module is responsible for Gossiping cluster information . The abstraction maintains the list of live
2012-01-30 11:41:41 +01:00
* and dead members . Periodically i . e . every 1 second this module chooses a random member and initiates a round
2011-10-26 08:48:16 +02:00
* of Gossip with it . Whenever it gets gossip updates it updates the Failure Detector with the liveness
* information .
* < p />
2012-01-30 11:41:41 +01:00
* During each of these runs the member initiates gossip exchange according to following rules ( as defined in the
2011-10-26 08:48:16 +02:00
* Cassandra documentation [ http :// wiki . apache . org / cassandra / ArchitectureGossip ] :
* < pre >
2012-01-30 11:41:41 +01:00
* 1 ) Gossip to random live member ( if any )
* 2 ) Gossip to random unreachable member with certain probability depending on number of unreachable and live members
2012-02-07 16:53:49 +01:00
* 3 ) If the member gossiped to at ( 1 ) was not deputy , or the number of live members is less than number of deputy list ,
* gossip to random deputy with certain probability depending on number of unreachable , deputy and live members .
2011-10-26 08:48:16 +02:00
* </ pre >
2012-02-29 10:02:00 +01:00
*
* Example :
* { { {
2012-03-02 09:55:54 +01:00
* val node = Node ( system )
2012-02-29 10:02:00 +01:00
*
* if ( node . isLeader ) { . . . }
* } } }
2011-10-26 08:48:16 +02:00
*/
2012-03-02 09:55:54 +01:00
class Node ( system : ExtendedActorSystem ) extends Extension {
2012-02-22 18:40:16 +01:00
2011-10-26 08:48:16 +02:00
/* *
2012-02-20 15:45:50 +01:00
* Represents the state for this Node . Implemented using optimistic lockless concurrency ,
2011-10-26 08:48:16 +02:00
* all state is represented by this immutable case class and managed by an AtomicReference .
*/
private case class State (
2012-02-09 15:59:10 +01:00
self : Member ,
2012-02-08 14:14:01 +01:00
latestGossip : Gossip ,
2012-02-09 15:59:10 +01:00
memberMembershipChangeListeners : Set [ MembershipChangeListener ] = Set . empty [ MembershipChangeListener ] )
2011-10-26 08:48:16 +02:00
2012-02-29 10:02:00 +01:00
if ( ! system . provider . isInstanceOf [ RemoteActorRefProvider ] )
throw new ConfigurationException ( "ActorSystem[" + system + "] needs to have a 'RemoteActorRefProvider' enabled in the configuration" )
private val remote : RemoteActorRefProvider = system . provider . asInstanceOf [ RemoteActorRefProvider ]
2012-01-30 19:40:28 +01:00
2012-03-02 16:20:30 +01:00
val remoteSettings = new RemoteSettings ( system . settings . config , system . name )
val clusterSettings = new ClusterSettings ( system . settings . config , system . name )
2012-02-07 16:53:49 +01:00
2012-02-29 10:02:00 +01:00
private val remoteAddress = remote . transport . address
private val vclockNode = VectorClock . Node ( remoteAddress . toString )
2012-02-07 16:53:49 +01:00
2012-02-29 10:02:00 +01:00
private val gossipInitialDelay = clusterSettings . GossipInitialDelay
private val gossipFrequency = clusterSettings . GossipFrequency
2012-02-08 16:15:31 +01:00
2012-02-29 10:02:00 +01:00
implicit private val memberOrdering = Ordering . fromLessThan [ Member ] ( _ . address . toString < _ . address . toString )
implicit private val defaultTimeout = Timeout ( remoteSettings . RemoteSystemDaemonAckTimeout )
2012-01-24 12:09:32 +01:00
2012-02-18 17:48:07 +01:00
val failureDetector = new AccrualFailureDetector (
system , remoteAddress , clusterSettings . FailureDetectorThreshold , clusterSettings . FailureDetectorMaxSampleSize )
2011-01-01 01:50:33 +01:00
private val nrOfDeputyNodes = clusterSettings . NrOfDeputyNodes
2012-02-20 15:26:12 +01:00
private val nrOfGossipDaemons = clusterSettings . NrOfGossipDaemons
2012-02-09 15:59:10 +01:00
private val nodeToJoin : Option [ Address ] = clusterSettings . NodeToJoin filter ( _ != remoteAddress )
2011-10-26 08:48:16 +02:00
2012-01-31 15:00:46 +01:00
private val serialization = remote . serialization
2012-01-30 11:41:41 +01:00
private val isRunning = new AtomicBoolean ( true )
2012-02-20 15:45:50 +01:00
private val log = Logging ( system , "Node" )
2011-10-27 15:14:15 +02:00
private val random = SecureRandom . getInstance ( "SHA1PRNG" )
2012-01-30 11:41:41 +01:00
2012-03-02 16:20:30 +01:00
// create superisor for daemons under path "/system/cluster"
private val clusterDaemons = {
val createChild = CreateChild ( Props ( new ClusterDaemonSupervisor ( system , this ) ) , "cluster" )
Await . result ( system . systemGuardian ? createChild , defaultTimeout . duration ) match {
case a : ActorRef ⇒ a
case e : Exception ⇒ throw e
}
}
2012-02-20 15:26:12 +01:00
2012-03-02 16:20:30 +01:00
private val clusterCommandDaemon = system . actorFor ( "/system/cluster/commands" )
2012-02-08 14:14:01 +01:00
private val state = {
val member = Member ( remoteAddress , MemberStatus . Joining )
2012-02-20 15:45:50 +01:00
val gossip = Gossip ( members = SortedSet . empty [ Member ] + member ) + vclockNode // add me as member and update my vector clock
2012-02-09 15:59:10 +01:00
new AtomicReference [ State ] ( State ( member , gossip ) )
2012-02-08 14:14:01 +01:00
}
2011-10-26 08:48:16 +02:00
2012-02-14 20:50:12 +01:00
import Versioned.latestVersionOf
2012-02-20 15:45:50 +01:00
log . info ( "Node [{}] - Starting cluster Node..." , remoteAddress )
2012-01-24 12:09:32 +01:00
2012-02-08 14:14:01 +01:00
// try to join the node defined in the 'akka.cluster.node-to-join' option
2012-02-28 17:04:48 +01:00
autoJoin ( )
2012-01-24 12:09:32 +01:00
2012-02-14 20:50:12 +01:00
// start periodic gossip to random nodes in cluster
2011-01-01 01:50:33 +01:00
private val gossipCanceller = system . scheduler . schedule ( gossipInitialDelay , gossipFrequency ) {
2012-02-08 14:14:01 +01:00
gossip ( )
}
2012-02-14 20:50:12 +01:00
// start periodic cluster scrutinization (moving nodes condemned by the failure detector to unreachable list)
2011-01-01 01:50:33 +01:00
private val scrutinizeCanceller = system . scheduler . schedule ( gossipInitialDelay , gossipFrequency ) {
2012-02-15 15:51:27 +01:00
scrutinize ( )
2012-02-08 14:14:01 +01:00
}
2012-01-24 12:09:32 +01:00
2012-02-14 20:50:12 +01:00
// ======================================================
// ===================== PUBLIC API =====================
// ======================================================
/* *
* Latest gossip .
*/
def latestGossip : Gossip = state . get . latestGossip
/* *
* Member status for this node .
*/
def self : Member = state . get . self
2012-02-20 17:22:07 +01:00
/* *
* Is this node the leader ?
*/
def isLeader : Boolean = {
val currentState = state . get
remoteAddress == currentState . latestGossip . members . head . address
}
2012-02-14 20:50:12 +01:00
/* *
* Is this node a singleton cluster ?
*/
def isSingletonCluster : Boolean = isSingletonCluster ( state . get )
2011-01-01 01:50:33 +01:00
/* *
* Checks if we have a cluster convergence .
*
* @returns Some ( convergedGossip ) if convergence have been reached and None if not
*/
def convergence : Option [ Gossip ] = convergence ( latestGossip )
2012-01-24 12:09:32 +01:00
/* *
2012-01-30 11:41:41 +01:00
* Shuts down all connections to other members , the cluster daemon and the periodic gossip and cleanup tasks .
2012-01-24 12:09:32 +01:00
*/
def shutdown ( ) {
2012-02-14 20:50:12 +01:00
// FIXME Cheating for now. Can't just shut down. Node must first gossip an Leave command, wait for Leader to do proper Handoff and then await an Exit command before switching to Removed
2012-01-30 11:41:41 +01:00
if ( isRunning . compareAndSet ( true , false ) ) {
2012-03-02 16:20:30 +01:00
log . info ( "Node [{}] - Shutting down Node and cluster daemons..." , remoteAddress )
2012-03-02 09:55:54 +01:00
gossipCanceller . cancel ( )
scrutinizeCanceller . cancel ( )
2012-03-02 16:20:30 +01:00
system . stop ( clusterDaemons )
2012-01-30 11:41:41 +01:00
}
2011-10-26 08:48:16 +02:00
}
2011-01-01 01:50:33 +01:00
/* *
* Registers a listener to subscribe to cluster membership changes .
*/
@tailrec
final def registerListener ( listener : MembershipChangeListener ) {
val localState = state . get
val newListeners = localState . memberMembershipChangeListeners + listener
val newState = localState copy ( memberMembershipChangeListeners = newListeners )
if ( ! state . compareAndSet ( localState , newState ) ) registerListener ( listener ) // recur
}
/* *
* Unsubscribes to cluster membership changes .
*/
@tailrec
final def unregisterListener ( listener : MembershipChangeListener ) {
val localState = state . get
val newListeners = localState . memberMembershipChangeListeners - listener
val newState = localState copy ( memberMembershipChangeListeners = newListeners )
if ( ! state . compareAndSet ( localState , newState ) ) unregisterListener ( listener ) // recur
}
2012-02-29 10:02:00 +01:00
/* *
* Send command to JOIN one node to another .
*/
def sendJoin ( address : Address ) {
clusterCommandDaemon ! ClusterAction . Join ( address )
}
/* *
* Send command to issue state transition to LEAVING .
*/
def sendLeave ( ) {
clusterCommandDaemon ! ClusterAction . Leave
}
/* *
* Send command to issue state transition to EXITING .
*/
def sendDown ( ) {
clusterCommandDaemon ! ClusterAction . Down
}
/* *
* Send command to issue state transition to REMOVED .
*/
def sendRemove ( ) {
clusterCommandDaemon ! ClusterAction . Remove
}
2011-01-01 01:50:33 +01:00
// ========================================================
// ===================== INTERNAL API =====================
// ========================================================
2012-02-08 14:14:01 +01:00
/* *
2012-02-28 17:04:48 +01:00
* State transition to JOINING .
2012-02-08 14:14:01 +01:00
* New node joining .
*/
@tailrec
2011-01-01 01:50:33 +01:00
private [ cluster ] final def joining ( node : Address ) {
2012-02-14 20:50:12 +01:00
log . info ( "Node [{}] - Node [{}] is joining" , remoteAddress , node )
2012-02-08 16:15:31 +01:00
2012-02-14 20:50:12 +01:00
val localState = state . get
val localGossip = localState . latestGossip
val localMembers = localGossip . members
2012-02-08 16:15:31 +01:00
2012-02-14 20:50:12 +01:00
val newMembers = localMembers + Member ( node , MemberStatus . Joining ) // add joining node as Joining
val newGossip = localGossip copy ( members = newMembers )
2012-02-20 15:45:50 +01:00
val versionedGossip = newGossip + vclockNode
2012-02-14 20:50:12 +01:00
val seenVersionedGossip = versionedGossip seen remoteAddress
val newState = localState copy ( latestGossip = seenVersionedGossip )
if ( ! state . compareAndSet ( localState , newState ) ) joining ( node ) // recur if we failed update
2012-02-18 22:14:53 +01:00
else {
2012-03-02 09:55:54 +01:00
failureDetector heartbeat node // update heartbeat in failure detector
2012-02-18 22:14:53 +01:00
if ( convergence ( newState . latestGossip ) . isDefined ) {
2012-03-02 09:55:54 +01:00
newState . memberMembershipChangeListeners foreach { _ notify newMembers }
2012-02-18 22:14:53 +01:00
}
}
2012-02-08 14:14:01 +01:00
}
2012-01-24 12:09:32 +01:00
2012-02-28 17:04:48 +01:00
/* *
* State transition to UP .
*/
private [ cluster ] final def up ( ) { }
/* *
* State transition to LEAVING .
*/
private [ cluster ] final def leaving ( ) { }
/* *
* State transition to EXITING .
*/
private [ cluster ] final def exiting ( ) { }
/* *
* State transition to REMOVED .
*/
private [ cluster ] final def removing ( ) { }
/* *
* State transition to DOWN .
*/
private [ cluster ] final def downing ( ) { }
2011-10-26 08:48:16 +02:00
/* *
2012-02-08 14:14:01 +01:00
* Receive new gossip .
2011-10-26 08:48:16 +02:00
*/
2012-02-09 15:59:10 +01:00
@tailrec
2011-01-01 01:50:33 +01:00
private [ cluster ] final def receive ( sender : Member , remoteGossip : Gossip ) {
2012-02-14 20:50:12 +01:00
val localState = state . get
val localGossip = localState . latestGossip
2012-02-09 15:59:10 +01:00
2012-02-14 20:50:12 +01:00
val winningGossip =
if ( remoteGossip . version <> localGossip . version ) {
// concurrent
val mergedGossip = merge ( remoteGossip , localGossip )
2012-02-20 15:45:50 +01:00
val versionedMergedGossip = mergedGossip + vclockNode
2012-02-09 15:59:10 +01:00
2012-02-18 22:14:53 +01:00
log . debug (
"Can't establish a causal relationship between \"remote\" gossip [{}] and \"local\" gossip [{}] - merging them into [{}]" ,
2012-02-14 20:50:12 +01:00
remoteGossip , localGossip , versionedMergedGossip )
versionedMergedGossip
} else if ( remoteGossip . version < localGossip . version ) {
// local gossip is newer
localGossip
} else {
// remote gossip is newer
remoteGossip
}
val newState = localState copy ( latestGossip = winningGossip seen remoteAddress )
2012-02-09 15:59:10 +01:00
// if we won the race then update else try again
2012-02-14 20:50:12 +01:00
if ( ! state . compareAndSet ( localState , newState ) ) receive ( sender , remoteGossip ) // recur if we fail the update
2012-02-18 22:14:53 +01:00
else {
2012-03-02 09:55:54 +01:00
log . debug ( "Node [{}] - Receiving gossip from [{}]" , remoteAddress , sender . address )
failureDetector heartbeat sender . address // update heartbeat in failure detector
2012-02-18 22:14:53 +01:00
if ( convergence ( newState . latestGossip ) . isDefined ) {
2012-03-02 09:55:54 +01:00
newState . memberMembershipChangeListeners foreach { _ notify newState . latestGossip . members }
2012-02-18 22:14:53 +01:00
}
}
2011-10-26 08:48:16 +02:00
}
2012-01-24 12:09:32 +01:00
/* *
2012-02-07 16:53:49 +01:00
* Joins the pre - configured contact point and retrieves current gossip state .
2012-01-24 12:09:32 +01:00
*/
2012-02-28 17:04:48 +01:00
private def autoJoin ( ) = nodeToJoin foreach { address ⇒
2012-02-20 15:26:12 +01:00
val connection = clusterCommandConnectionFor ( address )
2012-02-29 10:02:00 +01:00
val command = ClusterAction . Join ( remoteAddress )
2012-02-20 15:26:12 +01:00
log . info ( "Node [{}] - Sending [{}] to [{}] through connection [{}]" , remoteAddress , command , address , connection )
connection ! command
2012-01-24 12:09:32 +01:00
}
2011-10-26 08:48:16 +02:00
/* *
* Initates a new round of gossip .
*/
2012-02-08 14:14:01 +01:00
private def gossip ( ) {
2012-02-14 20:50:12 +01:00
val localState = state . get
val localGossip = localState . latestGossip
val localMembers = localGossip . members
if ( ! isSingletonCluster ( localState ) ) { // do not gossip if we are a singleton cluster
log . debug ( "Node [{}] - Initiating new round of gossip" , remoteAddress )
2012-02-08 16:15:31 +01:00
2012-02-14 20:50:12 +01:00
val localGossip = localState . latestGossip
val localMembers = localGossip . members
val localMembersSize = localMembers . size
val localUnreachableAddresses = localGossip . overview . unreachable
val localUnreachableSize = localUnreachableAddresses . size
2012-02-08 16:15:31 +01:00
// 1. gossip to alive members
2012-03-02 09:55:54 +01:00
val gossipedToDeputy = gossipToRandomNodeOf ( localMembers map { _ . address } )
2012-02-08 16:15:31 +01:00
2012-02-09 15:59:10 +01:00
// 2. gossip to unreachable members
2012-02-14 20:50:12 +01:00
if ( localUnreachableSize > 0 ) {
val probability : Double = localUnreachableSize / ( localMembersSize + 1 )
2012-03-02 09:55:54 +01:00
if ( random . nextDouble ( ) < probability ) gossipToRandomNodeOf ( localUnreachableAddresses )
2012-02-08 16:15:31 +01:00
}
2011-10-26 08:48:16 +02:00
2012-02-08 16:15:31 +01:00
// 3. gossip to a deputy nodes for facilitating partition healing
2011-01-01 01:50:33 +01:00
val deputies = deputyNodes
2012-02-14 20:50:12 +01:00
if ( ( ! gossipedToDeputy || localMembersSize < 1 ) && ! deputies . isEmpty ) {
if ( localMembersSize == 0 ) gossipToRandomNodeOf ( deputies )
2012-02-08 16:15:31 +01:00
else {
2012-02-14 20:50:12 +01:00
val probability = 1.0 / localMembersSize + localUnreachableSize
2012-02-08 16:15:31 +01:00
if ( random . nextDouble ( ) <= probability ) gossipToRandomNodeOf ( deputies )
}
2011-10-26 08:48:16 +02:00
}
}
}
2012-02-14 20:50:12 +01:00
/* *
* Merges two Gossip instances including membership tables , meta - data tables and the VectorClock histories .
*/
private def merge ( gossip1 : Gossip , gossip2 : Gossip ) : Gossip = {
val mergedVClock = gossip1 . version merge gossip2 . version
val mergedMembers = gossip1 . members union gossip2 . members
val mergedMeta = gossip1 . meta ++ gossip2 . meta
Gossip ( gossip2 . overview , mergedMembers , mergedMeta , mergedVClock )
}
2012-02-09 15:59:10 +01:00
/* *
2012-02-28 17:04:48 +01:00
* Switches the member status .
*
* @param newStatus the new member status
* @param oldState the state to change the member status in
* @return the updated new state with the new member status
2012-02-09 15:59:10 +01:00
*/
2012-02-28 17:04:48 +01:00
private def switchMemberStatusTo ( newStatus : MemberStatus , state : State ) : State = {
2012-02-08 15:11:06 +01:00
log . info ( "Node [{}] - Switching membership status to [{}]" , remoteAddress , newStatus )
2012-02-28 17:04:48 +01:00
val localSelf = state . self
2012-02-08 15:11:06 +01:00
2012-02-28 17:04:48 +01:00
val localGossip = state . latestGossip
2012-02-14 20:50:12 +01:00
val localMembers = localGossip . members
2012-02-08 15:11:06 +01:00
2012-02-14 20:50:12 +01:00
val newSelf = localSelf copy ( status = newStatus )
val newMembersSet = localMembers map { member ⇒
2012-02-08 15:11:06 +01:00
if ( member . address == remoteAddress ) newSelf
else member
}
2012-02-14 20:50:12 +01:00
2012-02-08 15:11:06 +01:00
// ugly crap to work around bug in scala colletions ('val ss: SortedSet[Member] = SortedSet.empty[Member] ++ aSet' does not compile)
2012-02-08 16:15:31 +01:00
val newMembersSortedSet = SortedSet [ Member ] ( newMembersSet . toList : _ * )
2012-02-14 20:50:12 +01:00
val newGossip = localGossip copy ( members = newMembersSortedSet )
2012-02-20 15:45:50 +01:00
val versionedGossip = newGossip + vclockNode
2012-02-14 20:50:12 +01:00
val seenVersionedGossip = versionedGossip seen remoteAddress
2012-02-08 15:11:06 +01:00
2012-02-28 17:04:48 +01:00
state copy ( self = newSelf , latestGossip = seenVersionedGossip )
2012-02-08 15:11:06 +01:00
}
2011-10-26 08:48:16 +02:00
/* *
2012-02-09 15:59:10 +01:00
* Gossips latest gossip to an address .
2012-02-08 14:14:01 +01:00
*/
2012-02-09 15:59:10 +01:00
private def gossipTo ( address : Address ) {
2012-02-20 15:26:12 +01:00
val connection = clusterGossipConnectionFor ( address )
log . debug ( "Node [{}] - Gossiping to [{}]" , remoteAddress , connection )
connection ! GossipEnvelope ( self , latestGossip )
2012-02-08 14:14:01 +01:00
}
/* *
* Gossips latest gossip to a random member in the set of members passed in as argument .
2012-01-24 12:09:32 +01:00
*
2012-02-07 16:53:49 +01:00
* @return 'true' if it gossiped to a "deputy" member .
2011-10-26 08:48:16 +02:00
*/
2012-03-02 09:55:54 +01:00
private def gossipToRandomNodeOf ( addresses : Iterable [ Address ] ) : Boolean = {
val peers = addresses filterNot ( _ == remoteAddress ) // filter out myself
2011-10-26 08:48:16 +02:00
val peer = selectRandomNode ( peers )
2012-02-08 14:14:01 +01:00
gossipTo ( peer )
2011-01-01 01:50:33 +01:00
deputyNodes exists ( peer == _ )
2011-10-26 08:48:16 +02:00
}
2012-02-07 16:53:49 +01:00
/* *
2012-02-18 17:48:07 +01:00
* Scrutinizes the cluster ; marks members detected by the failure detector as unreachable .
2011-10-26 08:48:16 +02:00
*/
@tailrec
final private def scrutinize ( ) {
2012-02-14 20:50:12 +01:00
val localState = state . get
if ( ! isSingletonCluster ( localState ) ) { // do not scrutinize if we are a singleton cluster
val localGossip = localState . latestGossip
val localOverview = localGossip . overview
2012-02-20 17:22:07 +01:00
val localSeen = localOverview . seen
2012-02-14 20:50:12 +01:00
val localMembers = localGossip . members
val localUnreachableAddresses = localGossip . overview . unreachable
val newlyDetectedUnreachableMembers = localMembers filterNot { member ⇒ failureDetector . isAvailable ( member . address ) }
2012-02-09 15:59:10 +01:00
val newlyDetectedUnreachableAddresses = newlyDetectedUnreachableMembers map { _ . address }
if ( ! newlyDetectedUnreachableAddresses . isEmpty ) { // we have newly detected members marked as unavailable
2012-02-08 16:15:31 +01:00
2012-02-14 20:50:12 +01:00
val newMembers = localMembers diff newlyDetectedUnreachableMembers
2012-02-15 15:51:27 +01:00
val newUnreachableAddresses : Set [ Address ] = localUnreachableAddresses ++ newlyDetectedUnreachableAddresses
2012-02-14 20:50:12 +01:00
2012-02-20 17:22:07 +01:00
val newSeen = newUnreachableAddresses . foldLeft ( localSeen ) ( ( currentSeen , address ) ⇒ currentSeen - address )
val newOverview = localOverview copy ( seen = newSeen , unreachable = newUnreachableAddresses )
2012-02-14 20:50:12 +01:00
val newGossip = localGossip copy ( overview = newOverview , members = newMembers )
2012-02-20 15:45:50 +01:00
val versionedGossip = newGossip + vclockNode
2012-02-14 20:50:12 +01:00
val seenVersionedGossip = versionedGossip seen remoteAddress
val newState = localState copy ( latestGossip = seenVersionedGossip )
2012-02-08 16:15:31 +01:00
// if we won the race then update else try again
2012-02-14 20:50:12 +01:00
if ( ! state . compareAndSet ( localState , newState ) ) scrutinize ( ) // recur
2012-02-08 16:15:31 +01:00
else {
2012-03-02 09:55:54 +01:00
log . info ( "Node [{}] - Marking node(s) an unreachable [{}]" , remoteAddress , newlyDetectedUnreachableAddresses . mkString ( ", " ) )
2012-02-18 22:14:53 +01:00
if ( convergence ( newState . latestGossip ) . isDefined ) {
2012-03-02 09:55:54 +01:00
newState . memberMembershipChangeListeners foreach { _ notify newMembers }
2012-02-18 22:14:53 +01:00
}
2012-02-08 16:15:31 +01:00
}
2011-10-26 08:48:16 +02:00
}
}
}
2012-02-18 22:14:53 +01:00
/* *
* Checks if we have a cluster convergence .
*
* @returns Some ( convergedGossip ) if convergence have been reached and None if not
*/
private def convergence ( gossip : Gossip ) : Option [ Gossip ] = {
2012-02-29 10:02:00 +01:00
val overview = gossip . overview
2012-02-29 11:34:11 +01:00
// if (overview.unreachable.isEmpty) { // if there are any unreachable nodes then we can't have a convergence -
// waiting for user to act (issuing DOWN) or leader to act (issuing DOWN through auto-down)
val seen = gossip . overview . seen
2012-03-02 09:55:54 +01:00
val views = seen . values . toSet
2012-02-29 11:34:11 +01:00
if ( views . size == 1 ) {
log . debug ( "Node [{}] - Cluster convergence reached" , remoteAddress )
Some ( gossip )
2012-02-18 22:14:53 +01:00
} else None
2012-02-29 11:34:11 +01:00
// } else None
2012-02-18 22:14:53 +01:00
}
2012-02-07 16:53:49 +01:00
/* *
2012-02-20 15:26:12 +01:00
* Sets up cluster command connection .
2012-02-07 16:53:49 +01:00
*/
2012-03-02 16:20:30 +01:00
private def clusterCommandConnectionFor ( address : Address ) : ActorRef = system . actorFor ( RootActorPath ( address ) / "system" / "cluster" / "commands" )
2012-02-07 16:53:49 +01:00
/* *
2012-02-20 15:26:12 +01:00
* Sets up cluster gossip connection .
2012-02-07 16:53:49 +01:00
*/
2012-03-02 16:20:30 +01:00
private def clusterGossipConnectionFor ( address : Address ) : ActorRef = system . actorFor ( RootActorPath ( address ) / "system" / "cluster" / "gossip" )
2012-01-24 12:09:32 +01:00
2012-03-02 09:55:54 +01:00
private def deputyNodes : Iterable [ Address ] = state . get . latestGossip . members . toIterable map ( _ . address ) drop 1 take nrOfDeputyNodes filter ( _ != remoteAddress )
2012-01-24 12:09:32 +01:00
2012-03-02 09:55:54 +01:00
private def selectRandomNode ( addresses : Iterable [ Address ] ) : Address = addresses . toSeq ( random nextInt addresses . size )
2012-02-14 20:50:12 +01:00
private def isSingletonCluster ( currentState : State ) : Boolean = currentState . latestGossip . members . size == 1
2011-10-26 08:48:16 +02:00
}