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

优化计数器代码

Browse Source
This commit is contained in:
刘祥超
2023-11-13 15:11:11 +08:00
parent d31dac75be
commit ecff37e080
1 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/utils/counters/counter.go
View File

@@ -66,19 +66,16 @@ func (this *Counter) Increase(key uint64, lifeSeconds int) uint64 {
var index = int(key % this.countMaps) var index = int(key % this.countMaps)
this.locker.RLock(index) this.locker.RLock(index)
var item = this.itemMaps[index][key] var item = this.itemMaps[index][key]
var l = len(this.itemMaps[index])
this.locker.RUnlock(index) this.locker.RUnlock(index)
if item == nil { // no need to care about duplication if item == nil { // no need to care about duplication
item = NewItem(lifeSeconds) if l > maxItemsPerGroup {
this.locker.Lock(index) return 1
// check again
oldItem, ok := this.itemMaps[index][key]
if !ok {
this.itemMaps[index][key] = item
} else {
item = oldItem
} }
item = NewItem(lifeSeconds)
this.locker.Lock(index)
this.itemMaps[index][key] = item
this.locker.Unlock(index) this.locker.Unlock(index)
} }