micro/store/memory/memory.go

package memory

import (
	"errors"
	"sync"

	"github.com/myodc/go-micro/store"
)

type mstore struct {
	sync.RWMutex
	store map[string]*store.Item
}

func (m *mstore) Get(key string) (*store.Item, error) {
	m.RLock()
	v, ok := m.store[key]
	m.RUnlock()
	if !ok {
		return nil, errors.New("key not found")
	}
	return v, nil
}

func (m *mstore) Del(key string) error {
	m.Lock()
	delete(m.store, key)
	m.Unlock()
	return nil
}

func (m *mstore) Put(item *store.Item) error {
	m.Lock()
	m.store[item.Key] = item
	m.Unlock()
	return nil
}

func NewStore(addrs []string, opt ...store.Option) store.Store {
	return &mstore{
		store: make(map[string]*store.Item),
	}
}