fix: concurrent lru

Author: lz1998

service/cache/cache.go

@@ -1,20 +1,45 @@
 package cache
 
 import (
+	"sync"
+
 	"github.com/golang/groupcache/lru"
 )
 
+type LruCache struct {
+	*lru.Cache
+	mutex sync.Mutex
+}
+
+func NewLruCache(maxEntries int) *LruCache {
+	return &LruCache{
+		Cache: lru.New(maxEntries),
+	}
+}
+
+func (l *LruCache) Add(key lru.Key, value interface{}) {
+	l.mutex.Lock()
+	l.Cache.Add(key, value)
+	l.mutex.Unlock()
+}
+func (l *LruCache) Get(key lru.Key) (value interface{}, ok bool) {
+	l.mutex.Lock()
+	value, ok = l.Cache.Get(key)
+	l.mutex.Unlock()
+	return
+}
+
 // int:PrivateMessage
-var PrivateMessageLru = lru.New(512)
+var PrivateMessageLru = NewLruCache(512)
 
 // int:GroupMessage
-var GroupMessageLru = lru.New(2048)
+var GroupMessageLru = NewLruCache(2048)
 
 // string:
-var FriendRequestLru = lru.New(128)
+var FriendRequestLru = NewLruCache(128)
 
 // string:
-var GroupRequestLru = lru.New(128)
+var GroupRequestLru = NewLruCache(128)
 
 // string:
-var GroupInvitedRequestLru = lru.New(16)
+var GroupInvitedRequestLru = NewLruCache(16)

service/cache/cache_test.go

@@ -0,0 +1,15 @@
+package cache
+
+import "testing"
+
+func TestLruCache_Add(t *testing.T) {
+	lruCache := NewLruCache(3)
+	lruCache.Add(10, 10)
+	v, ok := lruCache.Get(10)
+	t.Logf("1: %+v %+v", v, ok)
+	lruCache.Add(11, 10)
+	lruCache.Add(13, 10)
+	lruCache.Add(14, 10)
+	v, ok = lruCache.Get(10)
+	t.Logf("2: %+v %+v", v, ok)
+}