tgdrive/teldrive
1
1
Fork 0
mirror of https://github.com/tgdrive/teldrive.git synced 2025-01-08 08:10:05 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: Add read-write locks for cache operations

Browse source
This commit is contained in:
divyam234 2024-07-26 21:22:33 +05:30
parent 9ce54b0a50
commit 4db65dd942
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
internal/cache/cache.go vendored
View file

@ -2,6 +2,7 @@ package cache
import (
"context"
"sync"
"time"
"github.com/coocood/freecache"
@ -19,6 +20,7 @@ type Cacher interface {
type MemoryCache struct {
cache *freecache.Cache
prefix string
mu sync.RWMutex
}
func NewCache(ctx context.Context, conf *config.Config) Cacher {
@ -43,6 +45,8 @@ func NewMemoryCache(size int) *MemoryCache {
}
func (m *MemoryCache) Get(key string, value interface{}) error {
m.mu.RLock()
defer m.mu.RUnlock()
key = m.prefix + key
data, err := m.cache.Get([]byte(key))
if err != nil {
@ -52,6 +56,8 @@ func (m *MemoryCache) Get(key string, value interface{}) error {
}
func (m *MemoryCache) Set(key string, value interface{}, expiration time.Duration) error {
m.mu.RLock()
defer m.mu.RUnlock()
key = m.prefix + key
data, err := msgpack.Marshal(value)
if err != nil {
@ -61,6 +67,8 @@ func (m *MemoryCache) Set(key string, value interface{}, expiration time.Duratio
}
func (m *MemoryCache) Delete(keys ...string) error {
m.mu.RLock()
defer m.mu.RUnlock()
for _, key := range keys {
m.cache.Del([]byte(m.prefix + key))
}
@ -71,6 +79,7 @@ type RedisCache struct {
client *redis.Client
ctx context.Context
prefix string
mu sync.RWMutex
}
func NewRedisCache(ctx context.Context, client *redis.Client) *RedisCache {
@ -82,6 +91,8 @@ func NewRedisCache(ctx context.Context, client *redis.Client) *RedisCache {
}
func (r *RedisCache) Get(key string, value interface{}) error {
r.mu.RLock()
defer r.mu.RUnlock()
key = r.prefix + key
data, err := r.client.Get(r.ctx, key).Bytes()
if err != nil {
@ -91,6 +102,8 @@ func (r *RedisCache) Get(key string, value interface{}) error {
}
func (r *RedisCache) Set(key string, value interface{}, expiration time.Duration) error {
r.mu.RLock()
defer r.mu.RUnlock()
key = r.prefix + key
data, err := msgpack.Marshal(value)
if err != nil {
@ -100,6 +113,8 @@ func (r *RedisCache) Set(key string, value interface{}, expiration time.Duration
}
func (r *RedisCache) Delete(keys ...string) error {
r.mu.RLock()
defer r.mu.RUnlock()
for i := range keys {
keys[i] = r.prefix + keys[i]
}