TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-03 23:20:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
GoEdgeLab
2021-05-24 09:37:37 +08:00
parent 46ecb8e47d
commit 98c0f7113a
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/caches/item.go
View File

@@ -23,7 +23,7 @@ func (this *Item) IsExpired() bool {
} }
func (this *Item) TotalSize() int64 { func (this *Item) TotalSize() int64 {
return this.Size() + this.MetaSize + int64(len(this.Key)) return this.Size() + this.MetaSize + int64(len(this.Key)) + 64
} }
func (this *Item) Size() int64 { func (this *Item) Size() int64 {

12
internal/caches/storage_memory.go
View File

@@ -42,10 +42,10 @@ func NewMemoryStorage(policy *serverconfigs.HTTPCachePolicy) *MemoryStorage {
// Init 初始化 // Init 初始化
func (this *MemoryStorage) Init() error { func (this *MemoryStorage) Init() error {
this.list.OnAdd(func(item *Item) { this.list.OnAdd(func(item *Item) {
atomic.AddInt64(&this.totalSize, item.Size()) atomic.AddInt64(&this.totalSize, item.TotalSize())
}) })
this.list.OnRemove(func(item *Item) { this.list.OnRemove(func(item *Item) {
atomic.AddInt64(&this.totalSize, -item.Size()) atomic.AddInt64(&this.totalSize, -item.TotalSize())
}) })
if this.purgeDuration <= 0 { if this.purgeDuration <= 0 {
@@ -118,7 +118,7 @@ func (this *MemoryStorage) Delete(key string) error {
hash := this.hash(key) hash := this.hash(key)
this.locker.Lock() this.locker.Lock()
delete(this.valuesMap, hash) delete(this.valuesMap, hash)
this.list.Remove(fmt.Sprintf("%d", hash)) _ = this.list.Remove(fmt.Sprintf("%d", hash))
this.locker.Unlock() this.locker.Unlock()
return nil return nil
} }
@@ -137,7 +137,7 @@ func (this *MemoryStorage) Stat() (*Stat, error) {
func (this *MemoryStorage) CleanAll() error { func (this *MemoryStorage) CleanAll() error {
this.locker.Lock() this.locker.Lock()
this.valuesMap = map[uint64]*MemoryItem{} this.valuesMap = map[uint64]*MemoryItem{}
this.list.Reset() _ = this.list.Reset()
atomic.StoreInt64(&this.totalSize, 0) atomic.StoreInt64(&this.totalSize, 0)
this.locker.Unlock() this.locker.Unlock()
return nil return nil
@@ -173,7 +173,7 @@ func (this *MemoryStorage) Stop() {
defer this.locker.Unlock() defer this.locker.Unlock()
this.valuesMap = map[uint64]*MemoryItem{} this.valuesMap = map[uint64]*MemoryItem{}
this.list.Reset() _ = this.list.Reset()
if this.ticker != nil { if this.ticker != nil {
this.ticker.Stop() this.ticker.Stop()
} }
@@ -188,7 +188,7 @@ func (this *MemoryStorage) Policy() *serverconfigs.HTTPCachePolicy {
func (this *MemoryStorage) AddToList(item *Item) { func (this *MemoryStorage) AddToList(item *Item) {
item.MetaSize = int64(len(item.Key)) + 32 /** 32是我们评估的数据结构的长度 **/ item.MetaSize = int64(len(item.Key)) + 32 /** 32是我们评估的数据结构的长度 **/
hash := fmt.Sprintf("%d", this.hash(item.Key)) hash := fmt.Sprintf("%d", this.hash(item.Key))
this.list.Add(hash, item) _ = this.list.Add(hash, item)
} }
// TotalDiskSize 消耗的磁盘尺寸 // TotalDiskSize 消耗的磁盘尺寸