tiny/spring-state-machine-renderer
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

choice for plantuml as rectangles

Browse Source
This commit is contained in:
Kamil Patryk Kozakowski
2025-07-15 03:52:13 +02:00
parent 71ccb6ada8
commit d6449a8de1
2 changed files with 111 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/click/kamil/springstatemachineexporter/ast/out/PlantUml.java
View File

@@ -29,6 +29,7 @@ public class PlantUml implements StateMachineExporter {
Set<String> startStates,
Set<String> endStates,
boolean useLambdaGuards) {
boolean includeChoiceStates = false;
StringBuilder sb = new StringBuilder();
sb.append("@startuml\n");
sb.append("skinparam state {\n");
@@ -53,9 +54,14 @@ public class PlantUml implements StateMachineExporter {
// 3. Declare choice pseudostates & connect from original state
for (String state : statesWithChoice) {
String choiceState = state + "_choice";
sb.append("state ").append(choiceState).append(" <<choice>>\n");
sb.append(state).append(" --> ").append(choiceState).append("\n");
String choiceState = state;
if (includeChoiceStates) {
choiceState = state + "_choice";
sb.append("state ").append(choiceState).append(" <<choice>>\n");
sb.append(state).append(" --> ").append(choiceState).append("\n");
} else {
sb.append("state ").append(choiceState).append(" <<choice>>\n");
}
}
sb.append("\n");
@@ -91,9 +97,14 @@ public class PlantUml implements StateMachineExporter {
for (String rawTarget : targets) {
String target = simplify(rawTarget);
String transitionSource = "withChoice".equals(t.getType()) && statesWithChoice.contains(source)
? source + "_choice"
: source;
String transitionSource;
if (includeChoiceStates) {
transitionSource = "withChoice".equals(t.getType()) && statesWithChoice.contains(source)
? source + "_choice"
: source;
} else {
transitionSource = source;
}
String label = buildLabel(t, useLambdaGuards);
String color = getColor(t, source, choiceStateColorMap);

94
src/main/java/click/kamil/springstatemachineexporter/statemachine/KamilEnumStateMachineConfiguration.java Normal file
View File

@@ -0,0 +1,94 @@
package click.kamil.springstatemachineexporter.statemachine;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Configuration;
import org.springframework.statemachine.StateContext;
import org.springframework.statemachine.action.Action;
import org.springframework.statemachine.config.EnableStateMachineFactory;
import org.springframework.statemachine.config.StateMachineConfigurerAdapter;
import org.springframework.statemachine.config.builders.StateMachineConfigurationConfigurer;
import org.springframework.statemachine.config.builders.StateMachineStateConfigurer;
import org.springframework.statemachine.config.builders.StateMachineTransitionConfigurer;
import org.springframework.statemachine.guard.Guard;
import org.springframework.statemachine.listener.StateMachineListenerAdapter;
import org.springframework.statemachine.state.State;
@Configuration
@EnableStateMachineFactory(name = "simpleEnumStateMachineFactory")
@Slf4j
class KamilEnumStateMachineConfiguration extends StateMachineConfigurerAdapter<OrderStates, OrderEvents> {
@Override
public void configure(StateMachineTransitionConfigurer<OrderStates, OrderEvents> transitions) throws Exception {
Guard<OrderStates, OrderEvents> guard1 = null;
Action<OrderStates, OrderEvents> action2 = null;
transitions
.withExternal().source(OrderStates.SUBMITTED).target(OrderStates.PAID).event(OrderEvents.PAY)
.and()
.withExternal().source(OrderStates.PAID).target(OrderStates.FULFILLED).event(OrderEvents.FULFILL)
.and()
.withExternal().source(OrderStates.SUBMITTED).target(OrderStates.CANCELED).event(OrderEvents.CANCEL).guard(new Guard<OrderStates, OrderEvents>() {
@Override
public boolean evaluate(StateContext<OrderStates, OrderEvents> context) {
return false;
}
})
.and()
.withExternal().source(OrderStates.PAID).target(OrderStates.CANCELED).event(OrderEvents.CANCEL)
.and()
.withExternal().source(OrderStates.SUBMITTED).event(OrderEvents.ABCD)
// it needs to have target specified to get added to .transitions collections
.and()
.withExternal().source(OrderStates.SUBMITTED).target(OrderStates.PAID1).event(OrderEvents.ABCD)
//
.and()
.withChoice().source(OrderStates.PAID1)
.first(OrderStates.PAID2, new Guard<OrderStates, OrderEvents>() {
@Override
public boolean evaluate(StateContext<OrderStates, OrderEvents> context) {
return true;
}
})
.then(OrderStates.PAID3, guard1)
.then(OrderStates.HAPPEN, new Guard<OrderStates, OrderEvents>() {
@Override
public boolean evaluate(StateContext<OrderStates, OrderEvents> context) {
return false;
}
})
.last(OrderStates.CANCELED)
.and().withExternal().source(OrderStates.PAID2).target(OrderStates.CANCELED)
.and().withExternal().source(OrderStates.PAID3).target(OrderStates.CANCELED)
// .and().withLocal().source(OrderStates.SUBMITTED).target(OrderStates.CANCELED)
.and().withExternal().source(OrderStates.PAID2).event(OrderEvents.ABCD).action(new Action<OrderStates, OrderEvents>() {
@Override
public void execute(StateContext<OrderStates, OrderEvents> context) {
;
}
}).action(action2);
}
@Override
public void configure(StateMachineStateConfigurer<OrderStates, OrderEvents> states) throws Exception {
states.withStates()
.initial(OrderStates.SUBMITTED)
.state(OrderStates.PAID)
.state(OrderStates.PAID2)
.state(OrderStates.PAID3)
.end(OrderStates.FULFILLED)
.end(OrderStates.CANCELED);
}
@Override
public void configure(StateMachineConfigurationConfigurer<OrderStates, OrderEvents> config) throws Exception {
StateMachineListenerAdapter<OrderStates, OrderEvents> listenerAdapter = new StateMachineListenerAdapter<>() {
@Override
public void stateChanged(State<OrderStates, OrderEvents> from, State<OrderStates, OrderEvents> to) {
log.info("state changed from {} to {}", from, to);
}
};
config.withConfiguration()
.autoStartup(false)
.listener(listenerAdapter);
}
}