77
88import com .fasterxml .jackson .core .JsonProcessingException ;
99import com .fasterxml .jackson .databind .ObjectMapper ;
10- import com . google .protobuf .GeneratedMessage ;
11- import com . google .protobuf .InvalidProtocolBufferException ;
10+ import io . opentelemetry . testing . internal .protobuf .GeneratedMessage ;
11+ import io . opentelemetry . testing . internal .protobuf .InvalidProtocolBufferException ;
1212import com .google .protobuf .util .JsonFormat ;
1313import io .opentelemetry .javaagent .testing .common .AgentTestingExporterAccess ;
14- import io .opentelemetry .proto .collector .logs .v1 .ExportLogsServiceRequest ;
15- import io .opentelemetry .proto .collector .metrics .v1 .ExportMetricsServiceRequest ;
16- import io .opentelemetry .proto .collector .trace .v1 .ExportTraceServiceRequest ;
14+ import io .opentelemetry .testing .internal .proto .collector .logs .v1 .ExportLogsServiceRequest ;
15+ import io .opentelemetry .testing .internal .proto .collector .metrics .v1 .ExportMetricsServiceRequest ;
1716import io .opentelemetry .sdk .trace .data .SpanData ;
1817import io .opentelemetry .testing .internal .armeria .client .WebClient ;
18+ import io .opentelemetry .testing .internal .proto .collector .trace .v1 .ExportTraceServiceRequest ;
1919import java .util .Collection ;
2020import java .util .List ;
2121import java .util .concurrent .TimeUnit ;
2222import java .util .function .Supplier ;
2323import java .util .stream .Collectors ;
24- import java .util .stream .Stream ;
2524
2625public class JavaTelemetryRetriever {
2726 private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper ();
@@ -37,17 +36,10 @@ public void clearTelemetry() {
3736
3837 public List <SpanData > waitForTraces () {
3938 Collection <ExportTraceServiceRequest > collection =
40- waitForTelemetry ("get-traces" , () -> ExportTraceServiceRequest .newBuilder ());
41- Stream <io .opentelemetry .proto .trace .v1 .Span > objectStream =
42- collection .stream ()
43- .flatMap (
44- req ->
45- req .getResourceSpansList ().stream ()
46- .flatMap (
47- rs ->
48- rs .getScopeSpansList ().stream ()
49- .flatMap (ss -> ss .getSpansList ().stream ())));
50-
39+ waitForTelemetry ("get-traces" , () -> {
40+ ExportTraceServiceRequest .Builder builder = ExportTraceServiceRequest .newBuilder ();
41+ return builder ;
42+ });
5143 return AgentTestingExporterAccess .getSpanData (
5244 collection .stream ().flatMap (req -> req .getResourceSpansList ().stream ()));
5345 }
0 commit comments