掌握Go语言状态机编程，玩转状态转换与动作！

2023-12-14 19:32:48

引言

在计算机科学中，状态机（FSM）是一种数学模型，用于具有有限数量状态的系统。状态机由一组状态、一组输入和一组输出组成。当状态机收到输入时，它会根据当前状态和输入确定下一个状态和输出。

状态机在现实世界中有许多应用。例如，状态机可以用于模拟交通信号灯、电梯、洗衣机和其他具有有限数量状态的系统。

Go语言中的状态机

在Go语言中，可以使用switch语句或map结构来实现状态机。switch语句是一种条件语句，可以根据变量的值执行不同的代码块。map结构是一种数据结构，可以将键值对存储起来。

type StateMachine struct {
	currentState string
	transitions map[string]func(string) string
}

func NewStateMachine() *StateMachine {
	return &StateMachine{
		currentState: "start",
		transitions: map[string]func(string) string{
			"start": func(input string) string {
				if input == "a" {
					return "state1"
				} else {
					return "start"
				}
			},
			"state1": func(input string) string {
				if input == "b" {
					return "state2"
				} else {
					return "state1"
				}
			},
			"state2": func(input string) string {
				if input == "c" {
					return "end"
				} else {
					return "state2"
				}
			},
		},
	}
}

func (sm *StateMachine) ProcessInput(input string) {
	nextState := sm.transitions[sm.currentState](input)
	sm.currentState = nextState
}

上面的代码实现了一个简单的状态机，该状态机有三个状态：start、state1和state2。当状态机收到输入a时，它会从start状态转换到state1状态。当状态机收到输入b时，它会从state1状态转换到state2状态。当状态机收到输入c时，它会从state2状态转换到end状态。