!act,sam #3889 Adding Activator template FSM/become for Java with Lambda support

* Dining Hakkers Activator template for FSM and become
* Cleaup of FSM event matchers to be more usable and consistent
This commit is contained in:
Björn Antonsson 2014-02-24 12:11:02 +01:00
parent 6af33381b3
commit 07e361c684
30 changed files with 1026 additions and 103 deletions

View file

@ -0,0 +1,162 @@
package sample.become;
import akka.actor.*;
import akka.japi.pf.ReceiveBuilder;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import scala.concurrent.duration.Duration;
import scala.concurrent.duration.FiniteDuration;
import scala.PartialFunction;
import scala.runtime.BoxedUnit;
import static sample.become.Messages.*;
import static java.util.concurrent.TimeUnit.*;
// Akka adaptation of
// http://www.dalnefre.com/wp/2010/08/dining-philosophers-in-humus/
public class DiningHakkersOnBecome {
/*
* A Chopstick is an actor, it can be taken, and put back
*/
public static class Chopstick extends AbstractActor {
//When a Chopstick is taken by a hakker
//It will refuse to be taken by other hakkers
//But the owning hakker can put it back
PartialFunction<Object, BoxedUnit> takenBy(ActorRef hakker) {
return ReceiveBuilder.
match(Take.class,
t -> t.hakker.tell(new Busy(self()), self())).
match(Put.class, p -> p.hakker == hakker,
p -> context().become(available)).
build();
}
//When a Chopstick is available, it can be taken by a hakker
PartialFunction<Object, BoxedUnit> available = ReceiveBuilder.
match(Take.class, t -> {
context().become(takenBy(t.hakker));
t.hakker.tell(new Taken(self()), self());
}).build();
//A Chopstick begins its existence as available
public PartialFunction<Object, BoxedUnit> receive() {
return available;
}
}
/*
* A hakker is an awesome dude or dudette who either thinks about hacking or has to eat ;-)
*/
public static class Hakker extends AbstractActor {
private String name;
private ActorRef left;
private ActorRef right;
public Hakker(String name, ActorRef left, ActorRef right) {
this.name = name;
this.left = left;
this.right = right;
}
//When a hakker is eating, he can decide to start to think,
//then he puts down his chopsticks and starts to think
PartialFunction<Object, BoxedUnit> eating = ReceiveBuilder.
matchEquals(Think, m -> {
left.tell(new Put(self()), self());
right.tell(new Put(self()), self());
System.out.println(String.format("%s puts down his chopsticks and starts to think", name));
startThinking(Duration.create(5, SECONDS));
}).build();
//When a hakker is waiting for the last chopstick it can either obtain it
//and start eating, or the other chopstick was busy, and the hakker goes
//back to think about how he should obtain his chopsticks :-)
PartialFunction<Object, BoxedUnit> waitingFor(ActorRef chopstickToWaitFor, ActorRef otherChopstick) {
return ReceiveBuilder.
match(Taken.class, t -> t.chopstick == chopstickToWaitFor, t -> {
System.out.println(String.format("%s has picked up %s and %s and starts to eat",
name, left.path().name(), right.path().name()));
context().become(eating);
context().system().scheduler().scheduleOnce(Duration.create(5, SECONDS), self(), Think, context().system().dispatcher(), self());
}).
match(Busy.class, b -> {
otherChopstick.tell(new Put(self()), self());
startThinking(Duration.create(10, MILLISECONDS));
}).
build();
}
//When the results of the other grab comes back,
//he needs to put it back if he got the other one.
//Then go back and think and try to grab the chopsticks again
PartialFunction<Object, BoxedUnit> deniedAChopstick = ReceiveBuilder.
match(Taken.class, t -> {
t.chopstick.tell(new Put(self()), self());
startThinking(Duration.create(10, MILLISECONDS));
}).
match(Busy.class, b ->
startThinking(Duration.create(10, MILLISECONDS))).
build();
//When a hakker is hungry it tries to pick up its chopsticks and eat
//When it picks one up, it goes into wait for the other
//If the hakkers first attempt at grabbing a chopstick fails,
//it starts to wait for the response of the other grab
PartialFunction<Object, BoxedUnit> hungry = ReceiveBuilder.
match(Taken.class, t -> t.chopstick == left,
t -> context().become(waitingFor(right, left))).
match(Taken.class, t -> t.chopstick == right,
t -> context().become(waitingFor(left, right))).
match(Busy.class,
b -> context().become(deniedAChopstick)).
build();
//When a hakker is thinking it can become hungry
//and try to pick up its chopsticks and eat
PartialFunction<Object, BoxedUnit> thinking = ReceiveBuilder.
matchEquals(Eat, m -> {
context().become(hungry);
left.tell(new Take(self()), self());
right.tell(new Take(self()), self());
}).build();
//All hakkers start in a non-eating state
public PartialFunction<Object, BoxedUnit> receive() {
return ReceiveBuilder.matchEquals(Think, m -> {
System.out.println(String.format("%s starts to think", name));
startThinking(Duration.create(5, SECONDS));
}).build();
}
private void startThinking(FiniteDuration duration) {
context().become(thinking);
context().system().scheduler().scheduleOnce(duration, self(), Eat, context().system().dispatcher(), self());
}
}
/*
* Alright, here's our test-harness
*/
public static void main(String[] args) {
ActorSystem system = ActorSystem.create();
//Create 5 chopsticks
ActorRef[] chopsticks = new ActorRef[5];
for (int i = 0; i < 5; i++)
chopsticks[i] = system.actorOf(Props.create(Chopstick.class), "Chopstick" + i);
//Create 5 awesome hakkers and assign them their left and right chopstick
List<String> names = Arrays.asList("Ghosh", "Boner", "Klang", "Krasser", "Manie");
List<ActorRef> hakkers = new ArrayList<>();
int i = 0;
for (String name: names) {
hakkers.add(system.actorOf(Props.create(Hakker.class, name, chopsticks[i], chopsticks[(i + 1) % 5])));
i++;
}
//Signal all hakkers that they should start thinking, and watch the show
hakkers.stream().forEach(hakker -> hakker.tell(Think, ActorRef.noSender()));
}
}

View file

@ -0,0 +1,39 @@
package sample.become;
import akka.actor.ActorRef;
public class Messages {
public static final class Busy {
public final ActorRef chopstick;
public Busy(ActorRef chopstick){
this.chopstick = chopstick;
}
}
public static final class Put {
public final ActorRef hakker;
public Put(ActorRef hakker){
this.hakker = hakker;
}
}
public static final class Take {
public final ActorRef hakker;
public Take(ActorRef hakker){
this.hakker = hakker;
}
}
public static final class Taken {
public final ActorRef chopstick;
public Taken(ActorRef chopstick){
this.chopstick = chopstick;
}
}
private static interface EatMessage {};
public static final Object Eat = new EatMessage() {};
private static interface ThinkMessage {};
public static final Object Think = new ThinkMessage() {};
}

View file

@ -0,0 +1,205 @@
package sample.fsm;
import akka.actor.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import scala.concurrent.duration.Duration;
import scala.concurrent.duration.FiniteDuration;
import static java.util.concurrent.TimeUnit.*;
import static sample.fsm.Messages.*;
// Akka adaptation of
// http://www.dalnefre.com/wp/2010/08/dining-philosophers-in-humus/
public class DiningHakkersOnFsm {
/**
* Some states the chopstick can be in
*/
public static enum CS {
Available,
Taken
}
/**
* Some state container for the chopstick
*/
public static final class TakenBy {
public final ActorRef hakker;
public TakenBy(ActorRef hakker){
this.hakker = hakker;
}
}
/*
* A chopstick is an actor, it can be taken, and put back
*/
public static class Chopstick extends AbstractLoggingFSM<CS, TakenBy> {
{
// A chopstick begins its existence as available and taken by no one
startWith(CS.Available, new TakenBy(context().system().deadLetters()));
// When a chopstick is available, it can be taken by a some hakker
when(CS.Available,
matchEventEquals(Take, (take, data) ->
goTo(CS.Taken).using(new TakenBy(sender())).replying(new Taken(self()))));
// When a chopstick is taken by a hakker
// It will refuse to be taken by other hakkers
// But the owning hakker can put it back
when(CS.Taken,
matchEventEquals(Take, (take, data) ->
stay().replying(new Busy(self()))).
event((event, data) -> (event == Put) && (data.hakker == sender()), (event, data) ->
goTo(CS.Available).using(new TakenBy(context().system().deadLetters()))));
// Initialze the chopstick
initialize();
}
}
/**
* Some fsm hakker states
*/
public static enum HS {
Waiting,
Thinking,
Hungry,
WaitForOtherChopstick,
FirstChopstickDenied,
Eating
}
/**
* Some state container to keep track of which chopsticks we have
*/
public static final class TakenChopsticks {
public final ActorRef left;
public final ActorRef right;
public TakenChopsticks(ActorRef left, ActorRef right) {
this.left = left;
this.right = right;
}
}
/*
* A fsm hakker is an awesome dude or dudette who either thinks about hacking or has to eat ;-)
*/
public static class Hakker extends AbstractLoggingFSM<HS, TakenChopsticks> {
private String name;
private ActorRef left;
private ActorRef right;
public Hakker(String name, ActorRef left, ActorRef right) {
this.name = name;
this.left = left;
this.right = right;
}
{
//All hakkers start waiting
startWith(HS.Waiting, new TakenChopsticks(null, null));
when(HS.Waiting,
matchEventEquals(Think, (think, data) -> {
System.out.println(String.format("%s starts to think", name));
return startThinking(Duration.create(5, SECONDS));
}));
//When a hakker is thinking it can become hungry
//and try to pick up its chopsticks and eat
when(HS.Thinking,
matchEventEquals(StateTimeout(), (event, data) -> {
left.tell(Take, self());
right.tell(Take, self());
return goTo(HS.Hungry);
}));
// When a hakker is hungry it tries to pick up its chopsticks and eat
// When it picks one up, it goes into wait for the other
// If the hakkers first attempt at grabbing a chopstick fails,
// it starts to wait for the response of the other grab
when(HS.Hungry,
matchEvent(Taken.class, (taken, data) -> taken.chopstick == left,
(taken, data) -> goTo(HS.WaitForOtherChopstick).using(new TakenChopsticks(left, null))).
event(Taken.class, (taken, data) -> taken.chopstick == right,
(taken, data) -> goTo(HS.WaitForOtherChopstick).using(new TakenChopsticks(null, right))).
event(Busy.class,
(busy, data) -> goTo(HS.FirstChopstickDenied)));
// When a hakker is waiting for the last chopstick it can either obtain it
// and start eating, or the other chopstick was busy, and the hakker goes
// back to think about how he should obtain his chopsticks :-)
when(HS.WaitForOtherChopstick,
matchEvent(Taken.class,
(taken, data) -> (taken.chopstick == left && data.left == null && data.right != null),
(taken, data) -> startEating(left, right)).
event(Taken.class,
(taken, data) -> (taken.chopstick == right && data.left != null && data.right == null),
(taken, data) -> startEating(left, right)).
event(Busy.class, (busy, data) -> {
if (data.left != null) left.tell(Put, self());
if (data.right != null) right.tell(Put, self());
return startThinking(Duration.create(10, MILLISECONDS));
}));
// When the results of the other grab comes back,
// he needs to put it back if he got the other one.
// Then go back and think and try to grab the chopsticks again
when(HS.FirstChopstickDenied,
matchEvent(Taken.class, (taken, data) -> {
taken.chopstick.tell(Put, self());
return startThinking(Duration.create(10, MILLISECONDS));
}).
event(Busy.class, (busy, data) ->
startThinking(Duration.create(10, MILLISECONDS))));
// When a hakker is eating, he can decide to start to think,
// then he puts down his chopsticks and starts to think
when(HS.Eating,
matchEventEquals(StateTimeout(), (event, data) -> {
left.tell(Put, self());
right.tell(Put, self());
System.out.println(String.format("%s puts down his chopsticks and starts to think", name));
return startThinking(Duration.create(5, SECONDS));
}));
// Initialize the hakker
initialize();
}
private FSM.State<HS, TakenChopsticks> startEating(ActorRef left, ActorRef right) {
System.out.println(String.format("%s has picked up %s and %s and starts to eat",
name, left.path().name(), right.path().name()));
return goTo(HS.Eating).using(new TakenChopsticks(left, right)).forMax(Duration.create(5, SECONDS));
}
private FSM.State<HS, TakenChopsticks> startThinking(FiniteDuration duration) {
return goTo(HS.Thinking).using(new TakenChopsticks(null, null)).forMax(duration);
}
}
/*
* Alright, here's our test-harness
*/
public static void main(String[] args) {
ActorSystem system = ActorSystem.create();
//Create 5 chopsticks
ActorRef[] chopsticks = new ActorRef[5];
for (int i = 0; i < 5; i++)
chopsticks[i] = system.actorOf(Props.create(Chopstick.class), "Chopstick" + i);
//Create 5 awesome hakkers and assign them their left and right chopstick
List<String> names = Arrays.asList("Ghosh", "Boner", "Klang", "Krasser", "Manie");
List<ActorRef> hakkers = new ArrayList<>();
int i = 0;
for (String name: names) {
hakkers.add(system.actorOf(Props.create(Hakker.class, name, chopsticks[i], chopsticks[(i + 1) % 5]), name));
i++;
}
//Signal all hakkers that they should start thinking, and watch the show
hakkers.stream().forEach(hakker -> hakker.tell(Think, ActorRef.noSender()));
}
}

View file

@ -0,0 +1,35 @@
package sample.fsm;
import akka.actor.ActorRef;
public class Messages {
public static final class Busy {
public final ActorRef chopstick;
public Busy(ActorRef chopstick){
this.chopstick = chopstick;
}
}
private static interface PutMessage {};
public static final Object Put = new PutMessage() {
@Override
public String toString() { return "Put"; }
};
private static interface TakeMessage {};
public static final Object Take = new TakeMessage() {
@Override
public String toString() { return "Take"; }
};
public static final class Taken {
public final ActorRef chopstick;
public Taken(ActorRef chopstick){
this.chopstick = chopstick;
}
}
private static interface ThinkMessage {};
public static final Object Think = new ThinkMessage() {};
}