57 lines
980 B
Go
57 lines
980 B
Go
|
// Package memory is an in memory cache
|
||
|
|
package memory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/micro/go-micro/v2/cache"
|
||
|
|
"github.com/micro/go-micro/v2/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
type memoryCache struct {
|
||
|
|
// TODO: use a decent caching library
|
||
|
|
sync.RWMutex
|
||
|
|
values map[string]interface{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *memoryCache) Init(opts ...cache.Option) error {
|
||
|
|
// TODO: implement
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *memoryCache) Get(key string) (interface{}, error) {
|
||
|
|
m.RLock()
|
||
|
|
defer m.RUnlock()
|
||
|
|
|
||
|
|
v, ok := m.values[key]
|
||
|
|
if !ok {
|
||
|
|
return nil, errors.NotFound("go.micro.cache", key+" not found")
|
||
|
|
}
|
||
|
|
|
||
|
|
return v, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *memoryCache) Set(key string, val interface{}) error {
|
||
|
|
m.Lock()
|
||
|
|
m.values[key] = val
|
||
|
|
m.Unlock()
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *memoryCache) Delete(key string) error {
|
||
|
|
m.Lock()
|
||
|
|
delete(m.values, key)
|
||
|
|
m.Unlock()
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *memoryCache) String() string {
|
||
|
|
return "memory"
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewCache(opts ...cache.Option) cache.Cache {
|
||
|
|
return &memoryCache{
|
||
|
|
values: make(map[string]interface{}),
|
||
|
|
}
|
||
|
|
}
|