fsm_manager.go

package msm

import (
	"reflect"

	"go.uber.org/zap"
	"gorm.io/gorm"
)

type FSMManager struct {
	FSMs      map[string]*FSM
	Callbacks []ICallback
	Plugins   map[string]IPlugin
	DB        *gorm.DB
	logger    *zap.Logger
	respChCap int
}

type FSMManagerSettings func(mgr *FSMManager)

func NewFSMManager(db *gorm.DB, logger *zap.Logger, settings ...FSMManagerSettings) *FSMManager {
	mgr := &FSMManager{
		FSMs:      make(map[string]*FSM),
		Callbacks: make([]ICallback, 0),
		Plugins:   make(map[string]IPlugin),
		DB:        db,
		logger:    logger,
		respChCap: 32,
	}
	for _, setting := range settings {
		setting(mgr)
	}
	return mgr
}

func (m *FSMManager) Start() {
	for _, fsm := range m.FSMs {
		fsm.Start()
	}
}

func (m *FSMManager) Stop() {
	for _, fsm := range m.FSMs {
		fsm.Stop()
	}
}

func (m *FSMManager) Sync() {
	for _, fsm := range m.FSMs {
		fsm.Sync()
	}
}

func (m *FSMManager) NewFSM(name string, storageFactory StorageFactory) *FSM {
	storageType := reflect.TypeOf(storageFactory).Out(0)
	fsm := &FSM{
		Manager:        m,
		Name:           name,
		Table:          nil,
		Terminals:      map[State]bool{FinishState: true},
		cancel:         make(chan struct{}),
		response:       make(chan Response, m.respChCap),
		onTerminated:   make(map[FsmID][]TerminationCB),
		storageFactory: storageFactory,
		storageType:    storageType,
		logger:         m.logger.Named(name),
		DB:             m.DB,
	}
	m.FSMs[name] = fsm
	return fsm
}

func (m *FSMManager) CreateFSMInstance(fsmName string) FsmID {
	fsm, ok := m.FSMs[fsmName]
	if !ok {
		return FsmID{}
	}
	return fsm.CreateFSM()
}

func (m *FSMManager) Process(id FsmID, src string, event Event) {
	fsm, ok := m.FSMs[id.Fsm]
	if !ok {
		return
	}
	fsm.Process(id, src, event)
}

func (m *FSMManager) IsTerminated(id FsmID) bool {
	fsm, ok := m.FSMs[id.Fsm]
	if !ok {
		return false
	}
	return fsm.IsTerminated(id)
}

func (m *FSMManager) AddPlugin(plugin IPlugin) {
	m.Plugins[plugin.Name()] = plugin
}

func (m *FSMManager) AddCallback(cb ICallback) {
	m.Callbacks = append(m.Callbacks, cb)
}

func (m *FSMManager) AutoMigrate() error {
	for _, fsm := range m.FSMs {
		err := fsm.AutoMigrate()
		if err != nil {
			return err
		}
	}
	for _, plugin := range m.Plugins {
		err := plugin.AutoMigrate()
		if err != nil {
			return err
		}
	}
	for _, cb := range m.Callbacks {
		err := cb.AutoMigrate()
		if err != nil {
			return err
		}
	}
	return nil
}