/
example_bykof.go
115 lines (94 loc) · 2.58 KB
/
example_bykof.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package main
import (
"errors"
"log"
"github.com/bykof/stateful"
)
type (
MyMachine struct {
state stateful.State
amount int
}
AmountParams struct {
Amount int
}
)
func NewMyMachine() MyMachine {
return MyMachine{
state: A,
amount: 0,
}
}
func (mm MyMachine) State() stateful.State {
return mm.state
}
func (mm *MyMachine) SetState(state stateful.State) error {
mm.state = state
return nil
}
const (
A = stateful.DefaultState("A")
B = stateful.DefaultState("B")
)
func (mm *MyMachine) FromAToB(transitionArguments stateful.TransitionArguments) (stateful.State, error) {
amountParams, ok := transitionArguments.(AmountParams)
if !ok {
return nil, errors.New("could not parse AmountParams")
}
mm.amount += amountParams.Amount
return B, nil
}
func (mm *MyMachine) FromBToA(transitionArguments stateful.TransitionArguments) (stateful.State, error) {
amountParams, ok := transitionArguments.(AmountParams)
if !ok {
return nil, errors.New("could not parse AmountParams")
}
mm.amount -= amountParams.Amount
return A, nil
}
func (mm *MyMachine) FromAToNotExistingC(_ stateful.TransitionArguments) (stateful.State, error) {
return stateful.DefaultState("C"), nil
}
func main() {
myMachine := NewMyMachine()
stateMachine := &stateful.StateMachine{
StatefulObject: &myMachine,
}
stateMachine.AddTransition(
// The transition function
myMachine.FromAToB,
// SourceStates
stateful.States{A},
// DestinationStates
stateful.States{B},
)
stateMachine.AddTransition(
myMachine.FromBToA,
stateful.States{B},
stateful.States{A},
)
log.Printf("myMachine[state=%v, amount=%d]\n", myMachine.state, myMachine.amount)
var err error
err = stateMachine.Run(
// The transition function
myMachine.FromAToB,
// The transition params which will be passed to the transition function
stateful.TransitionArguments(AmountParams{Amount: 1}),
)
log.Printf("myMachine[state=%v, amount=%d], FromAtoB=%v\n", myMachine.state, myMachine.amount, err)
err = stateMachine.Run(
myMachine.FromBToA,
stateful.TransitionArguments(AmountParams{Amount: 1}),
)
log.Printf("myMachine[state=%v, amount=%d], FromBToA=%v\n", myMachine.state, myMachine.amount, err)
err = stateMachine.Run(
myMachine.FromBToA,
stateful.TransitionArguments(AmountParams{Amount: 1}),
)
log.Printf("myMachine[state=%v, amount=%d], FromBToA=%v\n", myMachine.state, myMachine.amount, err)
err = stateMachine.Run(
myMachine.FromAToNotExistingC,
stateful.TransitionArguments(nil),
)
log.Printf("myMachine[state=%v, amount=%d], FromAToNotExistingC=%v\n", myMachine.state, myMachine.amount, err)
}