59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
|
package rest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/md5"
|
||
|
|
"github.com/bradfitz/gomemcache/memcache"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Memcached connection settings
|
||
|
|
type MC struct {
|
||
|
|
Enabled bool `json:"enabled"`
|
||
|
|
Hosts []string `json:"hosts"`
|
||
|
|
Prefix string `json:"prefix"`
|
||
|
|
Client *memcache.Client
|
||
|
|
}
|
||
|
|
|
||
|
|
// InitConnection to memcached cluster
|
||
|
|
func (mc MC) InitConnection() *memcache.Client {
|
||
|
|
client := memcache.New(mc.Hosts...)
|
||
|
|
return client
|
||
|
|
}
|
||
|
|
|
||
|
|
// StoreToCache save data to memecache
|
||
|
|
func (mc MC) StoreToCache(key string, data []byte) error {
|
||
|
|
prefixed_key := strings.Join([]string{mc.Prefix, key}, "_")
|
||
|
|
|
||
|
|
var item memcache.Item
|
||
|
|
// item.Key = hashKey(prefixed_key)
|
||
|
|
item.Key = prefixed_key
|
||
|
|
item.Value = data
|
||
|
|
|
||
|
|
// store
|
||
|
|
err := mc.Client.Set(&item)
|
||
|
|
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// (mc MC) GetFromCache ...
|
||
|
|
func (mc MC) GetFromCache(key string) ([]byte, error) {
|
||
|
|
var data *memcache.Item
|
||
|
|
prefixed_key := strings.Join([]string{mc.Prefix, key}, "_")
|
||
|
|
|
||
|
|
// data, err := mc.Client.Get(hashKey(prefixed_key))
|
||
|
|
data, err := mc.Client.Get(prefixed_key)
|
||
|
|
if err != nil {
|
||
|
|
return []byte(""), err
|
||
|
|
}
|
||
|
|
|
||
|
|
return data.Value, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// hashKey return md5sum of prefixed_key
|
||
|
|
func hashKey(key string) string {
|
||
|
|
h := md5.New()
|
||
|
|
sum := h.Sum([]byte(key))
|
||
|
|
|
||
|
|
return string(sum)
|
||
|
|
}
|