From d55a1705708912240e177936eef4a881c7282982 Mon Sep 17 00:00:00 2001 From: yuan Date: Sat, 20 May 2023 15:19:09 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20memory=20=E5=B9=B6=E5=8F=91=E8=AF=BB?= =?UTF-8?q?=E5=86=99=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cache/memory.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/cache/memory.go b/cache/memory.go index 135d9d9..948fe17 100644 --- a/cache/memory.go +++ b/cache/memory.go @@ -26,25 +26,31 @@ func NewMemory() *Memory { // Get return cached value func (mem *Memory) Get(key string) interface{} { + mem.Lock() if ret, ok := mem.data[key]; ok { + mem.Unlock() if ret.Expired.Before(time.Now()) { mem.deleteKey(key) return nil } return ret.Data } + mem.Unlock() return nil } // IsExist check value exists in memcache. func (mem *Memory) IsExist(key string) bool { + mem.Lock() if ret, ok := mem.data[key]; ok { + mem.Unlock() if ret.Expired.Before(time.Now()) { mem.deleteKey(key) return false } return true } + mem.Unlock() return false }