-
Notifications
You must be signed in to change notification settings - Fork 0
/
slide.go
87 lines (73 loc) · 1.43 KB
/
slide.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package util
import (
"sync"
"time"
)
// 请求次数
var countLockSlide = 0
// 单次请求
type slideItem struct {
time time.Time // 请求时间
allow bool // 是否允许
}
/**
滑动窗口计数器
*/
type CounterSlide struct {
rate int //计数周期内最多允许的请求数
cycle time.Duration //计数周期
itemMap map[int]slideItem
lock sync.Mutex
}
func (l *CounterSlide) Allow() bool {
l.lock.Lock()
defer l.lock.Unlock()
countLockSlide++
tmp := slideItem{
time: time.Now(),
}
if l.checkItem(tmp.time) {
tmp.allow = true
} else {
tmp.allow = false
}
l.itemMap[countLockSlide] = tmp
return tmp.allow
}
// 处理map元素
func (l *CounterSlide) dealMap(now time.Time) {
for key, value := range l.itemMap {
thisTime := value.time
if now.Sub(thisTime) > l.cycle {
delete(l.itemMap, key)
}
}
}
// 检查itemArr 中的元素
// 这里也可以只记录允许通过的记录 用切片记录即可
func (l *CounterSlide) checkItem(now time.Time) bool {
var limit = 0
for key, value := range l.itemMap {
thisTime := value.time
if now.Sub(thisTime) > l.cycle {
delete(l.itemMap, key)
continue
}
// 允许
if value.allow {
limit++
}
}
if limit < l.rate {
return true
}
return false
}
func (l *CounterSlide) Set(r int, cycle time.Duration) {
if r <= 0 || cycle <= 0 {
panic("参数异常")
}
l.rate = r
l.cycle = cycle
l.itemMap = make(map[int]slideItem)
}