125 lines
3.7 KiB
Java
125 lines
3.7 KiB
Java
/**
|
||
* Copyright (C) 2009-2013 Typesafe Inc. <http://www.typesafe.com>
|
||
*/
|
||
package docs.jrouting;
|
||
|
||
import akka.actor.*;
|
||
import akka.remote.routing.RemoteRouterConfig;
|
||
import akka.routing.Broadcast;
|
||
import akka.routing.RoundRobinRouter;
|
||
import akka.testkit.JavaTestKit;
|
||
import docs.jrouting.RouterViaProgramExample.ExampleActor;
|
||
import docs.routing.RouterViaProgramDocSpec.Echo;
|
||
import java.io.Serializable;
|
||
import java.util.Arrays;
|
||
import java.util.HashMap;
|
||
import java.util.Map;
|
||
import org.junit.AfterClass;
|
||
import org.junit.Assert;
|
||
import org.junit.BeforeClass;
|
||
import org.junit.Test;
|
||
|
||
public class RouterViaProgramDocTestBase {
|
||
|
||
static ActorSystem system;
|
||
|
||
@BeforeClass
|
||
public static void setup() {
|
||
system = ActorSystem.create();
|
||
}
|
||
|
||
@AfterClass
|
||
public static void teardown() {
|
||
system.shutdown();
|
||
}
|
||
|
||
private static class JavaTestKitWithSelf extends JavaTestKit {
|
||
public JavaTestKitWithSelf(ActorSystem system) {
|
||
super(system);
|
||
}
|
||
/**
|
||
* Wrap `getRef()` so our examples look like they're within a normal actor.
|
||
*/
|
||
public ActorRef getSelf() {
|
||
return getRef();
|
||
}
|
||
}
|
||
|
||
@Test
|
||
public void demonstrateRouteesFromPaths() {
|
||
new JavaTestKit(system) {{
|
||
//#programmaticRoutingRouteePaths
|
||
ActorRef actor1 = system.actorOf(new Props(ExampleActor.class), "actor1");
|
||
ActorRef actor2 = system.actorOf(new Props(ExampleActor.class), "actor2");
|
||
ActorRef actor3 = system.actorOf(new Props(ExampleActor.class), "actor3");
|
||
Iterable<String> routees = Arrays.asList(
|
||
new String[] { "/user/actor1", "/user/actor2", "/user/actor3" });
|
||
ActorRef router = system.actorOf(
|
||
new Props().withRouter(new RoundRobinRouter(routees)));
|
||
//#programmaticRoutingRouteePaths
|
||
for (int i = 1; i <= 6; i++) {
|
||
router.tell(new ExampleActor.Message(i), null);
|
||
}
|
||
}};
|
||
}
|
||
|
||
@Test
|
||
public void demonstrateBroadcast() {
|
||
new JavaTestKitWithSelf(system) {{
|
||
ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
|
||
//#broadcastDavyJonesWarning
|
||
router.tell(new Broadcast("Watch out for Davy Jones' locker"), getSelf());
|
||
//#broadcastDavyJonesWarning
|
||
receiveN(5, duration("5 seconds"));
|
||
}};
|
||
}
|
||
|
||
@Test
|
||
public void demonstratePoisonPill() {
|
||
new JavaTestKitWithSelf(system) {{
|
||
ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
|
||
watch(router);
|
||
//#poisonPill
|
||
router.tell(PoisonPill.getInstance(), getSelf());
|
||
//#poisonPill
|
||
expectMsgClass(Terminated.class);
|
||
}};
|
||
}
|
||
|
||
@Test
|
||
public void demonstrateBroadcastOfPoisonPill() {
|
||
new JavaTestKitWithSelf(system) {{
|
||
ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
|
||
watch(router);
|
||
//#broadcastPoisonPill
|
||
router.tell(new Broadcast(PoisonPill.getInstance()), getSelf());
|
||
//#broadcastPoisonPill
|
||
expectMsgClass(Terminated.class);
|
||
}};
|
||
}
|
||
|
||
@Test
|
||
public void demonstrateKill() {
|
||
new JavaTestKitWithSelf(system) {{
|
||
ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
|
||
watch(router);
|
||
//#kill
|
||
router.tell(Kill.getInstance(), getSelf());
|
||
//#kill
|
||
expectMsgClass(Terminated.class);
|
||
}};
|
||
}
|
||
|
||
@Test
|
||
public void demonstrateBroadcastOfKill() {
|
||
new JavaTestKitWithSelf(system) {{
|
||
ActorRef router = system.actorOf(new Props(Echo.class).withRouter(new RoundRobinRouter(5)));
|
||
watch(router);
|
||
//#broadcastKill
|
||
router.tell(new Broadcast(Kill.getInstance()), getSelf());
|
||
//#broadcastKill
|
||
expectMsgClass(Terminated.class);
|
||
}};
|
||
}
|
||
|
||
}
|