This commit is contained in:
sabban 2023-07-07 15:46:30 +02:00
parent 0a4dcc5bd2
commit e8d6904399

View file

@ -1,9 +1,11 @@
package leakybucket
import (
"fmt"
"time"
"github.com/crowdsecurity/crowdsec/pkg/types"
"github.com/davecgh/go-spew/spew"
log "github.com/sirupsen/logrus"
)
@ -35,11 +37,13 @@ func TimeMachinePour(l *Leaky, msg types.Event) {
}
l.Last_ts = d
l.mutex.Unlock()
// fmt.Printf("event: %s", spew.Sdump(msg.Line.Raw))
if l.Limiter.AllowN(d, 1) {
fmt.Printf("event: %s", spew.Sdump(msg.Line.Raw))
l.logger.Tracef("Time-Pouring event %s (tokens:%f)", d, l.Limiter.GetTokensCount())
l.Queue.Add(msg)
} else {
fmt.Printf("overflow: %s", spew.Sdump(msg.Line.Raw))
l.Ovflw_ts = d
l.logger.Debugf("Bucket overflow at %s", l.Ovflw_ts)
l.Queue.Add(msg)