From 5a42b2b1f7fba1e109d0c0db57e1881ab41e8d6f Mon Sep 17 00:00:00 2001 From: lspgn Date: Tue, 10 Aug 2021 22:32:43 -0700 Subject: [PATCH] Enricher can set sampling rate --- cmd/enricher/main.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/cmd/enricher/main.go b/cmd/enricher/main.go index 23ef5db..e6379d7 100644 --- a/cmd/enricher/main.go +++ b/cmd/enricher/main.go @@ -43,6 +43,8 @@ var ( LogLevel = flag.String("loglevel", "info", "Log level") LogFmt = flag.String("logfmt", "normal", "Log formatter") + SamplingRate = flag.Int("samplingrate", 0, "Set sampling rate (values > 0)") + Format = flag.String("format", "json", fmt.Sprintf("Choose the format (available: %s)", strings.Join(format.GetFormats(), ", "))) Transport = flag.String("transport", "file", fmt.Sprintf("Choose the transport (available: %s)", strings.Join(transport.GetTransports(), ", "))) @@ -163,6 +165,10 @@ func main() { MapFlow(dbAsn, dbCountry, msg) + if *SamplingRate > 0 { + msg.SamplingRate = uint64(*SamplingRate) + } + key, data, err := formatter.Format(msg) if err != nil { log.Error(err)