7299d6b26c
The randomly selected port by Artery could be the same as the randomly selected port the test generates. Fixes #25403
36 lines
1.2 KiB
Scala
36 lines
1.2 KiB
Scala
/**
|
|
* Copyright (C) 2009-2018 Lightbend Inc. <https://www.lightbend.com>
|
|
*/
|
|
|
|
package akka.remote
|
|
|
|
import akka.actor.ActorSystem
|
|
import akka.actor.Address
|
|
import akka.actor.ExtendedActorSystem
|
|
import akka.actor.Extension
|
|
import akka.actor.ExtensionId
|
|
import akka.actor.ExtensionIdProvider
|
|
import akka.remote.artery.ArteryTransport
|
|
|
|
/**
|
|
* Extension provides access to bound addresses.
|
|
*/
|
|
object BoundAddressesExtension extends ExtensionId[BoundAddressesExtension] with ExtensionIdProvider {
|
|
override def get(system: ActorSystem): BoundAddressesExtension = super.get(system)
|
|
|
|
override def lookup = BoundAddressesExtension
|
|
|
|
override def createExtension(system: ExtendedActorSystem): BoundAddressesExtension =
|
|
new BoundAddressesExtension(system)
|
|
}
|
|
|
|
class BoundAddressesExtension(val system: ExtendedActorSystem) extends Extension {
|
|
/**
|
|
* Returns a mapping from a protocol to a set of bound addresses.
|
|
*/
|
|
def boundAddresses: Map[String, Set[Address]] = system.provider
|
|
.asInstanceOf[RemoteActorRefProvider].transport match {
|
|
case artery: ArteryTransport ⇒ Map(ArteryTransport.ProtocolName → Set(artery.bindAddress.address))
|
|
case remoting: Remoting ⇒ remoting.boundAddresses
|
|
}
|
|
}
|