Moved the cluster impls to their own modules

This commit is contained in:
Viktor Klang 2010-01-11 22:36:57 +01:00
parent a9371c229b
commit 02a859e551
8 changed files with 226 additions and 154 deletions

View file

@ -4,8 +4,6 @@
package se.scalablesolutions.akka.remote
import org.jgroups.{JChannel, View, Address, Message, ExtendedMembershipListener, Receiver}
import se.scalablesolutions.akka.Config.config
import se.scalablesolutions.akka.config.ScalaConfig._
import se.scalablesolutions.akka.remote.Cluster.{Node, RelayedMessage}
@ -83,149 +81,3 @@ object Cluster extends Cluster {
def shutdown = supervisor.foreach(_.stop)
}
/**
* JGroups Internal Cluster messages.
*/
private[remote] object JGroupsClusterActor {
sealed trait JGroupsClusterMessage
case object PapersPlease extends JGroupsClusterMessage
case class Papers(addresses: List[RemoteAddress]) extends JGroupsClusterMessage
case object Block extends JGroupsClusterMessage
case object Unblock extends JGroupsClusterMessage
case class Zombie(address: Address) extends JGroupsClusterMessage
case class RegisterLocalNode(server: RemoteAddress) extends JGroupsClusterMessage
case class DeregisterLocalNode(server: RemoteAddress) extends JGroupsClusterMessage
}
/**
* Clustering support via JGroups.
*/
class JGroupsClusterActor extends ClusterActor {
import JGroupsClusterActor._
import org.scala_tools.javautils.Implicits._
@volatile private var isActive = false
@volatile private var local: Node = Node(Nil)
@volatile private var channel: Option[JChannel] = None
@volatile private var remotes: Map[Address, Node] = Map()
override def init = {
log debug "Initiating JGroups-based cluster actor"
remotes = new HashMap[Address, Node]
val me = this
isActive = true
// Set up the JGroups local endpoint
channel = Some(new JChannel {
setReceiver(new Receiver with ExtendedMembershipListener {
def getState: Array[Byte] = null
def setState(state: Array[Byte]): Unit = ()
def receive(msg: Message): Unit = if (isActive) me send msg
def viewAccepted(view: View): Unit = if (isActive) me send view
def suspect(a: Address): Unit = if (isActive) me send Zombie(a)
def block: Unit = if (isActive) me send Block
def unblock: Unit = if (isActive) me send Unblock
})
})
channel.map(_.connect(name))
}
def lookup[T](handleRemoteAddress: PartialFunction[RemoteAddress, T]): Option[T] =
remotes.values.toList.flatMap(_.endpoints).find(handleRemoteAddress isDefinedAt _).map(handleRemoteAddress)
def registerLocalNode(hostname: String, port: Int): Unit =
send(RegisterLocalNode(RemoteAddress(hostname, port)))
def deregisterLocalNode(hostname: String, port: Int): Unit =
send(DeregisterLocalNode(RemoteAddress(hostname, port)))
def relayMessage(to: Class[_ <: Actor], msg: AnyRef): Unit =
send(RelayedMessage(to.getName, msg))
private def broadcast[T <: AnyRef](recipients: Iterable[Address], msg: T): Unit = {
lazy val m = Cluster.serializer out msg
for (c <- channel; r <- recipients) c.send(new Message(r, null, m))
}
private def broadcast[T <: AnyRef](msg: T): Unit =
//Don't broadcast if we are not connected anywhere...
if (!remotes.isEmpty) channel.map(_.send(new Message(null, null, Cluster.serializer out msg)))
def receive = {
case Zombie(x) => { //Ask the presumed zombie for papers and prematurely treat it as dead
log debug ("Killing Zombie Node: %s", x)
broadcast(x :: Nil, PapersPlease)
remotes = remotes - x
}
case v: View => {
// Not present in the cluster anymore = presumably zombies
// Nodes we have no prior knowledge existed = unknowns
val members = Set[Address]() ++ v.getMembers.asScala - channel.get.getAddress // Exclude ourselves
val zombies = Set[Address]() ++ remotes.keySet -- members
val unknown = members -- remotes.keySet
log debug v.printDetails
// Tell the zombies and unknowns to provide papers and prematurely treat the zombies as dead
broadcast(zombies ++ unknown, PapersPlease)
remotes = remotes -- zombies
}
case m: Message => {
if (m.getSrc != channel.map(_.getAddress).getOrElse(m.getSrc)) // Handle non-own messages only, and only if we're connected
(Cluster.serializer in (m.getRawBuffer, None)) match {
case PapersPlease => {
log debug ("Asked for papers by %s", m.getSrc)
broadcast(m.getSrc :: Nil, Papers(local.endpoints))
if (remotes.get(m.getSrc).isEmpty) // If we were asked for papers from someone we don't know, ask them!
broadcast(m.getSrc :: Nil, PapersPlease)
}
case Papers(x) => remotes = remotes + (m.getSrc -> Node(x))
case RelayedMessage(c, m) => ActorRegistry.actorsFor(c).map(_ send m)
case unknown => log debug ("Unknown message: %s", unknown.toString)
}
}
case rm @ RelayedMessage(_, _) => {
log debug ("Relaying message: %s", rm)
broadcast(rm)
}
case RegisterLocalNode(s) => {
log debug ("RegisterLocalNode: %s", s)
local = Node(local.endpoints + s)
broadcast(Papers(local.endpoints))
}
case DeregisterLocalNode(s) => {
log debug ("DeregisterLocalNode: %s", s)
local = Node(local.endpoints - s)
broadcast(Papers(local.endpoints))
}
case Block => log debug "UNSUPPORTED: JGroupsClusterActor::block" //TODO HotSwap to a buffering body
case Unblock => log debug "UNSUPPORTED: JGroupsClusterActor::unblock" //TODO HotSwap back and flush the buffer
}
override def shutdown = {
log debug ("Shutting down %s", toString)
isActive = false
channel.foreach(_.shutdown)
remotes = Map()
channel = None
}
}