77 lines
2.1 KiB
Go
77 lines
2.1 KiB
Go
package email
|
||
|
||
import (
|
||
"fmt"
|
||
"time"
|
||
)
|
||
|
||
//FastDetector detecte speed and unique analyze
|
||
type FastDetector struct {
|
||
lastSec int64
|
||
sameSecCnt int
|
||
sameSecThreshold int
|
||
overspeedCnt int
|
||
overspeedThreshold int
|
||
uniqueSpeed map[int64]int
|
||
fastUnique int64
|
||
}
|
||
|
||
//NewFastDetector new
|
||
func NewFastDetector(speedThreshold, overspeedThreshold int) *FastDetector {
|
||
return &FastDetector{
|
||
lastSec: time.Now().Unix(),
|
||
sameSecCnt: 0,
|
||
sameSecThreshold: speedThreshold,
|
||
overspeedCnt: 0,
|
||
overspeedThreshold: overspeedThreshold,
|
||
uniqueSpeed: map[int64]int{},
|
||
}
|
||
}
|
||
|
||
//String string info
|
||
func (fd *FastDetector) String() string {
|
||
return fmt.Sprintf("same_sec_cnt=%d,overspeed_cnt=%d,fast_unique=%d,unique_speed=%v",
|
||
fd.sameSecCnt, fd.overspeedCnt, fd.fastUnique, fd.uniqueSpeed)
|
||
}
|
||
|
||
//Detect 快慢探查, 超限名单只能被慢速/下一个超速名单/间隔5s后空名单代替，超限名单只有在(overspeedthreshold+1) * samesecthreshold时才确定，此时返回true
|
||
func (fd *FastDetector) Detect(unique int64) (fast bool) {
|
||
now := time.Now().Unix()
|
||
//连续n次超限
|
||
if now == fd.lastSec {
|
||
fd.sameSecCnt++
|
||
if fd.sameSecCnt == fd.sameSecThreshold {
|
||
fd.overspeedCnt++
|
||
}
|
||
} else {
|
||
if fd.sameSecCnt < fd.sameSecThreshold || (now-fd.lastSec > 5) {
|
||
fd.overspeedCnt = 0
|
||
fd.fastUnique = 0
|
||
fd.uniqueSpeed = map[int64]int{}
|
||
}
|
||
fd.sameSecCnt = 1
|
||
fd.lastSec = now
|
||
}
|
||
|
||
//连续超限后，最先超限的unique指定为超限名单
|
||
if fd.overspeedCnt == fd.overspeedThreshold && fd.sameSecCnt == fd.sameSecThreshold {
|
||
fd.uniqueSpeed[unique] = 0
|
||
}
|
||
if (fd.overspeedCnt == fd.overspeedThreshold && fd.sameSecCnt != fd.sameSecThreshold) || (fd.overspeedCnt > fd.overspeedThreshold) {
|
||
fd.uniqueSpeed[unique]++
|
||
if fd.uniqueSpeed[unique] >= fd.sameSecThreshold {
|
||
fast = true
|
||
fd.fastUnique = unique //指定超限名单
|
||
fd.uniqueSpeed = map[int64]int{}
|
||
fd.overspeedCnt = 0
|
||
}
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
//IsFastUnique 是否为超限名单
|
||
func (fd *FastDetector) IsFastUnique(unique int64) bool {
|
||
return fd.fastUnique == unique
|
||
}
|