cache

cache/free_cache.go

@@ -0,0 +1,55 @@
+package cache
+
+import (
+	"errors"
+	"time"
+
+	"github.com/coocood/freecache"
+)
+
+const defaultSize = 10 * 1024 * 1024 // 10M
+
+var _ MemCache = &freeCache{}
+
+type freeCache struct {
+	cache *freecache.Cache
+}
+
+func NewFreeCache(size int) *freeCache {
+	if size < defaultSize {
+		size = defaultSize
+	}
+
+	// debug.SetGCPercent(20)
+
+	c := freecache.NewCache(size)
+	return &freeCache{
+		cache: c,
+	}
+}
+
+func (c *freeCache) Get(key string) ([]byte, error) {
+	return c.cache.Get([]byte(key))
+}
+
+func (c *freeCache) Set(key string, value []byte, d time.Duration) error {
+	s := int(d.Seconds())
+	return c.cache.Set([]byte(key), value, s)
+}
+
+func (c *freeCache) Delete(key string) error {
+	affected := c.cache.Del([]byte(key))
+	if !affected {
+		return errors.New("不存在")
+	}
+
+	return nil
+}
+
+func (c *freeCache) Exist(key string) error {
+	return nil
+}
+
+func (c *freeCache) IsNotFound(err error) bool {
+	return errors.Is(err, freecache.ErrNotFound)
+}