fix: tracing subsegment when using fluent api Tracing.WithSubsegment

Author: hjgraca

examples/Tracing/src/HelloWorld/Function.cs

@@ -91,6 +91,14 @@ public async Task<APIGatewayProxyResponse> FunctionHandler(APIGatewayProxyReques
         gatewaySegment.AddAnnotation("AccountId", apigwProxyEvent.RequestContext.AccountId);
         gatewaySegment.AddMetadata("LookupRecord", lookupRecord);
 
+        Tracing.WithSubsegment("LoggingResponse",
+            subsegment =>
+            {
+                subsegment.AddAnnotation("AccountId", apigwProxyEvent.RequestContext.AccountId);
+                subsegment.AddMetadata("LookupRecord", lookupRecord);
+            });
+
+        
         try
         {
             await SaveRecordInDynamo(lookupRecord);

libraries/src/AWS.Lambda.Powertools.Tracing/Tracing.cs

@@ -154,8 +154,20 @@ public static void WithSubsegment(string nameSpace, string name, Action<TracingS
         XRayRecorder.Instance.SetNamespace(GetNamespaceOrDefault(nameSpace));
         try
         {
-            var entity = XRayRecorder.Instance.GetEntity() as TracingSubsegment;
-            subsegment?.Invoke(entity);
+            var entity = XRayRecorder.Instance.GetEntity();
+            var tracingSubsegment = new TracingSubsegment("## " + name);
+            
+            // Copy properties from the current entity
+            if (entity != null)
+            {
+                if (entity is Subsegment subsegmentEntity)
+                {
+                    tracingSubsegment.Namespace = subsegmentEntity.Namespace;
+                }
+                tracingSubsegment.Sampled = entity.Sampled;
+            }
+            
+            subsegment?.Invoke(tracingSubsegment);
         }
         finally
         {