mirror of
https://gitee.com/johng/gf.git
synced 2024-12-01 11:48:09 +08:00
105 lines
2.8 KiB
Go
105 lines
2.8 KiB
Go
// Copyright 2018 gf Author(https://github.com/gogf/gf). All Rights Reserved.
|
||
//
|
||
// This Source Code Form is subject to the terms of the MIT License.
|
||
// If a copy of the MIT was not distributed with this file,
|
||
// You can obtain one at https://github.com/gogf/gf.
|
||
|
||
package gcache
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/gogf/gf/g/container/glist"
|
||
"github.com/gogf/gf/g/container/gmap"
|
||
"github.com/gogf/gf/g/container/gtype"
|
||
"github.com/gogf/gf/g/os/gtimer"
|
||
"time"
|
||
)
|
||
|
||
// LRU算法实现对象，底层双向链表使用了标准库的list.List
|
||
type memCacheLru struct {
|
||
cache *memCache // 所属Cache对象
|
||
data *gmap.Map // 记录键名与链表中的位置项指针
|
||
list *glist.List // 键名历史记录链表
|
||
rawList *glist.List // 事件列表
|
||
closed *gtype.Bool // 是否关闭
|
||
}
|
||
|
||
// 创建LRU管理对象
|
||
func newMemCacheLru(cache *memCache) *memCacheLru {
|
||
lru := &memCacheLru {
|
||
cache : cache,
|
||
data : gmap.New(),
|
||
list : glist.New(),
|
||
rawList : glist.New(),
|
||
closed : gtype.NewBool(),
|
||
}
|
||
gtimer.AddSingleton(time.Second, lru.SyncAndClear)
|
||
return lru
|
||
}
|
||
|
||
// 关闭LRU对象
|
||
func (lru *memCacheLru) Close() {
|
||
lru.closed.Set(true)
|
||
}
|
||
|
||
// 删除指定数据项
|
||
func (lru *memCacheLru) Remove(key interface{}) {
|
||
if v := lru.data.Get(key); v != nil {
|
||
lru.data.Remove(key)
|
||
lru.list.Remove(v.(*glist.Element))
|
||
}
|
||
}
|
||
|
||
// 当前LRU数据大小
|
||
func (lru *memCacheLru) Size() int {
|
||
return lru.data.Size()
|
||
}
|
||
|
||
// 添加LRU数据项
|
||
func (lru *memCacheLru) Push(key interface{}) {
|
||
lru.rawList.PushBack(key)
|
||
}
|
||
|
||
// 从链表尾删除LRU数据项，并返回对应数据
|
||
func (lru *memCacheLru) Pop() interface{} {
|
||
if v := lru.list.PopBack(); v != nil {
|
||
lru.data.Remove(v)
|
||
return v
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 从链表头打印LRU链表值
|
||
func (lru *memCacheLru) Print() {
|
||
for _, v := range lru.list.FrontAll() {
|
||
fmt.Printf("%v ", v)
|
||
}
|
||
fmt.Println()
|
||
}
|
||
|
||
// 异步执行协程，将queue中的数据同步到list中
|
||
func (lru *memCacheLru) SyncAndClear() {
|
||
if lru.closed.Val() {
|
||
gtimer.Exit()
|
||
return
|
||
}
|
||
// 数据同步
|
||
for {
|
||
if v := lru.rawList.PopFront(); v != nil {
|
||
// 删除对应链表项
|
||
if v := lru.data.Get(v); v != nil {
|
||
lru.list.Remove(v.(*glist.Element))
|
||
}
|
||
// 将数据插入到链表头，并记录对应的链表项到哈希表中，便于检索
|
||
lru.data.Set(v, lru.list.PushFront(v))
|
||
} else {
|
||
break
|
||
}
|
||
}
|
||
// 数据清理
|
||
for i := lru.Size() - lru.cache.cap; i > 0; i-- {
|
||
if s := lru.Pop(); s != nil {
|
||
lru.cache.clearByKey(s, true)
|
||
}
|
||
}
|
||
} |