pekko/akka-remote/src/main/scala/akka/remote/NetworkEventStream.scala
Roland edceda8edf restructure remoting
- remove Remote, incorporate its few fields into RemoteActorRefProvider
- rename RemoteSupport to RemoteTransport to sync up with conf and
  the way we talk about it
- remove LocalAddress/RemoteAddress etc. and just have a final case
  class Address(protocol, system, host, port)
- split netty settings out or RemoteSettings into NettySettings
- split out from NettyRemoteSupport.scala: Server.scala, Client.scala,
  Settings.scala

plus a few fixes, including using the contextClassLoader when loading
the provider for ActorSystemImpl
2012-01-20 14:41:15 +01:00

79 lines
2.3 KiB
Scala

/**
* Copyright (C) 2009-2011 Typesafe Inc. <http://www.typesafe.com>
*/
package akka.remote
import scala.collection.mutable
import akka.actor.{ LocalActorRef, Actor, ActorRef, Props, newUuid }
import akka.actor.Actor._
import akka.actor.ActorSystemImpl
import akka.actor.Address
/**
* Stream of all kinds of network events, remote failure and connection events, cluster failure and connection events etc.
* Also provides API for sender listener management.
*/
object NetworkEventStream {
private sealed trait NetworkEventStreamEvent
private case class Register(listener: Listener, connectionAddress: Address)
extends NetworkEventStreamEvent
private case class Unregister(listener: Listener, connectionAddress: Address)
extends NetworkEventStreamEvent
/**
* Base trait for network event listener.
*/
trait Listener {
def notify(event: RemoteLifeCycleEvent)
}
/**
* Channel actor with a registry of listeners.
*/
private class Channel extends Actor {
val listeners = new mutable.HashMap[Address, mutable.Set[Listener]]() {
override def default(k: Address) = mutable.Set.empty[Listener]
}
def receive = {
case event: RemoteClientLifeCycleEvent
listeners(event.remoteAddress) foreach (_ notify event)
case event: RemoteServerLifeCycleEvent // FIXME handle RemoteServerLifeCycleEvent, ticket #1408 and #1190
case Register(listener, connectionAddress)
listeners(connectionAddress) += listener
case Unregister(listener, connectionAddress)
listeners(connectionAddress) -= listener
case _ //ignore other
}
}
}
class NetworkEventStream(system: ActorSystemImpl) {
import NetworkEventStream._
// FIXME: check that this supervision is correct, ticket #1408
private[akka] val sender =
system.systemActorOf(Props[Channel].withDispatcher("akka.remote.network-event-sender-dispatcher"), "network-event-sender")
/**
* Registers a network event stream listener (asyncronously).
*/
def register(listener: Listener, connectionAddress: Address) =
sender ! Register(listener, connectionAddress)
/**
* Unregisters a network event stream listener (asyncronously) .
*/
def unregister(listener: Listener, connectionAddress: Address) =
sender ! Unregister(listener, connectionAddress)
}