add tracing

rusq · Apr 27, 2023 · 84de11a · 84de11a
1 parent ebe2eb6
commit 84de11a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/internal/chunk/file.go b/internal/chunk/file.go
@@ -403,7 +403,9 @@ func (f *File) Sorted(ctx context.Context, descending bool, fn func(ts time.Time
 		}
 	}
 
-	sort.Slice(tsList, sf)
+	trace.WithRegion(ctx, "sorted.sort", func() {
+		sort.Slice(tsList, sf)
+	})
 
 	var (
 		prevOffset int64 // previous chunk offset, used to avoid seeking
@@ -420,7 +422,10 @@ func (f *File) Sorted(ctx context.Context, descending bool, fn func(ts time.Time
 			}
 			prevOffset = tmOff.Offset
 		}
-		if err := fn(structures.Int2Time(ts).UTC(), &chunk.Messages[tmOff.Index]); err != nil {
+		rgn := trace.StartRegion(ctx, "sorted.callback")
+		err = fn(structures.Int2Time(ts).UTC(), &chunk.Messages[tmOff.Index])
+		rgn.End()
+		if err != nil {
 			return err
 		}
 	}