fsm

README

FSM

FSM provides a lightweight finite state machine for Golang. It runs allows any number of transition checks you'd like the it runs them in parallel. It's tested and benchmarked too.

Install

go get github.com/utilitywarehouse/fsm

Usage

package main

import (
    "log"
    "fmt"
    "github.com/utilitywarehouse/fsm"
)

type Thing struct {
    State fsm.State

    // our machine cache
    machine *fsm.Machine
}

// Add methods to comply with the fsm.Stater interface
func (t *Thing) CurrentState() fsm.State { return t.State }
func (t *Thing) SetState(s fsm.State)    { t.State = s }

// A helpful function that lets us apply arbitrary rulesets to this
// instances state machine without reallocating the machine. While not
// required, it's something I like to have.
func (t *Thing) Apply(r *fsm.Ruleset) *fsm.Machine {
    if t.machine == nil {
        t.machine = &fsm.Machine{Subject: t}
    }

    t.machine.Rules = r
    return t.machine
}

func main() {

    some_thing := Thing{State: "pending"} // Our subject
    fmt.Println(some_thing)

    // Establish some rules for our FSM
    rules := fsm.Ruleset{}
    rules.AddTransition(fsm.T{"pending", "started"})
    rules.AddTransition(fsm.T{"started", "finished"})

    errs := some_thing.Apply(&rules).Transition("started")
    if err != nil {
        log.Fatal(errs)
    }

    fmt.Println(some_thing)
}

Note: FSM makes no effort to determine the default state for any ruleset. That's your job.

The Apply(r *fsm.Ruleset) *fsm.Machine method is absolutely optional. I like having it though. It solves a pretty common problem I usually have when working with permissions - some users aren't allowed to transition between certain states.

Since the rules are applied to the the subject (through the machine) I can have a simple lookup to determine the ruleset that the subject has to follow for a given user. As a result, I rarely need to use any complicated guards but I can if need be. I leave the lookup and the maintaining of independent rulesets as an exercise of the user.

Documentation

Index

• func WithRules(r Ruleset) func(*Machine)
• func WithSubject(s Stater) func(*Machine)
• type ErrInvalidTransition
  • func (e ErrInvalidTransition) Error() string
• type Guard
• type Machine
  • func New(opts ...func(*Machine)) Machine
  • func (m Machine) Transition(goal State) []error
• type Ruleset
  • func CreateRuleset(transitions ...Transition) Ruleset
  • func (r Ruleset) AddRule(t Transition, guards ...Guard)
  • func (r Ruleset) AddTransition(t Transition)
  • func (r Ruleset) IsValidTransition(subject Stater, goal State) []error
• type State
• type Stater
• type T
  • func (t T) Exit() State
  • func (t T) Origin() State
• type Transition

Functions

func WithRules

func WithRules(r Ruleset) func(*Machine)

WithRules is intended to be passed to New to set the Rules

func WithSubject

func WithSubject(s Stater) func(*Machine)

WithSubject is intended to be passed to New to set the Subject

Types

type ErrInvalidTransition

type ErrInvalidTransition struct {
	Transition
}

ErrInvalidTransition is returned when no transition is defined between

func (ErrInvalidTransition) Error

func (e ErrInvalidTransition) Error() string

Error returns ErrInvalidTransition error message

type Guard

type Guard func(subject Stater, goal State) error

Guard provides protection against transitioning to the goal State. Returning an error or nil indicates if the transition is permitted or not.

type Machine

type Machine struct {
	Rules   *Ruleset
	Subject Stater
}

Machine is a pairing of Rules and a Subject. The subject or rules may be changed at any time within the machine's lifecycle.

func New

func New(opts ...func(*Machine)) Machine

New initializes a machine

func (Machine) Transition

func (m Machine) Transition(goal State) []error

Transition attempts to move the Subject to the Goal state.

type Ruleset

type Ruleset map[Transition][]Guard

Ruleset stores the rules for the state machine.

func CreateRuleset

func CreateRuleset(transitions ...Transition) Ruleset

CreateRuleset will establish a ruleset with the provided transitions. This eases initialization when storing within another structure.

func (Ruleset) AddRule

func (r Ruleset) AddRule(t Transition, guards ...Guard)

AddRule adds Guards for the given Transition

func (Ruleset) AddTransition

func (r Ruleset) AddTransition(t Transition)

AddTransition adds a transition with a default rule

func (Ruleset) IsValidTransition

func (r Ruleset) IsValidTransition(subject Stater, goal State) []error

IsValidTransition determines if a transition is allowed. This occurs in parallel. NOTE: Guards are not halted if they are short-circuited for some transition. They may continue running *after* the outcome is determined. Returns a list of errors if the transition is invalid or nil otherwise

type State

type State string

type Stater

type Stater interface {
	CurrentState() State
	SetState(State)
}

Stater can be passed into the FSM. The Stater is reponsible for setting its own default state. Behavior of a Stater without a State is undefined.

type T

type T struct {
	O, E State
}

T implements the Transition interface; it provides a default implementation of a Transition.

func (T) Exit

func (t T) Exit() State

func (T) Origin

func (t T) Origin() State

type Transition

type Transition interface {
	Origin() State
	Exit() State
}

Transition is the change between States