p93542168
/
wheat-cache
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

feat(lru): update sizebyte

This commit is contained in:
bandl 2021-10-04 22:37:04 +08:00
parent 8bcac6f739
commit 1de8e1b141
1 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
pkg/lru/lru.go
View File

@ -23,37 +23,37 @@ type SingleCache struct {
li *list.List li *list.List
lruMap map[string]*list.Element lruMap map[string]*list.Element
lruDriver event.DriverInterface lruDriver event.DriverInterface
lruConsumer event.ConsumerInterface lruConsumer event.ConsumerInterface
lruCleanProduce event.ProduceInterface // 发送清理事件 lruCleanProduce event.ProduceInterface // 发送清理事件
} }
// UpdateLruSize 更新现在的长度 // UpdateLruSize 更新现在的长度
func (lru *SingleCache) UpdateLruSize(length int64) { func (lru *SingleCache) UpdateLruSize(length structure.UpdateLength) {
atomic.AddInt64(&lru.nowSize, length) atomic.AddInt64(&lru.nowSize, int64(length))
} }
func cacheInit() (int64, int64, event.DriverInterface) { func cacheInit() (int64, int64, event.DriverInterface) {
maxSize := viper.GetString("lruCache.maxSize") maxSize := viper.GetString("lruCache.maxSize")
retMaxSize, maxErr:= util.ParseSizeToBit(maxSize) retMaxSize, maxErr := util.ParseSizeToBit(maxSize)
if maxErr != nil{ if maxErr != nil {
return 0, 0, nil return 0, 0, nil
} }
if retMaxSize == 0{ if retMaxSize == 0 {
retMaxSize = lruMaxSize retMaxSize = lruMaxSize
} }
clearSize := viper.GetString("lruCache.clearSize") clearSize := viper.GetString("lruCache.clearSize")
retClearSize, clearErr := util.ParseSizeToBit(clearSize) retClearSize, clearErr := util.ParseSizeToBit(clearSize)
if clearErr != nil{ if clearErr != nil {
return 0, 0, nil return 0, 0, nil
} }
if retClearSize == 0{ if retClearSize == 0 {
retClearSize = lruClearSize retClearSize = lruClearSize
} }
maxDriver := viper.GetInt("lruCache.eventDriverSize") maxDriver := viper.GetInt("lruCache.eventDriverSize")
if maxDriver == 0{ if maxDriver == 0 {
maxDriver = lruEventDriver maxDriver = lruEventDriver
} }
lruDriver := event.NewDriver(maxDriver) lruDriver := event.NewDriver(maxDriver)
@ -66,14 +66,14 @@ func NewLRUCache() *SingleCache {
lruCacheOnce.Do(func() { lruCacheOnce.Do(func() {
_, _, lruDriver := cacheInit() _, _, lruDriver := cacheInit()
lru := &SingleCache{ lru := &SingleCache{
maxsize: maxSize, maxsize: maxSize,
clearSize: clearSize, clearSize: clearSize,
nowSize: 0, nowSize: 0,
li: list.New(), li: list.New(),
lruMap: make(map[string]*list.Element), lruMap: make(map[string]*list.Element),
lruDriver: lruDriver, lruDriver: lruDriver,
lruConsumer: event.NewConsumer(lruDrivers), lruConsumer: event.NewConsumer(lruDrivers),
lruCleanProduce: event.NewProduce(lruDrivers), lruCleanProduce: event.NewProduce(lruDrivers),
} }
lruCache = lru lruCache = lru
go lru.lruSingleWork() go lru.lruSingleWork()
@ -101,7 +101,7 @@ func (lru *SingleCache) Add(key string, val structure.KeyBaseInterface) {
valEl := lru.li.PushFront(keyBaseVal) valEl := lru.li.PushFront(keyBaseVal)
lru.lruMap[key] = valEl lru.lruMap[key] = valEl
//增加大小 //增加大小
lru.UpdateLruSize(valEl.Value.(*keyBaseValue).val.SizeByte()) lru.UpdateLruSize(structure.UpdateLength(valEl.Value.(*keyBaseValue).val.SizeByte()))
} }
// Get 查找key对应的value // Get 查找key对应的value
@ -125,7 +125,7 @@ func (lru *SingleCache) Del() error {
data := lru.li.Back() data := lru.li.Back()
delete(lru.lruMap, data.Value.(*keyBaseValue).key) delete(lru.lruMap, data.Value.(*keyBaseValue).key)
//删除大小 //删除大小
lru.UpdateLruSize(-1 * data.Value.(*keyBaseValue).val.SizeByte()) lru.UpdateLruSize(structure.UpdateLength(-1 * data.Value.(*keyBaseValue).val.SizeByte()))
lru.li.Remove(data) lru.li.Remove(data)
return nil return nil
} }