Change Adjuster To Work On Seq

Signed-off-by: Mahad Zaryab <[email protected]>
jaegertracing · Dec 24, 2024 · 1d721ac · 1d721ac
1 parent ffad629
commit 1d721ac
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/cmd/query/app/querysvc/query_service_v2.go b/cmd/query/app/querysvc/query_service_v2.go
@@ -144,8 +144,13 @@ func (qs QueryServiceV2) ArchiveTrace(ctx context.Context, traceID pcommon.Trace
 }
 
 // Adjust applies adjusters to the trace.
-func (qs QueryServiceV2) Adjust(trace ptrace.Traces) {
-	qs.options.Adjuster.Adjust(trace)
+func (qs QueryServiceV2) Adjust(tracesIter iter.Seq[[]ptrace.Traces]) {
+	tracesIter(func(traces []ptrace.Traces) bool {
+		for _, trace := range traces {
+			qs.options.Adjuster.Adjust(trace)
+		}
+		return true
+	})
 }
 
 // GetDependencies implements depstore.Reader.GetDependencies