update

cache/bigcache/big_cache.go

@@ -0,0 +1,93 @@
+package bigcache
+
+import (
+	"errors"
+	"time"
+
+	"github.com/allegro/bigcache/v3"
+	"github.com/charlienet/go-mixed/logx"
+)
+
+type BigCacheConfig struct {
+	Shards             int
+	LifeWindow         time.Duration
+	CleanWindow        time.Duration
+	MaxEntriesInWindow int
+	MaxEntrySize       int
+	HardMaxCacheSize   int
+	log                logx.Logger
+}
+
+type bigCacheClient struct {
+	cache *bigcache.BigCache
+}
+
+func NewBigCache(c BigCacheConfig) (*bigCacheClient, error) {
+	config := bigcache.DefaultConfig(time.Minute * 10)
+
+	config.LifeWindow = c.LifeWindow
+	config.LifeWindow = c.LifeWindow
+	config.CleanWindow = c.CleanWindow
+	config.MaxEntriesInWindow = c.MaxEntriesInWindow
+	config.MaxEntrySize = c.MaxEntrySize
+	config.HardMaxCacheSize = c.HardMaxCacheSize
+	config.Logger = c.log
+
+	if c.Shards > 0 {
+		config.Shards = c.Shards
+	}
+
+	bigCache, err := bigcache.NewBigCache(config)
+	if err != nil {
+		return nil, err
+	}
+
+	return &bigCacheClient{
+		cache: bigCache,
+	}, nil
+}
+
+func (c *bigCacheClient) Get(key string) ([]byte, bool) {
+	b, err := c.cache.Get(key)
+	if err == nil {
+		return b, false
+	}
+
+	return b, true
+}
+
+func (c *bigCacheClient) Set(key string, entry []byte, expire time.Duration) error {
+	return c.cache.Set(key, entry)
+}
+
+func (c *bigCacheClient) Delete(keys ...string) error {
+	ks := keys[:]
+	for _, k := range ks {
+		if err := c.cache.Delete(k); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (c *bigCacheClient) Exist(key string) bool {
+	_, err := c.cache.Get(key)
+	if err == nil {
+		return true
+	}
+
+	return !errors.Is(err, bigcache.ErrEntryNotFound)
+}
+
+func (c *bigCacheClient) Clear() {
+
+}
+
+func (c *bigCacheClient) IsNotFound(err error) bool {
+	if err == nil {
+		return true
+	}
+
+	return !errors.Is(err, bigcache.ErrEntryNotFound)
+}

cache/bigcache/big_cache_test.go

@@ -0,0 +1,27 @@
+package bigcache
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestBigCache(t *testing.T) {
+	r := require.New(t)
+
+	c, err := NewBigCache(BigCacheConfig{})
+	r.Nil(err)
+
+	cacheKey := "a"
+	cacheValue := "bbb"
+
+	c.Set(cacheKey, []byte(cacheValue), time.Second*5)
+
+	r.True(c.Exist(cacheKey))
+	r.False(c.Exist("abb"))
+
+	b, ok := c.Get(cacheKey)
+	r.True(ok)
+	r.Equal(cacheValue, string(b))
+}