Use internalSet from AttributesExtractorUtil in AWS SDK v1 SnsAttribu…

…tesExtractor.
open-telemetry · Dec 22, 2023 · 9a6ce1c · 9a6ce1c
1 parent 62cc322
commit 9a6ce1c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...y/src/main/java/io/opentelemetry/instrumentation/awssdk/v1_11/SnsAttributesExtractor.java b/...y/src/main/java/io/opentelemetry/instrumentation/awssdk/v1_11/SnsAttributesExtractor.java
@@ -11,16 +11,16 @@
 import io.opentelemetry.api.common.AttributesBuilder;
 import io.opentelemetry.context.Context;
 import io.opentelemetry.instrumentation.api.instrumenter.AttributesExtractor;
+import io.opentelemetry.instrumentation.api.internal.AttributesExtractorUtil;
 import io.opentelemetry.semconv.SemanticAttributes;
 import javax.annotation.Nullable;
 
 public class SnsAttributesExtractor implements AttributesExtractor<Request<?>, Response<?>> {
   @Override
   public void onStart(AttributesBuilder attributes, Context parentContext, Request<?> request) {
     String destination = findMessageDestination(request.getOriginalRequest());
-    if (destination != null) {
-      attributes.put(SemanticAttributes.MESSAGING_DESTINATION_NAME, destination);
-    }
+    AttributesExtractorUtil.internalSet(attributes, SemanticAttributes.MESSAGING_DESTINATION_NAME,
+        destination);
   }
 
   /*