memory_store.go

package scenario

import (
	"strings"
)

func NewMemoryStore() *MemoryStore {
	status := make(map[string]string)
	return &MemoryStore{status: status}
}

type MemoryStore struct {
	status map[string]string
	paused bool
}

func (sm *MemoryStore) Reset(name string) bool {
	if _, f := sm.status[strings.ToLower(name)]; f {
		sm.status[strings.ToLower(name)] = "not_started"
		return true
	}
	return false
}

func (sm *MemoryStore) ResetAll() {
	sm.status = make(map[string]string)
	sm.paused = false
}

func (sm *MemoryStore) SetState(name, status string) {
	if sm.paused {
		return
	}
	sm.status[strings.ToLower(name)] = strings.ToLower(status)
}

func (sm *MemoryStore) GetState(name string) string {
	if v, f := sm.status[strings.ToLower(name)]; f {
		return v
	}
	return "not_started"
}

func (sm *MemoryStore) GetPaused() bool {
	return sm.paused
}

func (sm *MemoryStore) SetPaused(newstate bool) {
	sm.paused = newstate
}