优化每次new一个独立的cache

This commit is contained in:
yun
2024-04-12 16:29:15 +08:00
parent 3ff2662b08
commit 6473f6c61d
+11 -16
View File
@@ -13,12 +13,9 @@ import (
type cache struct { type cache struct {
prefix string prefix string
store sync.Map
} }
var one sync.Once
var c *cache
var store sync.Map
type cacheData struct { type cacheData struct {
key string key string
data interface{} data interface{}
@@ -29,24 +26,22 @@ var ErrorEmpty error = errors.New("empty cache")
func NewCache(prefix string) *cache { func NewCache(prefix string) *cache {
c = &cache{ c := &cache{
prefix: prefix, prefix: prefix,
} }
one.Do(func() {
go func() { go func() {
for { for {
store.Range(func(key, value interface{}) bool { c.store.Range(func(key, value interface{}) bool {
if value.(*cacheData).expire.Before(time.Now()) { if value.(*cacheData).expire.Before(time.Now()) {
store.Delete(key) c.store.Delete(key)
} }
return true return true
}) })
time.Sleep(time.Second * 5) time.Sleep(time.Second * 5)
} }
}() }()
})
return c return c
} }
@@ -57,17 +52,17 @@ func (c *cache) Set(key string, value interface{}, expire time.Duration) {
} }
cd := &cacheData{key, value, time.Now().Add(expire)} cd := &cacheData{key, value, time.Now().Add(expire)}
key = c.prefix + key key = c.prefix + key
store.Store(key, cd) c.store.Store(key, cd)
} }
// 读取缓存 // 读取缓存
func (c *cache) Get(key string) (interface{}, error) { func (c *cache) Get(key string) (interface{}, error) {
key = c.prefix + key key = c.prefix + key
if v, ok := store.Load(key); ok { if v, ok := c.store.Load(key); ok {
cc := v.(*cacheData) cc := v.(*cacheData)
if cc.expire.Before(time.Now()) { if cc.expire.Before(time.Now()) {
store.Delete(key) c.store.Delete(key)
return nil, ErrorEmpty return nil, ErrorEmpty
} }
return cc.data, nil return cc.data, nil
@@ -78,15 +73,15 @@ func (c *cache) Get(key string) (interface{}, error) {
// 删除缓存 // 删除缓存
func (c *cache) Delete(key string) { func (c *cache) Delete(key string) {
key = c.prefix + key key = c.prefix + key
store.Delete(key) c.store.Delete(key)
} }
// 清空缓存 // 清空缓存
func (c *cache) Clear() { func (c *cache) Clear() {
store.Range(func(key, value interface{}) bool { c.store.Range(func(key, value interface{}) bool {
// 根据前缀删除 // 根据前缀删除
if key.(string)[:len(c.prefix)] == c.prefix { if key.(string)[:len(c.prefix)] == c.prefix {
store.Delete(key) c.store.Delete(key)
} }
return true return true
}) })