Skip to content

Commit 4b0942c

Browse files
committed
fix kafka stream receive span test
1 parent 9e5b02d commit 4b0942c

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

instrumentation/kafka/kafka-streams-0.11/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/kafkastreams/KafkaStreamsSuppressReceiveSpansTest.java

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -71,15 +71,14 @@ private static KStream<Integer, String> stream(Object builder)
7171
ClassNotFoundException {
7272
Method streamMethod;
7373
try {
74-
streamMethod =
75-
Class.forName("org.apache.kafka.streams.kstream.KStreamBuilder")
76-
.getMethod("stream", String[].class);
74+
Class.forName("org.apache.kafka.streams.kstream.KStreamBuilder");
75+
return ((org.apache.kafka.streams.kstream.KStreamBuilder) builder).stream(STREAM_PENDING);
7776
} catch (ClassNotFoundException e) {
7877
streamMethod =
7978
Class.forName("org.apache.kafka.streams.StreamsBuilder")
8079
.getMethod("stream", String.class);
80+
return (KStream<Integer, String>) streamMethod.invoke(builder, STREAM_PENDING);
8181
}
82-
return (KStream<Integer, String>) streamMethod.invoke(builder, STREAM_PENDING);
8382
}
8483

8584
@DisplayName("test kafka produce and consume with streams in-between")

0 commit comments

Comments
 (0)