test changes

DataDog · Nov 8, 2024 · 6f3a82a · 6f3a82a
1 parent 2c25759
commit 6f3a82a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/extension/extension.go b/internal/extension/extension.go
@@ -183,6 +183,10 @@ func (em *ExtensionManager) SendEndInvocationRequest(ctx context.Context, functi
 		}
 		req.Header.Set(string(DdTraceId), fmt.Sprint(functionExecutionSpan.Context().TraceID()))
 		req.Header.Set(string(DdSpanId), fmt.Sprint(functionExecutionSpan.Context().SpanID()))
+		if samplingPriority, ok := ctx.Value("_sampling_priority_v1").(string); ok {
+			fmt.Printf("priority: %s\n", priority)
+			req.Header.Set(string(DdSamplingPriority), samplingPriority)
+		}
 	}
 	fmt.Printf("[DEBUG lambda] extension.go context: %+v\n", ctx)