use apply distributed headers

kgorro · kgorro · commit 85741384ce5b · 2021-09-07T23:24:46.000+08:00
diff --git a/Src/StackifyLib/ProfileTracer.cs b/Src/StackifyLib/ProfileTracer.cs
@@ -217,15 +217,12 @@ public ProfileTracer IgnoreChildFrames(bool value = true)
         }
 
         [MethodImpl(MethodImplOptions.PreserveSig | MethodImplOptions.NoInlining | MethodImplOptions.NoOptimization)]
-        public void StartDistributedTracing()
+        public void ApplyDistributedHeaders(string traceparent,string tracestate)
         {
-            
-        }
-
-        [MethodImpl(MethodImplOptions.PreserveSig | MethodImplOptions.NoInlining | MethodImplOptions.NoOptimization)]
-        public void EndDistributedTracing()
-        {
-
+            DistributedHeader.Remove("traceparent");
+            DistributedHeader.Add("traceparent", traceparent);
+            DistributedHeader.Remove("tracestate");
+            DistributedHeader.Add("tracestate", tracestate);
         }
 
         //Method the profiler looks for

Original file line number	Diff line number	Diff line change
`@@ -217,15 +217,12 @@ public ProfileTracer IgnoreChildFrames(bool value = true)`
`217`	`217`	`}`
`218`	`218`
`219`	`219`	`[MethodImpl(MethodImplOptions.PreserveSig \| MethodImplOptions.NoInlining \| MethodImplOptions.NoOptimization)]`
`220`		`- public void StartDistributedTracing()`
	`220`	`+ public void ApplyDistributedHeaders(string traceparent,string tracestate)`
`221`	`221`	`{`
`222`		`-`
`223`		`- }`
`224`		`-`
`225`		`- [MethodImpl(MethodImplOptions.PreserveSig \| MethodImplOptions.NoInlining \| MethodImplOptions.NoOptimization)]`
`226`		`- public void EndDistributedTracing()`
`227`		`- {`
`228`		`-`
	`222`	`+ DistributedHeader.Remove("traceparent");`
	`223`	`+ DistributedHeader.Add("traceparent", traceparent);`
	`224`	`+ DistributedHeader.Remove("tracestate");`
	`225`	`+ DistributedHeader.Add("tracestate", tracestate);`
`229`	`226`	`}`
`230`	`227`
`231`	`228`	`//Method the profiler looks for`