TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-12-15 23:26:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
GoEdgeLab
2023-10-11 14:07:13 +08:00
parent bf68c29fa5
commit 0e45e75167
4 changed files with 45 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/ttlcache/piece.go
View File

@@ -24,11 +24,19 @@ func NewPiece[T any](maxItems int) *Piece[T] {
}
func (this *Piece[T]) Add(key uint64, item *Item[T]) (ok bool) {
this.locker.Lock()
this.locker.RLock()
if this.maxItems > 0 && len(this.m) >= this.maxItems {
this.locker.Unlock()
this.locker.RUnlock()
return
}
this.locker.RUnlock()
this.locker.Lock()
oldItem, exists := this.m[key]
if exists && oldItem.expiredAt == item.expiredAt {
this.locker.Unlock()
return true
}
this.m[key] = item
this.locker.Unlock()