41 lines
732 B
Go
41 lines
732 B
Go
package memory
|
|
|
|
import (
|
|
"context"
|
|
"github.com/abelith/gomax/state"
|
|
"sync"
|
|
)
|
|
|
|
type FSMStorage struct {
|
|
m sync.Map
|
|
}
|
|
|
|
type EventIsolation struct{}
|
|
|
|
func NewMemoryFSMStorage() *FSMStorage {
|
|
return &FSMStorage{}
|
|
}
|
|
|
|
func (f *FSMStorage) Get(_ context.Context, userID int) (state.Snapshot, error) {
|
|
st, ok := f.m.Load(userID)
|
|
if !ok {
|
|
return state.Snapshot{}, state.ErrStateNotFound
|
|
}
|
|
|
|
return st.(state.Snapshot), nil
|
|
}
|
|
|
|
func (f *FSMStorage) Put(_ context.Context, snapshot state.Snapshot) error {
|
|
f.m.Store(snapshot.ID, snapshot)
|
|
return nil
|
|
}
|
|
|
|
func (f *FSMStorage) Delete(_ context.Context, userID int) error {
|
|
f.m.Delete(userID)
|
|
return nil
|
|
}
|
|
|
|
func NewMemoryEventIsolation() *EventIsolation {
|
|
return &EventIsolation{}
|
|
}
|