Get archive traces for es

Signed-off-by: Emmanuel Emonueje Ebenezer <[email protected]>

Author: ekefan

plugin/storage/integration/integration.go

@@ -207,11 +207,12 @@ func (s *StorageIntegration) testArchiveTrace(t *testing.T) {
 	var actual *model.Trace
 	found := s.waitForCondition(t, func(_ *testing.T) bool {
 		var err error
-		iterTraces := s.ArchiveTraceReader.GetTraces(context.Background(), tracestore.GetTraceParams{TraceID: tID.ToOTELTraceID()})
-		traces, err := v1adapter.V1TracesFromSeq2(iterTraces)
-		if len(traces) > 0 {
-			actual = traces[0]
-		}
+		// iterTraces := s.ArchiveTraceReader.GetTraces(context.Background(), tracestore.GetTraceParams{TraceID: tID.ToOTELTraceID()})
+		// traces, err := v1adapter.V1TracesFromSeq2(iterTraces)
+		// if len(traces) > 0 {
+		// 	actual = traces[0]
+		// }
+		actual, err = s.ArchiveTraceReader.(*v1adapter.TraceReader).GetArchiveTraces(context.Background(), tracestore.GetTraceParams{TraceID: tID.ToOTELTraceID()})
 		return err == nil && len(actual.Spans) >= len(expected.Spans)
 	})
 	require.True(t, found)

storage_v2/v1adapter/tracereader.go

@@ -65,6 +65,11 @@ func (tr *TraceReader) GetTraces(
 	}
 }
 
+func (tr *TraceReader) GetArchiveTraces(ctx context.Context, traceID tracestore.GetTraceParams) (*model.Trace, error) {
+	tID := model.TraceIDFromOTEL(traceID.TraceID)
+	return tr.spanReader.GetTrace(ctx, spanstore.GetTraceParameters{TraceID: tID})
+}
+
 func (tr *TraceReader) GetServices(ctx context.Context) ([]string, error) {
 	return tr.spanReader.GetServices(ctx)
 }