[Sampler.AWS] Wrapped root sampler with ParentBasedSampler

src/OpenTelemetry.Sampler.AWS/.publicApi/PublicAPI.Unshipped.txt

@@ -2,7 +2,7 @@
 OpenTelemetry.Sampler.AWS.AWSXRayRemoteSampler
 OpenTelemetry.Sampler.AWS.AWSXRayRemoteSampler.Dispose() -> void
 OpenTelemetry.Sampler.AWS.AWSXRayRemoteSamplerBuilder
-OpenTelemetry.Sampler.AWS.AWSXRayRemoteSamplerBuilder.Build() -> OpenTelemetry.Sampler.AWS.AWSXRayRemoteSampler!
+OpenTelemetry.Sampler.AWS.AWSXRayRemoteSamplerBuilder.Build() -> OpenTelemetry.Trace.Sampler!
 OpenTelemetry.Sampler.AWS.AWSXRayRemoteSamplerBuilder.SetEndpoint(string! endpoint) -> OpenTelemetry.Sampler.AWS.AWSXRayRemoteSamplerBuilder!
 OpenTelemetry.Sampler.AWS.AWSXRayRemoteSamplerBuilder.SetPollingInterval(System.TimeSpan pollingInterval) -> OpenTelemetry.Sampler.AWS.AWSXRayRemoteSamplerBuilder!
 override OpenTelemetry.Sampler.AWS.AWSXRayRemoteSampler.ShouldSample(in OpenTelemetry.Trace.SamplingParameters samplingParameters) -> OpenTelemetry.Trace.SamplingResult

src/OpenTelemetry.Sampler.AWS/AWSXRayRemoteSamplerBuilder.cs

@@ -2,6 +2,7 @@
 // SPDX-License-Identifier: Apache-2.0
 
 using OpenTelemetry.Resources;
+using OpenTelemetry.Trace;
 
 namespace OpenTelemetry.Sampler.AWS;
 
@@ -66,8 +67,17 @@ public AWSXRayRemoteSamplerBuilder SetEndpoint(string endpoint)
     /// <summary>
     /// Returns a <see cref="AWSXRayRemoteSampler"/> with configuration of this builder.
     /// </summary>
-    /// <returns>an instance of <see cref="AWSXRayRemoteSampler"/>.</returns>
-    public AWSXRayRemoteSampler Build()
+    /// <returns>an instance of <see cref="Trace.Sampler"/>.</returns>
+    public Trace.Sampler Build()
+    {
+        using var rootSampler = this.BuildXraySampler();
+        return new ParentBasedSampler(rootSampler);
+    }
+
+    // This is intended for testing to check that the XRayRemoteSampler is built with the correct attributes
+    // Should not be exposed to public as the public build method should return the sampler wrapped inside
+    // ParentBasedSampler.
+    internal AWSXRayRemoteSampler BuildXraySampler()
     {
         return new AWSXRayRemoteSampler(this.resource, this.pollingInterval, this.endpoint, this.clock);
     }

src/OpenTelemetry.Sampler.AWS/CHANGELOG.md

@@ -17,6 +17,8 @@
 * Updated OpenTelemetry core component version(s) to `1.10.0`.
   ([#2317](https://github.com/open-telemetry/opentelemetry-dotnet-contrib/pull/2317))
 
+* `AWSXRayRemoteSamplerBuilder.Build()` now returns `ParentBasedSampler` which is of type `Trace.Sampler` instead of `AWSXRayRemoteSampler`
+
 ## 0.1.0-alpha.2
 
 Released 2024-Sep-09

src/OpenTelemetry.Sampler.AWS/FallbackSampler.cs

@@ -12,8 +12,8 @@ internal class FallbackSampler : Trace.Sampler
 
     public FallbackSampler(Clock clock)
     {
-        this.reservoirSampler = new ParentBasedSampler(new RateLimitingSampler(1, clock));
-        this.fixedRateSampler = new ParentBasedSampler(new TraceIdRatioBasedSampler(0.05));
+        this.reservoirSampler = new RateLimitingSampler(1, clock);
+        this.fixedRateSampler = new TraceIdRatioBasedSampler(0.05);
     }
 
     public override SamplingResult ShouldSample(in SamplingParameters samplingParameters)

src/OpenTelemetry.Sampler.AWS/SamplingRuleApplier.cs

@@ -20,7 +20,7 @@ public SamplingRuleApplier(string clientId, Clock clock, SamplingRule rule, Stat
         {
             // Until calling GetSamplingTargets, the default is to borrow 1/s if reservoir size is
             // positive.
-            this.ReservoirSampler = new ParentBasedSampler(new RateLimitingSampler(1, this.Clock));
+            this.ReservoirSampler = new RateLimitingSampler(1, this.Clock);
             this.Borrowing = true;
         }
         else
@@ -30,7 +30,7 @@ public SamplingRuleApplier(string clientId, Clock clock, SamplingRule rule, Stat
             this.Borrowing = false;
         }
 
-        this.FixedRateSampler = new ParentBasedSampler(new TraceIdRatioBasedSampler(rule.FixedRate));
+        this.FixedRateSampler = new TraceIdRatioBasedSampler(rule.FixedRate);
 
         // We either have no reservoir sampling or borrow until we get a quota so have no end time.
         this.ReservoirEndTime = DateTimeOffset.MaxValue;
@@ -191,15 +191,15 @@ public SamplingStatisticsDocument Snapshot(DateTimeOffset now)
     public SamplingRuleApplier WithTarget(SamplingTargetDocument target, DateTimeOffset now)
     {
         var newFixedRateSampler = target.FixedRate != null
-            ? new ParentBasedSampler(new TraceIdRatioBasedSampler(target.FixedRate.Value))
+            ? new TraceIdRatioBasedSampler(target.FixedRate.Value)
             : this.FixedRateSampler;
 
         Trace.Sampler newReservoirSampler = new AlwaysOffSampler();
         var newReservoirEndTime = DateTimeOffset.MaxValue;
         if (target.ReservoirQuota != null && target.ReservoirQuotaTTL != null)
         {
             newReservoirSampler = target.ReservoirQuota > 0
-                ? new ParentBasedSampler(new RateLimitingSampler(target.ReservoirQuota.Value, this.Clock))
+                ? new RateLimitingSampler(target.ReservoirQuota.Value, this.Clock)
                 : new AlwaysOffSampler();
 
             newReservoirEndTime = this.Clock.ToDateTime(target.ReservoirQuotaTTL.Value);

test/OpenTelemetry.Sampler.AWS.Tests/TestAWSXRayRemoteSampler.cs

@@ -22,7 +22,7 @@ public void TestSamplerWithConfiguration()
         var sampler = AWSXRayRemoteSampler.Builder(ResourceBuilder.CreateEmpty().Build())
             .SetPollingInterval(pollingInterval)
             .SetEndpoint(endpoint)
-            .Build();
+            .BuildXraySampler();
 
         Assert.Equal(pollingInterval, sampler.PollingInterval);
         Assert.Equal(endpoint, sampler.Endpoint);
@@ -33,7 +33,7 @@ public void TestSamplerWithConfiguration()
     [Fact]
     public void TestSamplerWithDefaults()
     {
-        var sampler = AWSXRayRemoteSampler.Builder(ResourceBuilder.CreateEmpty().Build()).Build();
+        var sampler = AWSXRayRemoteSampler.Builder(ResourceBuilder.CreateEmpty().Build()).BuildXraySampler();
 
         Assert.Equal(TimeSpan.FromMinutes(5), sampler.PollingInterval);
         Assert.Equal("http://localhost:2000", sampler.Endpoint);
@@ -53,7 +53,7 @@ public void TestSamplerUpdateAndSample()
             .SetPollingInterval(TimeSpan.FromMilliseconds(10))
             .SetEndpoint(mockServer.Url!)
             .SetClock(clock)
-            .Build();
+            .BuildXraySampler();
 
         // the sampler will use fallback sampler until rules are fetched.
         Assert.Equal(SamplingDecision.RecordAndSample, this.DoSample(sampler, "cat-service"));