61 lines
1.7 KiB
Go
61 lines
1.7 KiB
Go
|
package counter_util
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// CounterUtil
|
|||
|
|
// @description: 固定窗口计数器辅助类
|
|||
|
|
type CounterUtil struct {
|
|||
|
|
tag string // tag counter标识
|
|||
|
|
num int // num 当前计数
|
|||
|
|
warnNum int // warnNum 警告数量
|
|||
|
|
windowsTime time.Time // windowsTime 窗口时间
|
|||
|
|
checkSameWindowsFun func(t1, t2 time.Time) bool // checkSameWindowsFun 比较是否同一个时间窗口
|
|||
|
|
warnAction func(tag string, num int, t time.Time) // warnAction 监控回调方法
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewCounterUtil
|
|||
|
|
// @description: 构造计数器
|
|||
|
|
// parameter:
|
|||
|
|
// @_tag:tag标识，会在WarnAction中传递回来
|
|||
|
|
// @_warnNum:告警数量
|
|||
|
|
// @_checkWindowFun:比较是否同一个时间窗口
|
|||
|
|
// @_warnAction:指定数量触发回调
|
|||
|
|
// return:
|
|||
|
|
// @*CounterUtil:固定窗口计数器辅助类
|
|||
|
|
func NewCounterUtil(_tag string, _warnNum int, _checkWindowFun func(t1, t2 time.Time) bool, _warnAction func(tag string, num int, t time.Time)) *CounterUtil {
|
|||
|
|
r := &CounterUtil{
|
|||
|
|
tag: _tag,
|
|||
|
|
warnNum: _warnNum,
|
|||
|
|
windowsTime: time.Now(),
|
|||
|
|
checkSameWindowsFun: _checkWindowFun,
|
|||
|
|
warnAction: _warnAction,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return r
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AddNum
|
|||
|
|
// @description: 添加数量
|
|||
|
|
// parameter:
|
|||
|
|
// @receiver c:计数器
|
|||
|
|
// @n:增加的数量
|
|||
|
|
// return:
|
|||
|
|
// @int:计数器当前的数量
|
|||
|
|
func (c *CounterUtil) AddNum(n int) int {
|
|||
|
|
if !c.checkSameWindowsFun(c.windowsTime, time.Now()) {
|
|||
|
|
c.num = 0
|
|||
|
|
c.windowsTime = time.Now()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 增加次数
|
|||
|
|
c.num += n
|
|||
|
|
if c.num >= c.warnNum {
|
|||
|
|
c.warnAction(c.tag, c.num, time.Now())
|
|||
|
|
c.num = 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return c.num
|
|||
|
|
}
|