TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-12-30 01:37:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化计数器相关代码

Browse Source
This commit is contained in:
GoEdgeLab
2023-10-05 16:29:02 +08:00
parent 72f2fde473
commit 451138cd0a
2 changed files with 34 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
internal/utils/counters/item.go
View File

@@ -33,14 +33,17 @@ func NewItem(lifeSeconds int) *Item {
}
}
func (this *Item) Increase() uint64 {
func (this *Item) Increase() (result uint64) {
var currentTime = fasttime.Now().Unix()
var currentSpanIndex = this.calculateSpanIndex(currentTime)
// return quickly
if this.lastUpdateTime == currentTime {
this.spans[currentSpanIndex]++
return this.Sum()
for _, count := range this.spans {
result += count
}
return
}
if this.lastUpdateTime > 0 {
@@ -50,16 +53,17 @@ func (this *Item) Increase() uint64 {
}
} else {
var lastSpanIndex = this.calculateSpanIndex(this.lastUpdateTime)
var countSpans = len(this.spans)
// reset values between LAST and CURRENT
for index := lastSpanIndex + 1; ; index++ {
var realIndex = index % countSpans
if realIndex <= currentSpanIndex {
if lastSpanIndex != currentSpanIndex {
var countSpans = len(this.spans)
// reset values between LAST and CURRENT
for index := lastSpanIndex + 1; ; index++ {
var realIndex = index % countSpans
this.spans[realIndex] = 0
}
if realIndex == currentSpanIndex {
break
if realIndex == currentSpanIndex {
break
}
}
}
}
@@ -68,7 +72,11 @@ func (this *Item) Increase() uint64 {
this.spans[currentSpanIndex]++
this.lastUpdateTime = currentTime
return this.Sum()
for _, count := range this.spans {
result += count
}
return
}
func (this *Item) Sum() (result uint64) {
@@ -84,10 +92,11 @@ func (this *Item) Sum() (result uint64) {
} else {
var lastSpanIndex = this.calculateSpanIndex(this.lastUpdateTime)
var countSpans = len(this.spans)
for index := 0; index < countSpans; index++ {
if (currentSpanIndex >= lastSpanIndex && (index <= lastSpanIndex || index >= currentSpanIndex /** a >=b **/)) ||
(currentSpanIndex < lastSpanIndex && index >= currentSpanIndex && index <= lastSpanIndex /** a < b **/) {
result += this.spans[index]
for index := currentSpanIndex + 1; ; index++ {
var realIndex = index % countSpans
result += this.spans[realIndex]
if realIndex == lastSpanIndex {
break
}
}
}