micro/store/memcached_store.go

66 lines
1.1 KiB
Go
Raw Normal View History

2015-01-14 02:31:27 +03:00
package store
import (
"errors"
"os"
mc "github.com/bradfitz/gomemcache/memcache"
)
type MemcacheStore struct {
Client *mc.Client
}
func (m *MemcacheStore) Get(key string) (Item, error) {
kv, err := m.Client.Get(key)
if err != nil && err == mc.ErrCacheMiss {
return nil, errors.New("key not found")
} else if err != nil {
return nil, err
}
if kv == nil {
return nil, errors.New("key not found")
}
return &MemcacheItem{
key: kv.Key,
value: kv.Value,
}, nil
}
func (m *MemcacheStore) Del(key string) error {
return m.Client.Delete(key)
}
func (m *MemcacheStore) Put(item Item) error {
return m.Client.Set(&mc.Item{
Key: item.Key(),
Value: item.Value(),
})
}
func (m *MemcacheStore) NewItem(key string, value []byte) Item {
return &MemcacheItem{
key: key,
value: value,
}
}
func NewMemcacheStore() Store {
server := os.Getenv("MEMCACHED_SERVICE_HOST")
port := os.Getenv("MEMCACHED_SERVICE_PORT")
if len(server) == 0 {
server = "127.0.0.1"
}
if len(port) == 0 {
port = "11211"
}
return &MemcacheStore{
Client: mc.New(server + ":" + port),
}
}