refactor: complete core features

common/cache.go

@@ -0,0 +1,63 @@
+package common
+
+import (
+	"bytes"
+	"encoding/gob"
+	"sync"
+
+	"github.com/coocood/freecache"
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/types"
+)
+
+type CommonCache struct {
+	cache *freecache.Cache
+	mu    sync.RWMutex
+}
+
+var Cache *CommonCache
+
+func initCache() {
+	gob.Register(types.File{})
+	gob.Register(tg.InputDocumentFileLocation{})
+	Cache = &CommonCache{cache: freecache.NewCache(10 * 1024 * 1024)}
+}
+
+func GetCache() *CommonCache {
+	return Cache
+}
+
+func (c *CommonCache) Get(key string, value *types.File) error {
+	c.mu.RLock()
+	defer c.mu.RUnlock()
+	data, err := Cache.cache.Get([]byte(key))
+	if err != nil {
+		return err
+	}
+	dec := gob.NewDecoder(bytes.NewReader(data))
+	err = dec.Decode(&value)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (c *CommonCache) Set(key string, value *types.File, expireSeconds int) error {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	var buf bytes.Buffer
+	enc := gob.NewEncoder(&buf)
+	err := enc.Encode(value)
+	if err != nil {
+		return err
+	}
+	Cache.cache.Set([]byte(key), buf.Bytes(), expireSeconds)
+	return nil
+}
+
+func (c *CommonCache) Delete(key string) error {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	Cache.cache.Del([]byte(key))
+	return nil
+}