Performance: Fix OM hierarchy to use strings for relative paths instead of URI

Microsoft.Azure.Cosmos/src/CosmosClient.cs

@@ -93,7 +93,7 @@ namespace Microsoft.Azure.Cosmos
     /// <seealso href="https://docs.microsoft.com/azure/cosmos-db/request-units">Request Units</seealso>
     public class CosmosClient : IDisposable
     {
-        private readonly Uri DatabaseRootUri = new Uri(Paths.Databases_Root, UriKind.Relative);
+        private readonly string DatabaseRootUri = Paths.Databases_Root;
         private ConsistencyLevel? accountConsistencyLevel;
         private bool isDisposed = false;
 

Microsoft.Azure.Cosmos/src/Diagnostics/PointOperationStatistics.cs

@@ -19,7 +19,7 @@ public PointOperationStatistics(
             double requestCharge,
             string errorMessage,
             HttpMethod method,
-            Uri requestUri,
+            string requestUri,
             string requestSessionToken,
             string responseSessionToken)
         {
@@ -42,7 +42,7 @@ public PointOperationStatistics(
         public double RequestCharge { get; }
         public string ErrorMessage { get; }
         public HttpMethod Method { get; }
-        public Uri RequestUri { get; }
+        public string RequestUri { get; }
         public string RequestSessionToken { get; }
         public string ResponseSessionToken { get; }
 

Microsoft.Azure.Cosmos/src/Handler/RequestInvokerHandler.cs

@@ -61,7 +61,7 @@ public override async Task<ResponseMessage> SendAsync(
         }
 
         public virtual async Task<T> SendAsync<T>(
-            Uri resourceUri,
+            string resourceUri,
             ResourceType resourceType,
             OperationType operationType,
             RequestOptions requestOptions,
@@ -79,7 +79,7 @@ public virtual async Task<T> SendAsync<T>(
             }
 
             ResponseMessage responseMessage = await this.SendAsync(
-                resourceUri: resourceUri,
+                resourceUriString: resourceUri,
                 resourceType: resourceType,
                 operationType: operationType,
                 requestOptions: requestOptions,
@@ -94,7 +94,7 @@ public virtual async Task<T> SendAsync<T>(
         }
 
         public virtual async Task<ResponseMessage> SendAsync(
-            Uri resourceUri,
+            string resourceUriString,
             ResourceType resourceType,
             OperationType operationType,
             RequestOptions requestOptions,
@@ -105,9 +105,9 @@ public virtual async Task<ResponseMessage> SendAsync(
             CosmosDiagnosticsContext diagnosticsContext,
             CancellationToken cancellationToken)
         {
-            if (resourceUri == null)
+            if (resourceUriString == null)
             {
-                throw new ArgumentNullException(nameof(resourceUri));
+                throw new ArgumentNullException(nameof(resourceUriString));
             }
 
             // DEVNOTE: Non-Item operations need to be refactored to always pass
@@ -124,7 +124,7 @@ public virtual async Task<ResponseMessage> SendAsync(
                 HttpMethod method = RequestInvokerHandler.GetHttpMethod(operationType);
                 RequestMessage request = new RequestMessage(
                         method,
-                        resourceUri,
+                        resourceUriString,
                         diagnosticsContext)
                 {
                     OperationType = operationType,

Microsoft.Azure.Cosmos/src/Handler/RequestMessage.cs

@@ -42,23 +42,24 @@ public RequestMessage()
         public RequestMessage(HttpMethod method, Uri requestUri)
         {
             this.Method = method;
-            this.RequestUri = requestUri;
+            this.RequestUriString = requestUri?.OriginalString;
+            this.InternalRequestUri = requestUri;
             this.DiagnosticsContext = new CosmosDiagnosticsContextCore();
         }
 
         /// <summary>
         /// Create a <see cref="RequestMessage"/>
         /// </summary>
         /// <param name="method">The http method</param>
-        /// <param name="requestUri">The requested URI</param>
+        /// <param name="requestUriString">The requested URI</param>
         /// <param name="diagnosticsContext">The diagnostics object used to track the request</param>
         internal RequestMessage(
             HttpMethod method,
-            Uri requestUri,
+            string requestUriString,
             CosmosDiagnosticsContext diagnosticsContext)
         {
             this.Method = method;
-            this.RequestUri = requestUri;
+            this.RequestUriString = requestUriString;
             this.DiagnosticsContext = diagnosticsContext ?? throw new ArgumentNullException(nameof(diagnosticsContext));
         }
 
@@ -70,7 +71,18 @@ internal RequestMessage(
         /// <summary>
         /// Gets the <see cref="Uri"/> for the current request.
         /// </summary>
-        public virtual Uri RequestUri { get; private set; }
+        public virtual Uri RequestUri
+        {
+            get
+            {
+                if (this.InternalRequestUri == null)
+                {
+                    this.InternalRequestUri = new Uri(this.RequestUriString, UriKind.Relative);
+                }
+
+                return this.InternalRequestUri;
+            }
+        }
 
         /// <summary>
         /// Gets the current <see cref="RequestMessage"/> HTTP headers.
@@ -90,6 +102,10 @@ public virtual Stream Content
             }
         }
 
+        internal string RequestUriString { get; }
+
+        internal Uri InternalRequestUri { get; private set; }
+
         internal CosmosDiagnosticsContext DiagnosticsContext { get; }
 
         internal RequestOptions RequestOptions { get; set; }
@@ -237,7 +253,7 @@ internal DocumentServiceRequest ToDocumentServiceRequest()
                 }
                 else
                 {
-                    serviceRequest = new DocumentServiceRequest(this.OperationType, this.ResourceType, this.RequestUri?.ToString(), this.Content, AuthorizationTokenType.PrimaryMasterKey, this.Headers.CosmosMessageHeaders);
+                    serviceRequest = new DocumentServiceRequest(this.OperationType, this.ResourceType, this.RequestUriString, this.Content, AuthorizationTokenType.PrimaryMasterKey, this.Headers.CosmosMessageHeaders);
                 }
 
                 if (this.UseGatewayMode.HasValue)

Microsoft.Azure.Cosmos/src/Handler/ResponseMessage.cs

@@ -163,7 +163,7 @@ public void Dispose()
 
         internal string GetResourceAddress()
         {
-            string resourceLink = this.RequestMessage?.RequestUri.OriginalString;
+            string resourceLink = this.RequestMessage?.RequestUriString;
             if (PathsHelper.TryParsePathSegments(
                 resourceLink,
                 out bool isFeed,

Microsoft.Azure.Cosmos/src/Query/Core/ExecutionContext/CosmosQueryExecutionContextFactory.cs

@@ -175,7 +175,7 @@ private static async Task<TryCatch<CosmosQueryExecutionContext>> TryCreateCoreCo
                                 // Only thing that matters is that we target the correct range.
                                 Documents.PartitionKeyDefinition partitionKeyDefinition = GetPartitionKeyDefinition(inputParameters, containerQueryProperties);
                                 List<Documents.PartitionKeyRange> targetRanges = await cosmosQueryContext.QueryClient.GetTargetPartitionKeyRangesByEpkStringAsync(
-                                    cosmosQueryContext.ResourceLink.OriginalString,
+                                    cosmosQueryContext.ResourceLink,
                                     containerQueryProperties.ResourceId,
                                     inputParameters.PartitionKey.Value.InternalKey.GetEffectivePartitionKeyString(partitionKeyDefinition));
 
@@ -237,7 +237,7 @@ public static async Task<TryCatch<CosmosQueryExecutionContext>> TryCreateFromPar
 
             List<Documents.PartitionKeyRange> targetRanges = await CosmosQueryExecutionContextFactory.GetTargetPartitionKeyRangesAsync(
                    cosmosQueryContext.QueryClient,
-                   cosmosQueryContext.ResourceLink.OriginalString,
+                   cosmosQueryContext.ResourceLink,
                    partitionedQueryExecutionInfo,
                    containerQueryProperties,
                    inputParameters.Properties,

Microsoft.Azure.Cosmos/src/Query/Core/ExecutionContext/CosmosQueryExecutionContextWithNameCacheStaleRetry.cs

@@ -50,7 +50,7 @@ public override async Task<QueryResponseCore> ExecuteNextAsync(CancellationToken
                 !this.alreadyRetried)
             {
                 await this.cosmosQueryContext.QueryClient.ForceRefreshCollectionCacheAsync(
-                        this.cosmosQueryContext.ResourceLink.OriginalString,
+                        this.cosmosQueryContext.ResourceLink,
                         cancellationToken);
                 this.alreadyRetried = true;
                 this.currentCosmosQueryExecutionContext.Dispose();

Microsoft.Azure.Cosmos/src/Query/Core/QueryClient/CosmosQueryClient.cs

@@ -18,7 +18,7 @@ internal abstract class CosmosQueryClient
         public abstract Action<IQueryable> OnExecuteScalarQueryCallback { get; }
 
         public abstract Task<ContainerQueryProperties> GetCachedContainerQueryPropertiesAsync(
-            Uri containerLink,
+            string containerLink,
             PartitionKey? partitionKey,
             CancellationToken cancellationToken);
 
@@ -44,7 +44,7 @@ public abstract Task<TryCatch<PartitionedQueryExecutionInfo>> TryGetPartitionedQ
             CancellationToken cancellationToken);
 
         public abstract Task<QueryResponseCore> ExecuteItemQueryAsync(
-            Uri resourceUri,
+            string resourceUri,
             Documents.ResourceType resourceType,
             Documents.OperationType operationType,
             Guid clientQueryCorrelationId,
@@ -58,7 +58,7 @@ public abstract Task<QueryResponseCore> ExecuteItemQueryAsync(
             CancellationToken cancellationToken);
 
         public abstract Task<PartitionedQueryExecutionInfo> ExecuteQueryPlanRequestAsync(
-            Uri resourceUri,
+            string resourceUri,
             Documents.ResourceType resourceType,
             Documents.OperationType operationType,
             SqlQuerySpec sqlQuerySpec,

Microsoft.Azure.Cosmos/src/Query/Core/QueryClient/CosmosQueryContext.cs

@@ -21,7 +21,7 @@ internal abstract class CosmosQueryContext
         public virtual Type ResourceType { get; }
         public virtual bool IsContinuationExpected { get; }
         public virtual bool AllowNonValueAggregateQuery { get; }
-        public virtual Uri ResourceLink { get; }
+        public virtual string ResourceLink { get; }
         public virtual string ContainerResourceId { get; set; }
         public virtual Guid CorrelatedActivityId { get; }
 
@@ -34,7 +34,7 @@ public CosmosQueryContext(
             ResourceType resourceTypeEnum,
             OperationType operationType,
             Type resourceType,
-            Uri resourceLink,
+            string resourceLink,
             Guid correlatedActivityId,
             bool isContinuationExpected,
             bool allowNonValueAggregateQuery,
@@ -62,7 +62,7 @@ internal abstract Task<QueryResponseCore> ExecuteQueryAsync(
             CancellationToken cancellationToken);
 
         internal abstract Task<PartitionedQueryExecutionInfo> ExecuteQueryPlanRequestAsync(
-            Uri resourceUri,
+            string resourceUri,
             Documents.ResourceType resourceType,
             Documents.OperationType operationType,
             SqlQuerySpec sqlQuerySpec,

Microsoft.Azure.Cosmos/src/Query/Core/QueryPlan/QueryPlanRetriever.cs

@@ -79,7 +79,7 @@ public static async Task<PartitionedQueryExecutionInfo> GetQueryPlanWithServiceI
         public static Task<PartitionedQueryExecutionInfo> GetQueryPlanThroughGatewayAsync(
             CosmosQueryContext queryContext,
             SqlQuerySpec sqlQuerySpec,
-            Uri resourceLink,
+            string resourceLink,
             PartitionKey? partitionKey,
             CancellationToken cancellationToken = default)
         {

Microsoft.Azure.Cosmos/src/Query/v3Query/CosmosQueryClientCore.cs

@@ -48,12 +48,12 @@ public CosmosQueryClientCore(
         public override Action<IQueryable> OnExecuteScalarQueryCallback => this.documentClient.OnExecuteScalarQueryCallback;
 
         public override async Task<ContainerQueryProperties> GetCachedContainerQueryPropertiesAsync(
-            Uri containerLink,
+            string containerLink,
             PartitionKey? partitionKey,
             CancellationToken cancellationToken)
         {
             ContainerProperties containerProperties = await this.clientContext.GetCachedContainerPropertiesAsync(
-                containerLink.OriginalString,
+                containerLink,
                 cancellationToken);
 
             string effectivePartitionKeyString = null;
@@ -117,7 +117,7 @@ public override async Task<TryCatch<PartitionedQueryExecutionInfo>> TryGetPartit
         }
 
         public override async Task<QueryResponseCore> ExecuteItemQueryAsync(
-            Uri resourceUri,
+            string resourceUri,
             ResourceType resourceType,
             OperationType operationType,
             Guid clientQueryCorrelationId,
@@ -165,7 +165,7 @@ public override async Task<QueryResponseCore> ExecuteItemQueryAsync(
         }
 
         public override async Task<PartitionedQueryExecutionInfo> ExecuteQueryPlanRequestAsync(
-            Uri resourceUri,
+            string resourceUri,
             ResourceType resourceType,
             OperationType operationType,
             SqlQuerySpec sqlQuerySpec,

Microsoft.Azure.Cosmos/src/Query/v3Query/CosmosQueryContextCore.cs

@@ -25,7 +25,7 @@ public CosmosQueryContextCore(
             ResourceType resourceTypeEnum,
             OperationType operationType,
             Type resourceType,
-            Uri resourceLink,
+            string resourceLink,
             Guid correlatedActivityId,
             bool isContinuationExpected,
             bool allowNonValueAggregateQuery,
@@ -92,7 +92,7 @@ internal override Task<QueryResponseCore> ExecuteQueryAsync(
         }
 
         internal override Task<PartitionedQueryExecutionInfo> ExecuteQueryPlanRequestAsync(
-            Uri resourceUri,
+            string resourceUri,
             Documents.ResourceType resourceType,
             Documents.OperationType operationType,
             SqlQuerySpec sqlQuerySpec,

Microsoft.Azure.Cosmos/src/Query/v3Query/QueryIterator.cs

@@ -46,7 +46,7 @@ public static QueryIterator Create(
             string continuationToken,
             FeedRangeInternal feedRangeInternal,
             QueryRequestOptions queryRequestOptions,
-            Uri resourceLink,
+            string resourceLink,
             bool isContinuationExpected,
             bool allowNonValueAggregateQuery,
             bool forcePassthrough,

Microsoft.Azure.Cosmos/src/Resource/ClientContextCore.cs

@@ -151,7 +151,7 @@ internal static CosmosClientContext Create(
         /// <param name="uriPathSegment">The URI path segment</param>
         /// <param name="id">The id of the resource</param>
         /// <returns>A resource link in the format of {parentLink}/this.UriPathSegment/this.Name with this.Name being a Uri escaped version</returns>
-        internal override Uri CreateLink(
+        internal override string CreateLink(
             string parentLink,
             string uriPathSegment,
             string id)
@@ -160,6 +160,8 @@ internal override Uri CreateLink(
             int parentLinkLength = parentLink?.Length ?? 0;
             string idUriEscaped = Uri.EscapeUriString(id);
 
+            Debug.Assert(parentLinkLength == 0 || !parentLink.EndsWith("/"));
+
             StringBuilder stringBuilder = new StringBuilder(parentLinkLength + 2 + uriPathSegment.Length + idUriEscaped.Length);
             if (parentLinkLength > 0)
             {
@@ -170,7 +172,7 @@ internal override Uri CreateLink(
             stringBuilder.Append(uriPathSegment);
             stringBuilder.Append("/");
             stringBuilder.Append(idUriEscaped);
-            return new Uri(stringBuilder.ToString(), UriKind.Relative);
+            return stringBuilder.ToString();
         }
 
         internal override void ValidateResource(string resourceId)
@@ -211,7 +213,7 @@ internal override CosmosDiagnosticsContext CreateDiagnosticContext(
         }
 
         internal override Task<ResponseMessage> ProcessResourceOperationStreamAsync(
-            Uri resourceUri,
+            string resourceUri,
             ResourceType resourceType,
             OperationType operationType,
             RequestOptions requestOptions,
@@ -261,7 +263,7 @@ internal override Task<ResponseMessage> ProcessResourceOperationStreamAsync(
         }
 
         internal override Task<ResponseMessage> ProcessResourceOperationStreamAsync(
-            Uri resourceUri,
+            string resourceUri,
             ResourceType resourceType,
             OperationType operationType,
             RequestOptions requestOptions,
@@ -274,7 +276,7 @@ internal override Task<ResponseMessage> ProcessResourceOperationStreamAsync(
         {
             this.ThrowIfDisposed();
             return this.RequestHandler.SendAsync(
-                resourceUri: resourceUri,
+                resourceUriString: resourceUri,
                 resourceType: resourceType,
                 operationType: operationType,
                 requestOptions: requestOptions,
@@ -287,7 +289,7 @@ internal override Task<ResponseMessage> ProcessResourceOperationStreamAsync(
         }
 
         internal override Task<T> ProcessResourceOperationAsync<T>(
-            Uri resourceUri,
+            string resourceUri,
             ResourceType resourceType,
             OperationType operationType,
             RequestOptions requestOptions,