Tracing: Adds ITrace as the default tracing implementation for CosmosDiagnostics

Microsoft.Azure.Cosmos/src/Authorization/AuthorizationTokenProvider.cs

@@ -6,9 +6,8 @@ namespace Microsoft.Azure.Cosmos
 {
     using System;
     using System.Globalization;
-    using System.Net.Http;
-    using System.Text;
     using System.Threading.Tasks;
+    using Microsoft.Azure.Cosmos.Tracing;
     using Microsoft.Azure.Documents;
     using Microsoft.Azure.Documents.Collections;
 
@@ -49,7 +48,7 @@ public abstract ValueTask<string> GetUserAuthorizationTokenAsync(
             string requestVerb,
             INameValueCollection headers,
             AuthorizationTokenType tokenType,
-            CosmosDiagnosticsContext diagnosticsContext);
+            ITrace trace);
 
         public abstract void TraceUnauthorized(
             DocumentClientException dce,

Microsoft.Azure.Cosmos/src/Authorization/AuthorizationTokenProviderMasterKey.cs

@@ -7,11 +7,11 @@ namespace Microsoft.Azure.Cosmos
     using System;
     using System.Globalization;
     using System.Net;
-    using System.Net.Http;
     using System.Security;
     using System.Text;
     using System.Threading.Tasks;
     using Microsoft.Azure.Cosmos.Core.Trace;
+    using Microsoft.Azure.Cosmos.Tracing;
     using Microsoft.Azure.Documents;
     using Microsoft.Azure.Documents.Collections;
 
@@ -94,7 +94,7 @@ public override ValueTask<string> GetUserAuthorizationTokenAsync(
             string requestVerb,
             INameValueCollection headers,
             AuthorizationTokenType tokenType,
-            CosmosDiagnosticsContext diagnosticsContext)
+            ITrace trace)
         {
             // this is masterkey authZ
             headers[HttpConstants.HttpHeaders.XDate] = DateTime.UtcNow.ToString("r", CultureInfo.InvariantCulture);

Microsoft.Azure.Cosmos/src/Authorization/AuthorizationTokenProviderResourceToken.cs

@@ -5,9 +5,9 @@
 namespace Microsoft.Azure.Cosmos
 {
     using System;
-    using System.Net.Http;
     using System.Threading.Tasks;
     using Microsoft.Azure.Cosmos.Core.Trace;
+    using Microsoft.Azure.Cosmos.Tracing;
     using Microsoft.Azure.Documents;
     using Microsoft.Azure.Documents.Collections;
 
@@ -44,7 +44,7 @@ public override ValueTask<string> GetUserAuthorizationTokenAsync(
             string requestVerb,
             INameValueCollection headers,
             AuthorizationTokenType tokenType,
-            CosmosDiagnosticsContext diagnosticsContext)
+            ITrace trace)
         {
             // If the input auth token is a resource token, then use it as a bearer-token.
             return this.urlEncodedAuthKeyResourceTokenValueTask;

Microsoft.Azure.Cosmos/src/Authorization/AuthorizationTokenProviderTokenCredential.cs

@@ -9,6 +9,7 @@ namespace Microsoft.Azure.Cosmos
     using System.Threading.Tasks;
     using global::Azure.Core;
     using Microsoft.Azure.Cosmos.Core.Trace;
+    using Microsoft.Azure.Cosmos.Tracing;
     using Microsoft.Azure.Documents;
     using Microsoft.Azure.Documents.Collections;
 
@@ -38,7 +39,7 @@ public AuthorizationTokenProviderTokenCredential(
             AuthorizationTokenType tokenType)
         {
             string token = AuthorizationTokenProviderTokenCredential.GenerateAadAuthorizationSignature(
-                    await this.tokenCredentialCache.GetTokenAsync(EmptyCosmosDiagnosticsContext.Singleton));
+                    await this.tokenCredentialCache.GetTokenAsync(NoOpTrace.Singleton));
             return (token, default);
         }
 
@@ -48,10 +49,10 @@ public override async ValueTask<string> GetUserAuthorizationTokenAsync(
             string requestVerb,
             INameValueCollection headers,
             AuthorizationTokenType tokenType,
-            CosmosDiagnosticsContext diagnosticsContext)
+            ITrace trace)
         {
             return AuthorizationTokenProviderTokenCredential.GenerateAadAuthorizationSignature(
-                    await this.tokenCredentialCache.GetTokenAsync(diagnosticsContext));
+                    await this.tokenCredentialCache.GetTokenAsync(trace));
         }
 
         public override async ValueTask AddAuthorizationHeaderAsync(
@@ -61,7 +62,7 @@ public override async ValueTask AddAuthorizationHeaderAsync(
             AuthorizationTokenType tokenType)
         {
             string token = AuthorizationTokenProviderTokenCredential.GenerateAadAuthorizationSignature(
-                    await this.tokenCredentialCache.GetTokenAsync(EmptyCosmosDiagnosticsContext.Singleton));
+                    await this.tokenCredentialCache.GetTokenAsync(NoOpTrace.Singleton));
 
             headersCollection.Add(HttpConstants.HttpHeaders.Authorization, token);
         }

Microsoft.Azure.Cosmos/src/Authorization/ICosmosAuthorizationTokenProvider.cs

@@ -5,6 +5,7 @@
 namespace Microsoft.Azure.Cosmos
 {
     using System.Threading.Tasks;
+    using Microsoft.Azure.Cosmos.Tracing;
     using Microsoft.Azure.Documents;
     using Microsoft.Azure.Documents.Collections;
 
@@ -23,6 +24,6 @@ ValueTask<string> GetUserAuthorizationTokenAsync(
             string requestVerb,
             INameValueCollection headers,
             AuthorizationTokenType tokenType,
-            CosmosDiagnosticsContext diagnosticsContext);
+            ITrace trace);
     }
 }