Revert "Revert baseline and use interface for tracedatum"

This reverts commit 3680d0f.

Microsoft.Azure.Cosmos/src/Tracing/ITrace.cs

@@ -7,7 +7,6 @@ namespace Microsoft.Azure.Cosmos.Tracing
     using System;
     using System.Collections.Generic;
     using System.Runtime.CompilerServices;
-    using Microsoft.Azure.Documents;
 
     /// <summary>
     /// Interface to represent a single node in a trace tree.
@@ -130,6 +129,6 @@ ITrace StartChild(
         /// Update region contacted information to the parent Itrace
         /// </summary>
         /// <param name="traceDatum"></param>
-        void UpdateRegionContacted(IClientSideRequestStatistics traceDatum);
+        void UpdateRegionContacted(TraceDatum traceDatum);
     }
 }

Microsoft.Azure.Cosmos/src/Tracing/NoOpTrace.cs

@@ -6,7 +6,6 @@ namespace Microsoft.Azure.Cosmos.Tracing
 {
     using System;
     using System.Collections.Generic;
-    using Microsoft.Azure.Documents;
 
     internal sealed class NoOpTrace : ITrace
     {
@@ -93,7 +92,7 @@ public void AddChild(ITrace trace)
             // NoOp
         }
 
-        public void UpdateRegionContacted(IClientSideRequestStatistics traceDatum)
+        public void UpdateRegionContacted(TraceDatum traceDatum)
         {
             // NoOp
         }

Microsoft.Azure.Cosmos/src/Tracing/Trace.cs

@@ -10,7 +10,6 @@ namespace Microsoft.Azure.Cosmos.Tracing
     using System.Linq;
     using System.Runtime.CompilerServices;
     using Microsoft.Azure.Cosmos.Tracing.TraceData;
-    using Microsoft.Azure.Documents;
 
     internal sealed class Trace : ITrace
     {
@@ -89,14 +88,17 @@ private Trace(
         /// <summary>
         /// Update region contacted information to the parent Itrace
         /// </summary>
-        /// <param name="clientSideRequestStatisticsTraceDatum"></param>
-        public void UpdateRegionContacted(IClientSideRequestStatistics clientSideRequestStatisticsTraceDatum)
+        /// <param name="traceDatum"></param>
+        public void UpdateRegionContacted(TraceDatum traceDatum)
         {
-            if (clientSideRequestStatisticsTraceDatum.RegionsContacted == null || clientSideRequestStatisticsTraceDatum.RegionsContacted.Count == 0)
+            if (traceDatum is ClientSideRequestStatisticsTraceDatum clientSideRequestStatisticsTraceDatum)
             {
-                return;
+                if (clientSideRequestStatisticsTraceDatum.RegionsContacted == null || clientSideRequestStatisticsTraceDatum.RegionsContacted.Count == 0)
+                {
+                    return;
+                }
+                this.RegionsContacted = clientSideRequestStatisticsTraceDatum.RegionsContacted?.ToList();
             }
-            this.RegionsContacted = clientSideRequestStatisticsTraceDatum.RegionsContacted?.ToList();
         }
 
         public void Dispose()
@@ -174,10 +176,7 @@ public static Trace GetRootTrace(
         public void AddDatum(string key, TraceDatum traceDatum)
         {
             this.data.Add(key, traceDatum);
-            if (traceDatum is ClientSideRequestStatisticsTraceDatum clientSideRequestStatisticsTraceDatum)
-            {
-                this.UpdateRegionContacted(clientSideRequestStatisticsTraceDatum);
-            }
+            this.UpdateRegionContacted(traceDatum);
         }
 
         public void AddDatum(string key, object value)