30 lines
870 B
Go
30 lines
870 B
Go
|
package leakybucket
|
||
|
|
|
||
|
|
import "github.com/crowdsecurity/crowdsec/pkg/types"
|
||
|
|
|
||
|
|
type Processor interface {
|
||
|
|
OnBucketInit(Bucket *BucketFactory) error
|
||
|
|
OnBucketPour(Bucket *BucketFactory) func(types.Event, *Leaky) *types.Event
|
||
|
|
OnBucketOverflow(Bucket *BucketFactory) func(*Leaky, types.SignalOccurence, *Queue) (types.SignalOccurence, *Queue)
|
||
|
|
}
|
||
|
|
|
||
|
|
type DumbProcessor struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DumbProcessor) OnBucketInit(b *BucketFactory) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DumbProcessor) OnBucketPour(b *BucketFactory) func(types.Event, *Leaky) *types.Event {
|
||
|
|
return func(msg types.Event, l *Leaky) *types.Event {
|
||
|
|
return &msg
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DumbProcessor) OnBucketOverflow(b *BucketFactory) func(*Leaky, types.SignalOccurence, *Queue) (types.SignalOccurence, *Queue) {
|
||
|
|
return func(l *Leaky, s types.SignalOccurence, q *Queue) (types.SignalOccurence, *Queue) {
|
||
|
|
return s, q
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|