raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

Removed implicit conversions from ActorRef to AssociationEventListener and HandleEventListener

Browse source
This commit is contained in:
Endre Sándor Varga 2012-12-18 12:54:17 +01:00
parent 752e43a0e7
commit 4c7691fe40
13 changed files with 72 additions and 69 deletions
Download patch file Download diff file Expand all files Collapse all files

2
akka-remote/src/main/scala/akka/remote/Endpoint.scala
View file

@ -248,7 +248,7 @@ private[remote] class EndpointWriter(
reader = Some( reader = Some(
context.watch(context.actorOf(Props(new EndpointReader(readerCodec, readerLocalAddress, readerDispatcher)), context.watch(context.actorOf(Props(new EndpointReader(readerCodec, readerLocalAddress, readerDispatcher)),
"endpointReader-" + AddressUrlEncoder(remoteAddress) + "-" + readerId.next()))) "endpointReader-" + AddressUrlEncoder(remoteAddress) + "-" + readerId.next())))
h.readHandlerPromise.success(reader.get) h.readHandlerPromise.success(ActorHandleEventListener(reader.get))
case None ⇒ throw new EndpointException("Internal error: No handle was present during creation of the endpoint" + case None ⇒ throw new EndpointException("Internal error: No handle was present during creation of the endpoint" +
"reader.", null) "reader.", null)
} }

11
akka-remote/src/main/scala/akka/remote/RemoteActorRefProvider.scala
View file

@ -26,13 +26,13 @@ object RemoteActorRefProvider {
private class RemotingTerminator extends Actor with FSM[TerminatorState, Option[Internals]] { private class RemotingTerminator extends Actor with FSM[TerminatorState, Option[Internals]] {
import context.dispatcher import context.dispatcher
val systemGuardian = context.system.asInstanceOf[ExtendedActorSystem].provider.systemGuardian val systemGuardian = context.actorFor("/system")
startWith(Uninitialized, None) startWith(Uninitialized, None)
when(Uninitialized) { when(Uninitialized) {
case Event(i: Internals, _) ⇒ case Event(i: Internals, _) ⇒
systemGuardian.tell(RegisterTerminationHook, self) systemGuardian ! RegisterTerminationHook
goto(Idle) using Some(i) goto(Idle) using Some(i)
} }
@ -46,8 +46,7 @@ object RemoteActorRefProvider {
// TODO: state timeout // TODO: state timeout
when(WaitDaemonShutdown) { when(WaitDaemonShutdown) {
case Event(TerminationHookDone, Some(internals)) ⇒ case Event(TerminationHookDone, Some(internals)) ⇒
log.info("Remote daemon shut down.") log.info("Remote daemon shut down; proceeding with flushing remote transports.")
log.info("Shutting down remoting.")
internals.transport.shutdown() pipeTo self internals.transport.shutdown() pipeTo self
goto(WaitTransportShutdown) goto(WaitTransportShutdown)
} }
@ -55,7 +54,7 @@ object RemoteActorRefProvider {
when(WaitTransportShutdown) { when(WaitTransportShutdown) {
case Event((), _) ⇒ case Event((), _) ⇒
log.info("Remoting shut down.") log.info("Remoting shut down.")
systemGuardian.tell(TerminationHookDone, self) systemGuardian ! TerminationHookDone
stop() stop()
} }
@ -141,8 +140,8 @@ class RemoteActorRefProvider(
}).get }).get
}) })
remotingTerminator ! internals
_internals = internals _internals = internals
remotingTerminator ! internals
_log = Logging(eventStream, "RemoteActorRefProvider") _log = Logging(eventStream, "RemoteActorRefProvider")

5
akka-remote/src/main/scala/akka/remote/Remoting.scala
View file

@ -8,8 +8,7 @@ import akka.japi.Util.immutableSeq
import akka.pattern.{ gracefulStop, pipe, ask } import akka.pattern.{ gracefulStop, pipe, ask }
import akka.remote.EndpointManager._ import akka.remote.EndpointManager._
import akka.remote.Remoting.TransportSupervisor import akka.remote.Remoting.TransportSupervisor
import akka.remote.transport.Transport.AssociationEventListener import akka.remote.transport.Transport.{ ActorAssociationEventListener, AssociationEventListener, InboundAssociation }
import akka.remote.transport.Transport.InboundAssociation
import akka.remote.transport._ import akka.remote.transport._
import akka.util.Timeout import akka.util.Timeout
import com.typesafe.config.Config import com.typesafe.config.Config
@ -386,7 +385,7 @@ private[remote] class EndpointManager(conf: Config, log: LoggingAdapter) extends
// Register to each transport as listener and collect mapping to addresses // Register to each transport as listener and collect mapping to addresses
val transportsAndAddresses = results map { val transportsAndAddresses = results map {
case (transport, address, promise) ⇒ case (transport, address, promise) ⇒
promise.success(self) promise.success(ActorAssociationEventListener(self))
transport -> address transport -> address
} }
addressesPromise.success(transportsAndAddresses) addressesPromise.success(transportsAndAddresses)

9
akka-remote/src/main/scala/akka/remote/transport/AbstractTransportAdapter.scala
View file

@ -54,7 +54,7 @@ trait SchemeAugmenter {
/** /**
* An adapter that wraps a transport and provides interception * An adapter that wraps a transport and provides interception
*/ */
abstract class AbstractTransportAdapter(protected val wrappedTransport: Transport, implicit val ec: ExecutionContext) abstract class AbstractTransportAdapter(protected val wrappedTransport: Transport)(implicit val ec: ExecutionContext)
extends Transport with SchemeAugmenter { extends Transport with SchemeAugmenter {
protected def maximumOverhead: Int protected def maximumOverhead: Int
@ -125,7 +125,7 @@ object ActorTransportAdapter {
} }
abstract class ActorTransportAdapter(wrappedTransport: Transport, system: ActorSystem) abstract class ActorTransportAdapter(wrappedTransport: Transport, system: ActorSystem)
extends AbstractTransportAdapter(wrappedTransport, system.dispatcher) { extends AbstractTransportAdapter(wrappedTransport)(system.dispatcher) {
import ActorTransportAdapter._ import ActorTransportAdapter._
private implicit val timeout = new Timeout(3 seconds) private implicit val timeout = new Timeout(3 seconds)
@ -140,9 +140,12 @@ abstract class ActorTransportAdapter(wrappedTransport: Transport, system: ActorS
override def interceptListen(listenAddress: Address, override def interceptListen(listenAddress: Address,
listenerPromise: Future[AssociationEventListener]): Future[AssociationEventListener] = { listenerPromise: Future[AssociationEventListener]): Future[AssociationEventListener] = {
registerManager().map { mgr ⇒ registerManager().map { mgr ⇒
// Side effecting: storing the manager instance in volatile var
// This is done only once: during the initialization of the protocol stack. The variable manager is not read
// before listen is called.
manager = mgr manager = mgr
manager ! ListenUnderlying(listenAddress, listenerPromise) manager ! ListenUnderlying(listenAddress, listenerPromise)
manager ActorAssociationEventListener(manager)
} }
} }

4
akka-remote/src/main/scala/akka/remote/transport/AkkaProtocolTransport.scala
View file

@ -280,7 +280,7 @@ private[transport] class ProtocolStateActor(initialData: InitialProtocolStateDat
startWith(Closed, d) startWith(Closed, d)
case d: InboundUnassociated ⇒ case d: InboundUnassociated ⇒
d.wrappedHandle.readHandlerPromise.success(self) d.wrappedHandle.readHandlerPromise.success(ActorHandleEventListener(self))
startWith(WaitActivity, d) startWith(WaitActivity, d)
} }
@ -296,7 +296,7 @@ private[transport] class ProtocolStateActor(initialData: InitialProtocolStateDat
stop() stop()
case Event(Ready(wrappedHandle), OutboundUnassociated(_, statusPromise, _)) ⇒ case Event(Ready(wrappedHandle), OutboundUnassociated(_, statusPromise, _)) ⇒
wrappedHandle.readHandlerPromise.success(self) wrappedHandle.readHandlerPromise.success(ActorHandleEventListener(self))
sendAssociate(wrappedHandle) sendAssociate(wrappedHandle)
failureDetector.heartbeat() failureDetector.heartbeat()
initTimers() initTimers()

9
akka-remote/src/main/scala/akka/remote/transport/FailureInjectorTransportAdapter.scala
View file

@ -33,7 +33,7 @@ private[remote] object FailureInjectorTransportAdapter {
} }
private[remote] class FailureInjectorTransportAdapter(wrappedTransport: Transport, val extendedSystem: ExtendedActorSystem) private[remote] class FailureInjectorTransportAdapter(wrappedTransport: Transport, val extendedSystem: ExtendedActorSystem)
extends AbstractTransportAdapter(wrappedTransport, extendedSystem.dispatcher) with AssociationEventListener { extends AbstractTransportAdapter(wrappedTransport)(extendedSystem.dispatcher) with AssociationEventListener {
import extendedSystem.dispatcher import extendedSystem.dispatcher
@ -62,6 +62,9 @@ private[remote] class FailureInjectorTransportAdapter(wrappedTransport: Transpor
listenerFuture: Future[AssociationEventListener]): Future[AssociationEventListener] = { listenerFuture: Future[AssociationEventListener]): Future[AssociationEventListener] = {
log.warning("FailureInjectorTransport is active on this system. Gremlins might munch your packets.") log.warning("FailureInjectorTransport is active on this system. Gremlins might munch your packets.")
listenerFuture.onSuccess { listenerFuture.onSuccess {
// Side effecting: As this class is not an actor, the only way to safely modify state is through volatile vars.
// Listen is called only during the initialization of the stack, and upstreamListener is not read before this
// finishes.
case listener: AssociationEventListener ⇒ upstreamListener = Some(listener) case listener: AssociationEventListener ⇒ upstreamListener = Some(listener)
} }
Future.successful(this) Future.successful(this)
@ -126,8 +129,8 @@ private[remote] case class FailureInjectorHandle(_wrappedHandle: AssociationHand
wrappedHandle.readHandlerPromise.success(this) wrappedHandle.readHandlerPromise.success(this)
} }
override def write(payload: ByteString): Boolean = if (!gremlinAdapter.shouldDropOutbound(wrappedHandle.remoteAddress)) override def write(payload: ByteString): Boolean =
wrappedHandle.write(payload) if (!gremlinAdapter.shouldDropOutbound(wrappedHandle.remoteAddress)) wrappedHandle.write(payload)
else true else true
override def disassociate(): Unit = wrappedHandle.disassociate() override def disassociate(): Unit = wrappedHandle.disassociate()

6
akka-remote/src/main/scala/akka/remote/transport/ThrottlerTransportAdapter.scala
View file

@ -7,7 +7,7 @@ import akka.remote.transport.ActorTransportAdapter.AssociateUnderlying
import akka.remote.transport.ActorTransportAdapter.ListenUnderlying import akka.remote.transport.ActorTransportAdapter.ListenUnderlying
import akka.remote.transport.ActorTransportAdapter.ListenerRegistered import akka.remote.transport.ActorTransportAdapter.ListenerRegistered
import akka.remote.transport.AkkaPduCodec.Associate import akka.remote.transport.AkkaPduCodec.Associate
import akka.remote.transport.AssociationHandle.{ Disassociated, InboundPayload, HandleEventListener } import akka.remote.transport.AssociationHandle.{ ActorHandleEventListener, Disassociated, InboundPayload, HandleEventListener }
import akka.remote.transport.ThrottledAssociation._ import akka.remote.transport.ThrottledAssociation._
import akka.remote.transport.ThrottlerManager.Checkin import akka.remote.transport.ThrottlerManager.Checkin
import akka.remote.transport.ThrottlerTransportAdapter.SetThrottle import akka.remote.transport.ThrottlerTransportAdapter.SetThrottle
@ -255,14 +255,14 @@ private[transport] class ThrottledAssociation(
override def postStop(): Unit = originalHandle.disassociate() override def postStop(): Unit = originalHandle.disassociate()
if (inbound) startWith(WaitExposedHandle, Uninitialized) else { if (inbound) startWith(WaitExposedHandle, Uninitialized) else {
originalHandle.readHandlerPromise.success(self) originalHandle.readHandlerPromise.success(ActorHandleEventListener(self))
startWith(WaitModeAndUpstreamListener, Uninitialized) startWith(WaitModeAndUpstreamListener, Uninitialized)
} }
when(WaitExposedHandle) { when(WaitExposedHandle) {
case Event(handle: ThrottlerHandle, Uninitialized) ⇒ case Event(handle: ThrottlerHandle, Uninitialized) ⇒
// register to downstream layer and wait for origin // register to downstream layer and wait for origin
originalHandle.readHandlerPromise.success(self) originalHandle.readHandlerPromise.success(ActorHandleEventListener(self))
goto(WaitOrigin) using ExposedHandle(handle) goto(WaitOrigin) using ExposedHandle(handle)
} }

4
akka-remote/src/main/scala/akka/remote/transport/Transport.scala
View file

@ -68,8 +68,6 @@ object Transport {
override def notify(ev: AssociationEvent): Unit = actor ! ev override def notify(ev: AssociationEvent): Unit = actor ! ev
} }
implicit def actorRef2HandleEventListener(actor: ActorRef): AssociationEventListener =
ActorAssociationEventListener(actor)
} }
/** /**
@ -199,8 +197,6 @@ object AssociationHandle {
case class ActorHandleEventListener(actor: ActorRef) extends HandleEventListener { case class ActorHandleEventListener(actor: ActorRef) extends HandleEventListener {
override def notify(ev: HandleEvent): Unit = actor ! ev override def notify(ev: HandleEvent): Unit = actor ! ev
} }
implicit def actorRef2HandleEventListener(actor: ActorRef): HandleEventListener = ActorHandleEventListener(actor)
} }
/** /**

2
akka-remote/src/test/scala/akka/remote/RemoteConfigSpec.scala
View file

@ -19,7 +19,7 @@ class RemoteConfigSpec extends AkkaSpec(
} }
""") { """) {
// These tests are ignored as it tests configuration specific to the old remoting. // FIXME: These tests are ignored as it tests configuration specific to the old remoting.
"Remoting" must { "Remoting" must {
"be able to parse generic remote config elements" ignore { "be able to parse generic remote config elements" ignore {

24
akka-remote/src/test/scala/akka/remote/transport/AkkaProtocolSpec.scala
View file

@ -3,7 +3,7 @@ package akka.remote.transport
import akka.actor.{ ExtendedActorSystem, Address, Props } import akka.actor.{ ExtendedActorSystem, Address, Props }
import akka.remote.transport.AkkaPduCodec.{ Disassociate, Associate, Heartbeat } import akka.remote.transport.AkkaPduCodec.{ Disassociate, Associate, Heartbeat }
import akka.remote.transport.AkkaProtocolSpec.TestFailureDetector import akka.remote.transport.AkkaProtocolSpec.TestFailureDetector
import akka.remote.transport.AssociationHandle.{ Disassociated, InboundPayload } import akka.remote.transport.AssociationHandle.{ ActorHandleEventListener, Disassociated, InboundPayload }
import akka.remote.transport.TestTransport._ import akka.remote.transport.TestTransport._
import akka.remote.transport.Transport._ import akka.remote.transport.Transport._
import akka.remote.{ RemoteProtocol, RemoteActorRefProvider, FailureDetector } import akka.remote.{ RemoteProtocol, RemoteActorRefProvider, FailureDetector }
@ -123,7 +123,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
system.actorOf(Props(new ProtocolStateActor( system.actorOf(Props(new ProtocolStateActor(
localAddress, localAddress,
handle, handle,
testActor, ActorAssociationEventListener(testActor),
new AkkaProtocolSettings(conf), new AkkaProtocolSettings(conf),
codec, codec,
failureDetector))) failureDetector)))
@ -137,7 +137,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
val reader = system.actorOf(Props(new ProtocolStateActor( val reader = system.actorOf(Props(new ProtocolStateActor(
localAddress, localAddress,
handle, handle,
testActor, ActorAssociationEventListener(testActor),
new AkkaProtocolSettings(conf), new AkkaProtocolSettings(conf),
codec, codec,
failureDetector))) failureDetector)))
@ -150,7 +150,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
case InboundAssociation(h) ⇒ h case InboundAssociation(h) ⇒ h
} }
wrappedHandle.readHandlerPromise.success(testActor) wrappedHandle.readHandlerPromise.success(ActorHandleEventListener(testActor))
failureDetector.called must be(true) failureDetector.called must be(true)
@ -170,7 +170,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
val reader = system.actorOf(Props(new ProtocolStateActor( val reader = system.actorOf(Props(new ProtocolStateActor(
localAddress, localAddress,
handle, handle,
testActor, ActorAssociationEventListener(testActor),
new AkkaProtocolSettings(conf), new AkkaProtocolSettings(conf),
codec, codec,
failureDetector))) failureDetector)))
@ -257,7 +257,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
val reader = system.actorOf(Props(new ProtocolStateActor( val reader = system.actorOf(Props(new ProtocolStateActor(
localAddress, localAddress,
handle, handle,
testActor, ActorAssociationEventListener(testActor),
new AkkaProtocolSettings(ConfigFactory.parseString("akka.remoting.require-cookie = on").withFallback(conf)), new AkkaProtocolSettings(ConfigFactory.parseString("akka.remoting.require-cookie = on").withFallback(conf)),
codec, codec,
failureDetector))) failureDetector)))
@ -276,7 +276,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
val reader = system.actorOf(Props(new ProtocolStateActor( val reader = system.actorOf(Props(new ProtocolStateActor(
localAddress, localAddress,
handle, handle,
testActor, ActorAssociationEventListener(testActor),
new AkkaProtocolSettings(ConfigFactory.parseString("akka.remoting.require-cookie = on").withFallback(conf)), new AkkaProtocolSettings(ConfigFactory.parseString("akka.remoting.require-cookie = on").withFallback(conf)),
codec, codec,
failureDetector))) failureDetector)))
@ -288,7 +288,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
case InboundAssociation(h) ⇒ h case InboundAssociation(h) ⇒ h
} }
wrappedHandle.readHandlerPromise.success(testActor) wrappedHandle.readHandlerPromise.success(ActorHandleEventListener(testActor))
failureDetector.called must be(true) failureDetector.called must be(true)
@ -350,7 +350,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
case _ ⇒ fail() case _ ⇒ fail()
} }
wrappedHandle.readHandlerPromise.success(testActor) wrappedHandle.readHandlerPromise.success(ActorHandleEventListener(testActor))
lastActivityIsAssociate(registry, None) must be(true) lastActivityIsAssociate(registry, None) must be(true)
@ -388,7 +388,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
case _ ⇒ fail() case _ ⇒ fail()
} }
wrappedHandle.readHandlerPromise.success(testActor) wrappedHandle.readHandlerPromise.success(ActorHandleEventListener(testActor))
Thread.sleep(100) //FIXME: Remove this Thread.sleep(100) //FIXME: Remove this
@ -421,7 +421,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
case _ ⇒ fail() case _ ⇒ fail()
} }
wrappedHandle.readHandlerPromise.success(testActor) wrappedHandle.readHandlerPromise.success(ActorHandleEventListener(testActor))
lastActivityIsAssociate(registry, None) must be(true) lastActivityIsAssociate(registry, None) must be(true)
@ -459,7 +459,7 @@ class AkkaProtocolSpec extends AkkaSpec("""akka.actor.provider = "akka.remote.Re
stateActor ! Disassociated stateActor ! Disassociated
wrappedHandle.readHandlerPromise.success(testActor) wrappedHandle.readHandlerPromise.success(ActorHandleEventListener(testActor))
expectMsg(Disassociated) expectMsg(Disassociated)

2
akka-remote/src/test/scala/akka/remote/transport/AkkaProtocolStressTest.scala
View file

@ -80,7 +80,7 @@ class AkkaProtocolStressTest extends AkkaSpec(configA) with ImplicitSender with
expectMsgPF(30 seconds) { expectMsgPF(30 seconds) {
case (received: Int, lost: Int) ⇒ case (received: Int, lost: Int) ⇒
log.warning(s" ######## Received ${received - lost} messages from ${received} ########") log.debug(s" ######## Received ${received - lost} messages from ${received} ########")
} }
} }
} }

37
akka-remote/src/test/scala/akka/remote/transport/GenericTransportSpec.scala
View file

@ -1,17 +1,20 @@
package akka.remote.transport package akka.remote.transport
import akka.actor.{ ExtendedActorSystem, Address } import akka.actor.{ ExtendedActorSystem, Address }
import akka.remote.transport.AssociationHandle.Disassociated import akka.remote.transport.AssociationHandle.{ ActorHandleEventListener, Disassociated, InboundPayload }
import akka.remote.transport.AssociationHandle.InboundPayload
import akka.remote.transport.TestTransport._ import akka.remote.transport.TestTransport._
import akka.remote.transport.Transport.Fail import akka.remote.transport.Transport._
import akka.remote.transport.Transport.InboundAssociation
import akka.remote.transport.Transport.Ready
import akka.remote.transport.Transport.Status
import akka.testkit.{ ImplicitSender, DefaultTimeout, AkkaSpec } import akka.testkit.{ ImplicitSender, DefaultTimeout, AkkaSpec }
import akka.util.ByteString import akka.util.ByteString
import scala.concurrent.{ Future, Await } import scala.concurrent.{ Future, Await }
import akka.remote.RemoteActorRefProvider import akka.remote.RemoteActorRefProvider
import akka.remote.transport.Transport.InboundAssociation
import akka.remote.transport.TestTransport.DisassociateAttempt
import akka.remote.transport.TestTransport.WriteAttempt
import akka.remote.transport.TestTransport.ListenAttempt
import akka.remote.transport.Transport.Fail
import akka.remote.transport.TestTransport.AssociateAttempt
import akka.remote.transport.Transport.Ready
abstract class GenericTransportSpec(withAkkaProtocol: Boolean = false) abstract class GenericTransportSpec(withAkkaProtocol: Boolean = false)
extends AkkaSpec("""akka.actor.provider = "akka.remote.RemoteActorRefProvider" """) extends AkkaSpec("""akka.actor.provider = "akka.remote.RemoteActorRefProvider" """)
@ -61,8 +64,8 @@ abstract class GenericTransportSpec(withAkkaProtocol: Boolean = false)
val transportB = newTransportB(registry) val transportB = newTransportB(registry)
// Must complete the returned promise to receive events // Must complete the returned promise to receive events
Await.result(transportA.listen, timeout.duration)._2.success(self) Await.result(transportA.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
Await.result(transportB.listen, timeout.duration)._2.success(self) Await.result(transportB.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
awaitCond(registry.transportsReady(addressATest, addressBTest)) awaitCond(registry.transportsReady(addressATest, addressBTest))
@ -79,7 +82,7 @@ abstract class GenericTransportSpec(withAkkaProtocol: Boolean = false)
val registry = new AssociationRegistry val registry = new AssociationRegistry
val transportA = newTransportA(registry) val transportA = newTransportA(registry)
Await.result(transportA.listen, timeout.duration)._2.success(self) Await.result(transportA.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
awaitCond(registry.transportsReady(addressATest)) awaitCond(registry.transportsReady(addressATest))
Await.result(transportA.associate(nonExistingAddress), timeout.duration) match { Await.result(transportA.associate(nonExistingAddress), timeout.duration) match {
@ -93,8 +96,8 @@ abstract class GenericTransportSpec(withAkkaProtocol: Boolean = false)
val transportA = newTransportA(registry) val transportA = newTransportA(registry)
val transportB = newTransportB(registry) val transportB = newTransportB(registry)
Await.result(transportA.listen, timeout.duration)._2.success(self) Await.result(transportA.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
Await.result(transportB.listen, timeout.duration)._2.success(self) Await.result(transportB.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
awaitCond(registry.transportsReady(addressATest, addressBTest)) awaitCond(registry.transportsReady(addressATest, addressBTest))
@ -106,8 +109,8 @@ abstract class GenericTransportSpec(withAkkaProtocol: Boolean = false)
val Ready(handleA) = Await.result(associate, timeout.duration) val Ready(handleA) = Await.result(associate, timeout.duration)
// Initialize handles // Initialize handles
handleA.readHandlerPromise.success(self) handleA.readHandlerPromise.success(ActorHandleEventListener(self))
handleB.readHandlerPromise.success(self) handleB.readHandlerPromise.success(ActorHandleEventListener(self))
val payload = ByteString("PDU") val payload = ByteString("PDU")
val pdu = if (withAkkaProtocol) AkkaPduProtobufCodec.constructPayload(payload) else payload val pdu = if (withAkkaProtocol) AkkaPduProtobufCodec.constructPayload(payload) else payload
@ -131,8 +134,8 @@ abstract class GenericTransportSpec(withAkkaProtocol: Boolean = false)
val transportA = newTransportA(registry) val transportA = newTransportA(registry)
val transportB = newTransportB(registry) val transportB = newTransportB(registry)
Await.result(transportA.listen, timeout.duration)._2.success(self) Await.result(transportA.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
Await.result(transportB.listen, timeout.duration)._2.success(self) Await.result(transportB.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
awaitCond(registry.transportsReady(addressATest, addressBTest)) awaitCond(registry.transportsReady(addressATest, addressBTest))
@ -144,8 +147,8 @@ abstract class GenericTransportSpec(withAkkaProtocol: Boolean = false)
val Ready(handleA) = Await.result(associate, timeout.duration) val Ready(handleA) = Await.result(associate, timeout.duration)
// Initialize handles // Initialize handles
handleA.readHandlerPromise.success(self) handleA.readHandlerPromise.success(ActorHandleEventListener(self))
handleB.readHandlerPromise.success(self) handleB.readHandlerPromise.success(ActorHandleEventListener(self))
awaitCond(registry.existsAssociation(addressATest, addressBTest)) awaitCond(registry.existsAssociation(addressATest, addressBTest))

24
akka-remote/src/test/scala/akka/remote/transport/TestTransportSpec.scala
View file

@ -6,7 +6,7 @@ import akka.actor.Address
import akka.remote.transport.Transport._ import akka.remote.transport.Transport._
import akka.remote.transport.TestTransport._ import akka.remote.transport.TestTransport._
import akka.util.ByteString import akka.util.ByteString
import akka.remote.transport.AssociationHandle.{ Disassociated, InboundPayload } import akka.remote.transport.AssociationHandle.{ ActorHandleEventListener, Disassociated, InboundPayload }
class TestTransportSpec extends AkkaSpec with DefaultTimeout with ImplicitSender { class TestTransportSpec extends AkkaSpec with DefaultTimeout with ImplicitSender {
@ -37,8 +37,8 @@ class TestTransportSpec extends AkkaSpec with DefaultTimeout with ImplicitSender
val transportB = new TestTransport(addressB, registry) val transportB = new TestTransport(addressB, registry)
// Must complete the returned promise to receive events // Must complete the returned promise to receive events
Await.result(transportA.listen, timeout.duration)._2.success(self) Await.result(transportA.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
Await.result(transportB.listen, timeout.duration)._2.success(self) Await.result(transportB.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
awaitCond(registry.transportsReady(addressA, addressB)) awaitCond(registry.transportsReady(addressA, addressB))
@ -54,7 +54,7 @@ class TestTransportSpec extends AkkaSpec with DefaultTimeout with ImplicitSender
val registry = new AssociationRegistry val registry = new AssociationRegistry
var transportA = new TestTransport(addressA, registry) var transportA = new TestTransport(addressA, registry)
Await.result(transportA.listen, timeout.duration)._2.success(self) Await.result(transportA.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
Await.result(transportA.associate(nonExistingAddress), timeout.duration) match { Await.result(transportA.associate(nonExistingAddress), timeout.duration) match {
case Fail(_) ⇒ case Fail(_) ⇒
case _ ⇒ fail() case _ ⇒ fail()
@ -66,8 +66,8 @@ class TestTransportSpec extends AkkaSpec with DefaultTimeout with ImplicitSender
val transportA = new TestTransport(addressA, registry) val transportA = new TestTransport(addressA, registry)
val transportB = new TestTransport(addressB, registry) val transportB = new TestTransport(addressB, registry)
Await.result(transportA.listen, timeout.duration)._2.success(self) Await.result(transportA.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
Await.result(transportB.listen, timeout.duration)._2.success(self) Await.result(transportB.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
awaitCond(registry.transportsReady(addressA, addressB)) awaitCond(registry.transportsReady(addressA, addressB))
@ -79,8 +79,8 @@ class TestTransportSpec extends AkkaSpec with DefaultTimeout with ImplicitSender
val Ready(handleA) = Await.result(associate, timeout.duration) val Ready(handleA) = Await.result(associate, timeout.duration)
// Initialize handles // Initialize handles
handleA.readHandlerPromise.success(self) handleA.readHandlerPromise.success(ActorHandleEventListener(self))
handleB.readHandlerPromise.success(self) handleB.readHandlerPromise.success(ActorHandleEventListener(self))
val akkaPDU = ByteString("AkkaPDU") val akkaPDU = ByteString("AkkaPDU")
@ -103,8 +103,8 @@ class TestTransportSpec extends AkkaSpec with DefaultTimeout with ImplicitSender
val transportA = new TestTransport(addressA, registry) val transportA = new TestTransport(addressA, registry)
val transportB = new TestTransport(addressB, registry) val transportB = new TestTransport(addressB, registry)
Await.result(transportA.listen, timeout.duration)._2.success(self) Await.result(transportA.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
Await.result(transportB.listen, timeout.duration)._2.success(self) Await.result(transportB.listen, timeout.duration)._2.success(ActorAssociationEventListener(self))
awaitCond(registry.transportsReady(addressA, addressB)) awaitCond(registry.transportsReady(addressA, addressB))
@ -116,8 +116,8 @@ class TestTransportSpec extends AkkaSpec with DefaultTimeout with ImplicitSender
val Ready(handleA) = Await.result(associate, timeout.duration) val Ready(handleA) = Await.result(associate, timeout.duration)
// Initialize handles // Initialize handles
handleA.readHandlerPromise.success(self) handleA.readHandlerPromise.success(ActorHandleEventListener(self))
handleB.readHandlerPromise.success(self) handleB.readHandlerPromise.success(ActorHandleEventListener(self))
awaitCond(registry.existsAssociation(addressA, addressB)) awaitCond(registry.existsAssociation(addressA, addressB))