mirror of
https://github.com/akvorado/akvorado.git
synced 2025-12-12 06:24:10 +01:00
a449736a62
Done with: ``` git grep -l 'for.*:= 0.*++' \ | xargs sed -i -E 's/for (.*) := 0; \1 < (.*); \1\+\+/for \1 := range \2/' ``` And a few manual fixes due to unused variables. There is something fishy in BMP rib test. Add a comment about that. This is not equivalent (as with range, random is evaluated once, while in the original loop, it is evaluated at each iteration). I believe the intent was to behave like with range.
93 lines
2.9 KiB
Go
93 lines
2.9 KiB
Go
// SPDX-FileCopyrightText: 2023 Free Mobile
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
package flow
|
|
|
|
import (
|
|
"net"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"akvorado/common/helpers"
|
|
"akvorado/common/reporter"
|
|
"akvorado/common/schema"
|
|
"akvorado/inlet/flow/decoder"
|
|
"akvorado/inlet/flow/decoder/netflow"
|
|
"akvorado/inlet/flow/decoder/sflow"
|
|
)
|
|
|
|
// The goal is to benchmark flow decoding + encoding to protobuf
|
|
|
|
func BenchmarkDecodeEncodeNetflow(b *testing.B) {
|
|
schema.DisableDebug(b)
|
|
r := reporter.NewMock(b)
|
|
sch := schema.NewMock(b)
|
|
nfdecoder := netflow.New(r, decoder.Dependencies{Schema: sch}, decoder.Option{TimestampSource: decoder.TimestampSourceUDP})
|
|
|
|
template := helpers.ReadPcapL4(b, filepath.Join("decoder", "netflow", "testdata", "options-template.pcap"))
|
|
got := nfdecoder.Decode(decoder.RawFlow{Payload: template, Source: net.ParseIP("127.0.0.1")})
|
|
if got == nil || len(got) != 0 {
|
|
b.Fatal("Decode() error on options template")
|
|
}
|
|
data := helpers.ReadPcapL4(b, filepath.Join("decoder", "netflow", "testdata", "options-data.pcap"))
|
|
got = nfdecoder.Decode(decoder.RawFlow{Payload: data, Source: net.ParseIP("127.0.0.1")})
|
|
if got == nil || len(got) != 0 {
|
|
b.Fatal("Decode() error on options data")
|
|
}
|
|
template = helpers.ReadPcapL4(b, filepath.Join("decoder", "netflow", "testdata", "template.pcap"))
|
|
got = nfdecoder.Decode(decoder.RawFlow{Payload: template, Source: net.ParseIP("127.0.0.1")})
|
|
if got == nil || len(got) != 0 {
|
|
b.Fatal("Decode() error on template")
|
|
}
|
|
data = helpers.ReadPcapL4(b, filepath.Join("decoder", "netflow", "testdata", "data.pcap"))
|
|
|
|
for _, withEncoding := range []bool{true, false} {
|
|
title := map[bool]string{
|
|
true: "with encoding",
|
|
false: "without encoding",
|
|
}[withEncoding]
|
|
b.Run(title, func(b *testing.B) {
|
|
for range b.N {
|
|
got = nfdecoder.Decode(decoder.RawFlow{Payload: data, Source: net.ParseIP("127.0.0.1")})
|
|
if withEncoding {
|
|
for _, flow := range got {
|
|
sch.ProtobufMarshal(flow)
|
|
}
|
|
}
|
|
}
|
|
if got[0].ProtobufDebug != nil {
|
|
b.Fatal("debug is enabled")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func BenchmarkDecodeEncodeSflow(b *testing.B) {
|
|
schema.DisableDebug(b)
|
|
r := reporter.NewMock(b)
|
|
sch := schema.NewMock(b)
|
|
sdecoder := sflow.New(r, decoder.Dependencies{Schema: sch}, decoder.Option{TimestampSource: decoder.TimestampSourceUDP})
|
|
data := helpers.ReadPcapL4(b, filepath.Join("decoder", "sflow", "testdata", "data-1140.pcap"))
|
|
|
|
for _, withEncoding := range []bool{true, false} {
|
|
title := map[bool]string{
|
|
true: "with encoding",
|
|
false: "without encoding",
|
|
}[withEncoding]
|
|
var got []*schema.FlowMessage
|
|
b.Run(title, func(b *testing.B) {
|
|
for range b.N {
|
|
got = sdecoder.Decode(decoder.RawFlow{Payload: data, Source: net.ParseIP("127.0.0.1")})
|
|
if withEncoding {
|
|
for _, flow := range got {
|
|
sch.ProtobufMarshal(flow)
|
|
}
|
|
}
|
|
}
|
|
if got[0].ProtobufDebug != nil {
|
|
b.Fatal("debug is enabled")
|
|
}
|
|
})
|
|
}
|
|
}
|