local_freecache.go

package g2cache

import (
	"github.com./coocood/freecache"
	"sync"
)

var (
	DefaultFreeCacheSize = 50 * 1024 * 1024 // 50MB
)

type FreeCache struct {
	storage  *freecache.Cache
	stop     chan struct{}
	stopOnce sync.Once
}

func NewFreeCache() *FreeCache {
	f := &FreeCache{
		storage: freecache.NewCache(DefaultFreeCacheSize),
		stop:    make(chan struct{}, 1),
	}
	return f
}

func (c *FreeCache) Set(key string, e *Entry) error {
	select {
	case <-c.stop:
		return LocalStorageClose
	default:
	}
	s, _ := json.Marshal(e)
	// local storage should set Obsolete time
	obsolete := e.GetObsoleteTTL()
	return c.storage.Set([]byte(key), s, int(obsolete))
}

func (c *FreeCache) Del(key string) error {
	select {
	case <-c.stop:
		return LocalStorageClose
	default:
	}
	c.storage.Del([]byte(key))
	return nil
}

func (c *FreeCache) Get(key string, obj interface{}) (*Entry, bool, error) {
	select {
	case <-c.stop:
		return nil, false, LocalStorageClose
	default:
	}
	b, err := c.storage.Get([]byte(key))
	if err != nil {
		if err == freecache.ErrNotFound {
			return nil, false, nil
		}
		return nil, false, err
	}
	e := new(Entry)
	e.Value = obj // Save the reflection structure of obj
	err = json.Unmarshal(b, e)
	if err != nil {
		return nil, false, err
	}
	return e, true, nil
}

func (c *FreeCache) close() {
	close(c.stop)
	c.storage.Clear()
}

func (c *FreeCache) Close() {
	c.stopOnce.Do(c.close)
}

func (c *FreeCache) ThreadSafe() {}