Propagate sampling decision as attribute

majanjua-amzn · majanjua-amzn · commit 24dad10044ff · 2025-08-22T15:02:15.000-07:00
diff --git a/awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/AttributePropagatingSpanProcessor.java b/awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/AttributePropagatingSpanProcessor.java
@@ -112,6 +112,7 @@ public void onStart(Context parentContext, ReadWriteSpan span) {
     if (propagationData != null) {
       span.setAttribute(propagationDataKey, propagationData);
     }
+    span.setAttribute(AwsAttributeKeys.AWS_TRACE_FLAG_SAMPLED, span.getSpanContext().isSampled());
   }
 
   private boolean isConsumerKind(ReadableSpan span) {
diff --git a/awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/AwsAttributeKeys.java b/awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/AwsAttributeKeys.java
@@ -106,9 +106,6 @@ private AwsAttributeKeys() {}
   static final AttributeKey<Boolean> AWS_TRACE_FLAG_SAMPLED =
       AttributeKey.booleanKey("aws.trace.flag.sampled");
 
-  static final AttributeKey<String> AWS_XRAY_SAMPLING_RULE =
-      AttributeKey.stringKey("aws.xray.sampling_rule");
-
   // use the same AWS Resource attribute name defined by OTel java auto-instr for aws_sdk_v_1_1
   // TODO: all AWS specific attributes should be defined in semconv package and reused cross all
   // otel packages. Related sim -

Original file line number	Diff line number	Diff line change
`@@ -112,6 +112,7 @@ public void onStart(Context parentContext, ReadWriteSpan span) {`
`112`	`112`	`if (propagationData != null) {`
`113`	`113`	`span.setAttribute(propagationDataKey, propagationData);`
`114`	`114`	`}`
	`115`	`+ span.setAttribute(AwsAttributeKeys.AWS_TRACE_FLAG_SAMPLED, span.getSpanContext().isSampled());`
`115`	`116`	`}`
`116`	`117`
`117`	`118`	`private boolean isConsumerKind(ReadableSpan span) {`