mirror of
https://github.com/crowdsecurity/crowdsec.git
synced 2025-05-10 20:05:55 +02:00
35 lines
1.2 KiB
Go
35 lines
1.2 KiB
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.RuntimeAlert, *types.Queue) (types.RuntimeAlert, *types.Queue)
|
|
|
|
AfterBucketPour(Bucket *BucketFactory) func(types.Event, *Leaky) *types.Event
|
|
}
|
|
|
|
type DumbProcessor struct{}
|
|
|
|
func (d *DumbProcessor) OnBucketInit(bucketFactory *BucketFactory) error {
|
|
return nil
|
|
}
|
|
|
|
func (d *DumbProcessor) OnBucketPour(bucketFactory *BucketFactory) func(types.Event, *Leaky) *types.Event {
|
|
return func(msg types.Event, leaky *Leaky) *types.Event {
|
|
return &msg
|
|
}
|
|
}
|
|
|
|
func (d *DumbProcessor) OnBucketOverflow(b *BucketFactory) func(*Leaky, types.RuntimeAlert, *types.Queue) (types.RuntimeAlert, *types.Queue) {
|
|
return func(leaky *Leaky, alert types.RuntimeAlert, queue *types.Queue) (types.RuntimeAlert, *types.Queue) {
|
|
return alert, queue
|
|
}
|
|
}
|
|
|
|
func (d *DumbProcessor) AfterBucketPour(bucketFactory *BucketFactory) func(types.Event, *Leaky) *types.Event {
|
|
return func(msg types.Event, leaky *Leaky) *types.Event {
|
|
return &msg
|
|
}
|
|
}
|