forked from actionpay/postmanq
-
Notifications
You must be signed in to change notification settings - Fork 2
/
cleaner.go
49 lines (44 loc) · 1.46 KB
/
cleaner.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
package limiter
import (
"context"
"sync/atomic"
"time"
)
// Cleaner чистильщик, проверяет значения ограничений и обнуляет значения ограничений
type Cleaner struct {
service *Service
ticker *time.Ticker
ctx context.Context
done context.CancelFunc
}
// создает нового чистильщика
func newCleaner(service *Service) *Cleaner {
ctx, done := context.WithCancel(context.Background())
return &Cleaner{service: service, ticker: time.NewTicker(time.Second), ctx: ctx, done: done}
}
// проверяет значения ограничений и обнуляет значения ограничений
func (c *Cleaner) run() {
for {
select {
case <-c.ctx.Done():
return
case now := <-c.ticker.C:
// смотрим все ограничения
for _, conf := range c.service.Configs {
for _, limit := range conf.Limits {
// проверяем дату последнего изменения ограничения
if !limit.isValidDuration(now) {
// если дата последнего изменения выходит за промежуток времени для проверки
// обнулям текущее количество отправленных писем
atomic.StoreInt32(&limit.currentValue, 0)
limit.modifyDate = time.Now()
}
}
}
}
}
}
func (c *Cleaner) stop() {
c.ticker.Stop()
c.done()
}