fixing possible NPE in Apache Camel instrumentation (#2250)

This commit is contained in:
Jakub Wach 2021-02-11 11:38:10 +01:00 committed by GitHub
parent 9592229e36
commit 29780d97be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 71 additions and 3 deletions

View File

@ -4,7 +4,9 @@ dependencies {
testImplementation project(':instrumentation:apache-camel-2.20:javaagent')
testImplementation group: 'org.apache.camel', name: 'camel-core', version: '2.20.1'
testImplementation group: 'org.apache.camel', name: 'camel-aws', version: '2.20.1'
testImplementation group: 'org.apache.camel', name: 'camel-http', version: '2.20.1'
testImplementation deps.opentelemetryTraceProps
testImplementation deps.opentelemetryExtAws
testImplementation group: 'org.assertj', name: 'assertj-core', version: '3.18.1'
}

View File

@ -7,21 +7,86 @@ package io.opentelemetry.javaagent.instrumentation.apachecamel;
import static org.assertj.core.api.Assertions.assertThat;
import io.opentelemetry.api.GlobalOpenTelemetry;
import io.opentelemetry.api.OpenTelemetry;
import io.opentelemetry.api.trace.Span;
import io.opentelemetry.api.trace.SpanContext;
import io.opentelemetry.context.Context;
import io.opentelemetry.context.propagation.ContextPropagators;
import io.opentelemetry.extension.trace.propagation.JaegerPropagator;
import java.net.URI;
import java.util.Collections;
import java.util.Map;
import org.apache.camel.Endpoint;
import org.apache.camel.component.aws.sqs.SqsComponent;
import org.apache.camel.component.aws.sqs.SqsConfiguration;
import org.apache.camel.component.aws.sqs.SqsEndpoint;
import org.apache.camel.component.http.HttpComponent;
import org.apache.camel.component.http.HttpEndpoint;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
public class CamelPropagationUtilTest {
@BeforeAll
public static void setUp() {
GlobalOpenTelemetry.set(
OpenTelemetry.getPropagating(ContextPropagators.create(JaegerPropagator.getInstance())));
}
@Test
public void shouldExtractAwsParent() {
public void shouldExtractHttpParentForHttpEndpoint() throws Exception {
// given
Endpoint endpoint = new HttpEndpoint("", new HttpComponent(), URI.create(""));
Map<String, Object> exchangeHeaders =
Collections.singletonMap(
"uber-trace-id", "1f7f8dab3f0043b1b9cf0a75caf57510:a13825abcb764bd3:0:1");
// when
Context parent = CamelPropagationUtil.extractParent(exchangeHeaders, endpoint);
// then
Span parentSpan = Span.fromContext(parent);
SpanContext parentSpanContext = parentSpan.getSpanContext();
assertThat(parentSpanContext.getTraceId()).isEqualTo("1f7f8dab3f0043b1b9cf0a75caf57510");
assertThat(parentSpanContext.getSpanId()).isEqualTo("a13825abcb764bd3");
}
@Test
public void shouldNotFailExtractingNullHttpParentForHttpEndpoint() throws Exception {
// given
Endpoint endpoint = new HttpEndpoint("", new HttpComponent(), URI.create(""));
Map<String, Object> exchangeHeaders = Collections.singletonMap("uber-trace-id", null);
// when
Context parent = CamelPropagationUtil.extractParent(exchangeHeaders, endpoint);
// then
Span parentSpan = Span.fromContext(parent);
SpanContext parentSpanContext = parentSpan.getSpanContext();
assertThat(parentSpanContext.isValid()).isEqualTo(false);
}
@Test
public void shouldNotFailExtractingNullAwsParentForSqsEndpoint() {
// given
Endpoint endpoint = new SqsEndpoint("", new SqsComponent(), new SqsConfiguration());
Map<String, Object> exchangeHeaders = Collections.singletonMap("AWSTraceHeader", null);
// when
Context parent = CamelPropagationUtil.extractParent(exchangeHeaders, endpoint);
// then
Span parentSpan = Span.fromContext(parent);
SpanContext parentSpanContext = parentSpan.getSpanContext();
assertThat(parentSpanContext.isValid()).isEqualTo(false);
}
@Test
public void shouldExtractAwsParentForSqsEndpoint() {
// given
Endpoint endpoint = new SqsEndpoint("", new SqsComponent(), new SqsConfiguration());

View File

@ -58,8 +58,9 @@ final class CamelPropagationUtil {
}
@Override
public String get(Map<String, Object> map, String s) {
return (map.containsKey(s) ? map.get(s).toString() : null);
public String get(Map<String, Object> map, String key) {
Object value = map.get(key);
return (value == null ? null : value.toString());
}
}