a3e52078df
Seems when did the changes for 2018 it intro introduced a space in all after, hence so many changes.
146 lines
4.1 KiB
Scala
146 lines
4.1 KiB
Scala
/*
|
|
* Copyright (C) 2018 Lightbend Inc. <https://www.lightbend.com>
|
|
*/
|
|
|
|
package akka.cluster.metrics.sample
|
|
|
|
import akka.actor.Props
|
|
import akka.cluster.Cluster
|
|
import akka.cluster.ClusterEvent.{ CurrentClusterState, MemberUp }
|
|
|
|
import scala.concurrent.duration._
|
|
import scala.language.postfixOps
|
|
|
|
//#MultiNodeConfig
|
|
import akka.remote.testkit.MultiNodeConfig
|
|
import com.typesafe.config.ConfigFactory
|
|
|
|
object StatsSampleSpecConfig extends MultiNodeConfig {
|
|
// register the named roles (nodes) of the test
|
|
val first = role("first")
|
|
val second = role("second")
|
|
val third = role("third")
|
|
|
|
def nodeList = Seq(first, second, third)
|
|
|
|
// Extract individual sigar library for every node.
|
|
nodeList foreach { role ⇒
|
|
nodeConfig(role) {
|
|
ConfigFactory.parseString(s"""
|
|
# Enable metrics extension in akka-cluster-metrics.
|
|
akka.extensions=["akka.cluster.metrics.ClusterMetricsExtension"]
|
|
# Sigar native library extract location during tests.
|
|
akka.cluster.metrics.native-library-extract-folder=target/native/${role.name}
|
|
""")
|
|
}
|
|
}
|
|
|
|
// this configuration will be used for all nodes
|
|
// note that no fixed host names and ports are used
|
|
commonConfig(ConfigFactory.parseString("""
|
|
akka.actor.provider = cluster
|
|
akka.remote.log-remote-lifecycle-events = off
|
|
akka.cluster.roles = [compute]
|
|
#//#router-lookup-config
|
|
akka.actor.deployment {
|
|
/statsService/workerRouter {
|
|
router = consistent-hashing-group
|
|
routees.paths = ["/user/statsWorker"]
|
|
cluster {
|
|
enabled = on
|
|
allow-local-routees = on
|
|
use-roles = ["compute"]
|
|
}
|
|
}
|
|
}
|
|
#//#router-lookup-config
|
|
"""))
|
|
|
|
}
|
|
//#MultiNodeConfig
|
|
|
|
//#concrete-tests
|
|
// need one concrete test class per node
|
|
class StatsSampleSpecMultiJvmNode1 extends StatsSampleSpec
|
|
class StatsSampleSpecMultiJvmNode2 extends StatsSampleSpec
|
|
class StatsSampleSpecMultiJvmNode3 extends StatsSampleSpec
|
|
//#concrete-tests
|
|
|
|
//#abstract-test
|
|
import akka.remote.testkit.MultiNodeSpec
|
|
import akka.testkit.ImplicitSender
|
|
import org.scalatest.{ BeforeAndAfterAll, Matchers, WordSpecLike }
|
|
|
|
abstract class StatsSampleSpec extends MultiNodeSpec(StatsSampleSpecConfig)
|
|
with WordSpecLike with Matchers with BeforeAndAfterAll
|
|
with ImplicitSender {
|
|
|
|
import StatsSampleSpecConfig._
|
|
|
|
override def initialParticipants = roles.size
|
|
|
|
override def beforeAll() = multiNodeSpecBeforeAll()
|
|
|
|
override def afterAll() = multiNodeSpecAfterAll()
|
|
|
|
//#abstract-test
|
|
|
|
"The stats sample" must {
|
|
|
|
//#startup-cluster
|
|
"illustrate how to startup cluster" in within(15 seconds) {
|
|
Cluster(system).subscribe(testActor, classOf[MemberUp])
|
|
expectMsgClass(classOf[CurrentClusterState])
|
|
|
|
//#addresses
|
|
val firstAddress = node(first).address
|
|
val secondAddress = node(second).address
|
|
val thirdAddress = node(third).address
|
|
//#addresses
|
|
|
|
//#join
|
|
Cluster(system) join firstAddress
|
|
//#join
|
|
|
|
system.actorOf(Props[StatsWorker], "statsWorker")
|
|
system.actorOf(Props[StatsService], "statsService")
|
|
|
|
receiveN(3).collect { case MemberUp(m) ⇒ m.address }.toSet should be(
|
|
Set(firstAddress, secondAddress, thirdAddress))
|
|
|
|
Cluster(system).unsubscribe(testActor)
|
|
|
|
testConductor.enter("all-up")
|
|
}
|
|
//#startup-cluster
|
|
|
|
//#test-statsService
|
|
"show usage of the statsService from one node" in within(15 seconds) {
|
|
runOn(second) {
|
|
assertServiceOk()
|
|
}
|
|
|
|
testConductor.enter("done-2")
|
|
}
|
|
|
|
def assertServiceOk(): Unit = {
|
|
val service = system.actorSelection(node(third) / "user" / "statsService")
|
|
// eventually the service should be ok,
|
|
// first attempts might fail because worker actors not started yet
|
|
awaitAssert {
|
|
service ! StatsJob("this is the text that will be analyzed")
|
|
expectMsgType[StatsResult](1.second).meanWordLength should be(
|
|
3.875 +- 0.001)
|
|
}
|
|
|
|
}
|
|
//#test-statsService
|
|
|
|
"show usage of the statsService from all nodes" in within(15 seconds) {
|
|
assertServiceOk()
|
|
testConductor.enter("done-3")
|
|
}
|
|
|
|
}
|
|
|
|
}
|