tgdrive/teldrive
1
1
Fork 0
mirror of https://github.com/tgdrive/teldrive.git synced 2025-01-09 00:29:57 +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 ( import (
"context" "context"
"sync"
"time" "time"
"github.com/coocood/freecache" "github.com/coocood/freecache"
@ -19,6 +20,7 @@ type Cacher interface {
type MemoryCache struct { type MemoryCache struct {
cache *freecache.Cache cache *freecache.Cache
prefix string prefix string
mu sync.RWMutex
} }
func NewCache(ctx context.Context, conf *config.Config) Cacher { 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 { func (m *MemoryCache) Get(key string, value interface{}) error {
m.mu.RLock()
defer m.mu.RUnlock()
key = m.prefix + key key = m.prefix + key
data, err := m.cache.Get([]byte(key)) data, err := m.cache.Get([]byte(key))
if err != nil { 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 { func (m *MemoryCache) Set(key string, value interface{}, expiration time.Duration) error {
m.mu.RLock()
defer m.mu.RUnlock()
key = m.prefix + key key = m.prefix + key
data, err := msgpack.Marshal(value) data, err := msgpack.Marshal(value)
if err != nil { 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 { func (m *MemoryCache) Delete(keys ...string) error {
m.mu.RLock()
defer m.mu.RUnlock()
for _, key := range keys { for _, key := range keys {
m.cache.Del([]byte(m.prefix + key)) m.cache.Del([]byte(m.prefix + key))
} }
@ -71,6 +79,7 @@ type RedisCache struct {
client *redis.Client client *redis.Client
ctx context.Context ctx context.Context
prefix string prefix string
mu sync.RWMutex
} }
func NewRedisCache(ctx context.Context, client *redis.Client) *RedisCache { 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 { func (r *RedisCache) Get(key string, value interface{}) error {
r.mu.RLock()
defer r.mu.RUnlock()
key = r.prefix + key key = r.prefix + key
data, err := r.client.Get(r.ctx, key).Bytes() data, err := r.client.Get(r.ctx, key).Bytes()
if err != nil { 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 { func (r *RedisCache) Set(key string, value interface{}, expiration time.Duration) error {
r.mu.RLock()
defer r.mu.RUnlock()
key = r.prefix + key key = r.prefix + key
data, err := msgpack.Marshal(value) data, err := msgpack.Marshal(value)
if err != nil { 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 { func (r *RedisCache) Delete(keys ...string) error {
r.mu.RLock()
defer r.mu.RUnlock()
for i := range keys { for i := range keys {
keys[i] = r.prefix + keys[i] keys[i] = r.prefix + keys[i]
} }