akka.japi.pf.FSMTransitionHandlerBuilder Maven / Gradle / Ivy
/**
* Copyright (C) 2009-2014 Typesafe Inc.
*/
package akka.japi.pf;
import scala.PartialFunction;
import scala.runtime.BoxedUnit;
import scala.Tuple2;
/**
* Builder used to create a partial function for {@link akka.actor.FSM#onTransition}.
*
* @param the state type
*
* This is an EXPERIMENTAL feature and is subject to change until it has received more real world testing.
*/
public class FSMTransitionHandlerBuilder {
private UnitPFBuilder> builder =
new UnitPFBuilder>();
/**
* Add a case statement that matches on a from state and a to state.
*
* @param fromState the from state to match on, or null for any
* @param toState the to state to match on, or null for any
* @param apply an action to apply when the states match
* @return the builder with the case statement added
*/
public FSMTransitionHandlerBuilder state(final S fromState,
final S toState,
final FI.UnitApplyVoid apply) {
builder.match(Tuple2.class,
new FI.TypedPredicate() {
@Override
public boolean defined(Tuple2 t) {
return (fromState == null || fromState.equals(t._1()))
&& (toState == null || toState.equals(t._2()));
}
},
new FI.UnitApply() {
@Override
public void apply(Tuple2 t) throws Exception {
apply.apply();
}
}
);
return this;
}
/**
* Add a case statement that matches on a from state and a to state.
*
* @param fromState the from state to match on, or null for any
* @param toState the to state to match on, or null for any
* @param apply an action to apply when the states match
* @return the builder with the case statement added
*/
public FSMTransitionHandlerBuilder state(final S fromState,
final S toState,
final FI.UnitApply2 apply) {
builder.match(Tuple2.class,
new FI.TypedPredicate() {
@Override
public boolean defined(Tuple2 t) {
return (fromState == null || fromState.equals(t._1()))
&& (toState == null || toState.equals(t._2()));
}
},
new FI.UnitApply() {
@Override
public void apply(Tuple2 t) throws Exception {
@SuppressWarnings("unchecked")
S sf = (S) t._1();
@SuppressWarnings("unchecked")
S st = (S) t._2();
apply.apply(sf, st);
}
}
);
return this;
}
/**
* Build a {@link scala.PartialFunction} from this builder.
* After this call the builder will be reset.
*
* @return a PartialFunction for this builder.
*/
public PartialFunction, BoxedUnit> build() {
return builder.build();
}
}