Skip to content

Commit 34a74b3

Browse files
committed
Wrap level and generic field detection with pprof annotation
Signed-off-by: Christian Haudum <christian.haudum@gmail.com>
1 parent 202ecbc commit 34a74b3

File tree

1 file changed

+14
-9
lines changed

1 file changed

+14
-9
lines changed

‎pkg/distributor/distributor.go

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import (
66
"fmt"
77
"math"
88
"net/http"
9+
"runtime/pprof"
910
"slices"
1011
"sort"
1112
"strconv"
@@ -548,18 +549,22 @@ func (d *Distributor) Push(ctx context.Context, req *logproto.PushRequest) (*log
548549
}
549550
}
550551
if shouldDiscoverLevels {
551-
logLevel, ok := fieldDetector.extractLogLevel(lbs, structuredMetadata, entry)
552-
if ok {
553-
entry.StructuredMetadata = append(entry.StructuredMetadata, logLevel)
554-
}
552+
pprof.Do(ctx, pprof.Labels("action", "discover_log_level"), func(ctx context.Context) {
553+
logLevel, ok := fieldDetector.extractLogLevel(lbs, structuredMetadata, entry)
554+
if ok {
555+
entry.StructuredMetadata = append(entry.StructuredMetadata, logLevel)
556+
}
557+
})
555558
}
556559
if shouldDiscoverGenericFields {
557-
for field, hints := range fieldDetector.validationContext.discoverGenericFields {
558-
extracted, ok := fieldDetector.extractGenericField(field, hints, lbs, structuredMetadata, entry)
559-
if ok {
560-
entry.StructuredMetadata = append(entry.StructuredMetadata, extracted)
560+
pprof.Do(ctx, pprof.Labels("action", "discover_generic_fields"), func(ctx context.Context) {
561+
for field, hints := range fieldDetector.validationContext.discoverGenericFields {
562+
extracted, ok := fieldDetector.extractGenericField(field, hints, lbs, structuredMetadata, entry)
563+
if ok {
564+
entry.StructuredMetadata = append(entry.StructuredMetadata, extracted)
565+
}
561566
}
562-
}
567+
})
563568
}
564569
stream.Entries[n] = entry
565570

0 commit comments

Comments
 (0)