85 lines
2.8 KiB
Java
85 lines
2.8 KiB
Java
package docs.cluster;
|
|
|
|
import akka.cluster.routing.ClusterRouterGroup;
|
|
import akka.cluster.routing.ClusterRouterGroupSettings;
|
|
import akka.cluster.routing.ClusterRouterPool;
|
|
import akka.cluster.routing.ClusterRouterPoolSettings;
|
|
import akka.routing.ConsistentHashingGroup;
|
|
import akka.routing.ConsistentHashingPool;
|
|
import docs.cluster.StatsMessages.StatsJob;
|
|
import akka.actor.ActorRef;
|
|
import akka.actor.Props;
|
|
import akka.actor.UntypedActor;
|
|
import akka.routing.ConsistentHashingRouter.ConsistentHashableEnvelope;
|
|
import akka.routing.FromConfig;
|
|
|
|
import java.util.Collections;
|
|
|
|
//#service
|
|
public class StatsService extends UntypedActor {
|
|
|
|
// This router is used both with lookup and deploy of routees. If you
|
|
// have a router with only lookup of routees you can use Props.empty()
|
|
// instead of Props.create(StatsWorker.class).
|
|
ActorRef workerRouter = getContext().actorOf(
|
|
FromConfig.getInstance().props(Props.create(StatsWorker.class)),
|
|
"workerRouter");
|
|
|
|
@Override
|
|
public void onReceive(Object message) {
|
|
if (message instanceof StatsJob) {
|
|
StatsJob job = (StatsJob) message;
|
|
if (job.getText().equals("")) {
|
|
unhandled(message);
|
|
} else {
|
|
final String[] words = job.getText().split(" ");
|
|
final ActorRef replyTo = getSender();
|
|
|
|
// create actor that collects replies from workers
|
|
ActorRef aggregator = getContext().actorOf(
|
|
Props.create(StatsAggregator.class, words.length, replyTo));
|
|
|
|
// send each word to a worker
|
|
for (String word : words) {
|
|
workerRouter.tell(new ConsistentHashableEnvelope(word, word),
|
|
aggregator);
|
|
}
|
|
}
|
|
|
|
} else {
|
|
unhandled(message);
|
|
}
|
|
}
|
|
}
|
|
//#service
|
|
|
|
//not used, only for documentation
|
|
abstract class StatsService2 extends UntypedActor {
|
|
//#router-lookup-in-code
|
|
int totalInstances = 100;
|
|
Iterable<String> routeesPaths = Collections
|
|
.singletonList("/user/statsWorker");
|
|
boolean allowLocalRoutees = true;
|
|
String useRole = "compute";
|
|
ActorRef workerRouter = getContext().actorOf(
|
|
new ClusterRouterGroup(new ConsistentHashingGroup(routeesPaths),
|
|
new ClusterRouterGroupSettings(totalInstances, routeesPaths,
|
|
allowLocalRoutees, useRole)).props(), "workerRouter2");
|
|
//#router-lookup-in-code
|
|
}
|
|
|
|
//not used, only for documentation
|
|
abstract class StatsService3 extends UntypedActor {
|
|
//#router-deploy-in-code
|
|
int totalInstances = 100;
|
|
int maxInstancesPerNode = 3;
|
|
boolean allowLocalRoutees = false;
|
|
String useRole = "compute";
|
|
ActorRef workerRouter = getContext().actorOf(
|
|
new ClusterRouterPool(new ConsistentHashingPool(0),
|
|
new ClusterRouterPoolSettings(totalInstances, maxInstancesPerNode,
|
|
allowLocalRoutees, useRole)).props(Props
|
|
.create(StatsWorker.class)), "workerRouter3");
|
|
//#router-deploy-in-code
|
|
}
|
|
|