internal static async Task <DocumentServiceResponse> ParseResponseAsync(HttpResponseMessage responseMessage, JsonSerializerSettings serializerSettings = null, DocumentServiceRequest request = null)
{
using (responseMessage)
{
if ((int)responseMessage.StatusCode < 400)
{
INameValueCollection headers = GatewayStoreClient.ExtractResponseHeaders(responseMessage);
Stream contentStream = await GatewayStoreClient.BufferContentIfAvailableAsync(responseMessage);
return(new DocumentServiceResponse(contentStream, headers, responseMessage.StatusCode, serializerSettings));
}
else if (request != null &&
request.IsValidStatusCodeForExceptionlessRetry((int)responseMessage.StatusCode))
{
INameValueCollection headers = GatewayStoreClient.ExtractResponseHeaders(responseMessage);
Stream contentStream = await GatewayStoreClient.BufferContentIfAvailableAsync(responseMessage);
return(new DocumentServiceResponse(contentStream, headers, responseMessage.StatusCode, serializerSettings));
}
else
{
throw await GatewayStoreClient.CreateDocumentClientExceptionAsync(responseMessage);
}
}
}
public static async Task <DocumentServiceResponse> ParseResponseAsync(HttpResponseMessage responseMessage, JsonSerializerSettings serializerSettings = null, DocumentServiceRequest request = null)
{
using (responseMessage)
{
if ((int)responseMessage.StatusCode < 400)
{
MemoryStream bufferedStream = new MemoryStream();
await responseMessage.Content.CopyToAsync(bufferedStream);
bufferedStream.Position = 0;
INameValueCollection headers = ClientExtensions.ExtractResponseHeaders(responseMessage);
return(new DocumentServiceResponse(bufferedStream, headers, responseMessage.StatusCode, serializerSettings));
}
else if (request != null &&
request.IsValidStatusCodeForExceptionlessRetry((int)responseMessage.StatusCode))
{
INameValueCollection headers = ClientExtensions.ExtractResponseHeaders(responseMessage);
return(new DocumentServiceResponse(null, headers, responseMessage.StatusCode, serializerSettings));
}
else
{
throw await ClientExtensions.CreateDocumentClientException(responseMessage);
}
}
}
private async Task CaptureSessionTokenAndHandleSplitAsync(
HttpStatusCode?statusCode,
SubStatusCodes subStatusCode,
DocumentServiceRequest request,
INameValueCollection responseHeaders)
{
// Exceptionless can try to capture session token from CompleteResponse
if (request.IsValidStatusCodeForExceptionlessRetry((int)statusCode, subStatusCode))
{
// Not capturing on master resources
if (ReplicatedResourceClient.IsMasterResource(request.ResourceType))
{
return;
}
// Only capturing on 409, 412, 404 && !1002
if (statusCode != HttpStatusCode.PreconditionFailed &&
statusCode != HttpStatusCode.Conflict &&
(statusCode != HttpStatusCode.NotFound || subStatusCode == SubStatusCodes.ReadSessionNotAvailable))
{
return;
}
}
if (request.ResourceType == ResourceType.Collection && request.OperationType == OperationType.Delete)
{
string resourceId;
if (request.IsNameBased)
{
resourceId = responseHeaders[HttpConstants.HttpHeaders.OwnerId];
}
else
{
resourceId = request.ResourceId;
}
this.sessionContainer.ClearTokenByResourceId(resourceId);
}
else
{
this.sessionContainer.SetSessionToken(request, responseHeaders);
PartitionKeyRange detectedPartitionKeyRange = request.RequestContext.ResolvedPartitionKeyRange;
string partitionKeyRangeInResponse = responseHeaders[HttpConstants.HttpHeaders.PartitionKeyRangeId];
if (detectedPartitionKeyRange != null &&
!string.IsNullOrEmpty(partitionKeyRangeInResponse) &&
!string.IsNullOrEmpty(request.RequestContext.ResolvedCollectionRid) &&
!partitionKeyRangeInResponse.Equals(detectedPartitionKeyRange.Id, StringComparison.OrdinalIgnoreCase))
{
// The request ended up being on a different partition unknown to the client, so we better refresh the caches
await this.partitionKeyRangeCache.TryGetPartitionKeyRangeByIdAsync(
request.RequestContext.ResolvedCollectionRid,
partitionKeyRangeInResponse,
NoOpTrace.Singleton,
forceRefresh : true);
}
}
}
public void TestUseStatusCodeFor429()
{
using (DocumentServiceRequest request =
DocumentServiceRequest.Create(
OperationType.Query,
ResourceType.Document,
new Uri("https://foo.com/dbs/db1/colls/coll1", UriKind.Absolute),
new MemoryStream(Encoding.UTF8.GetBytes("content1")),
AuthorizationTokenType.PrimaryMasterKey,
null))
{
Assert.IsFalse(request.IsValidStatusCodeForExceptionlessRetry((int)StatusCodes.TooManyRequests));
request.UseStatusCodeFor429 = true;
Assert.IsTrue(request.IsValidStatusCodeForExceptionlessRetry((int)StatusCodes.TooManyRequests));
Assert.IsFalse(request.IsValidStatusCodeForExceptionlessRetry((int)HttpStatusCode.PreconditionFailed));
Assert.IsFalse(request.IsValidStatusCodeForExceptionlessRetry((int)HttpStatusCode.Conflict));
Assert.IsFalse(request.IsValidStatusCodeForExceptionlessRetry((int)HttpStatusCode.NotFound));
}
}
private void CaptureSessionToken(
HttpStatusCode?statusCode,
SubStatusCodes subStatusCode,
DocumentServiceRequest request,
INameValueCollection responseHeaders)
{
// Exceptionless can try to capture session token from CompleteResponse
if (request.IsValidStatusCodeForExceptionlessRetry((int)statusCode, subStatusCode))
{
// Not capturing on master resources
if (ReplicatedResourceClient.IsMasterResource(request.ResourceType))
{
return;
}
// Only capturing on 409, 412, 404 && !1002
if (statusCode != HttpStatusCode.PreconditionFailed &&
statusCode != HttpStatusCode.Conflict &&
(statusCode != HttpStatusCode.NotFound || subStatusCode == SubStatusCodes.ReadSessionNotAvailable))
{
return;
}
}
if (request.ResourceType == ResourceType.Collection && request.OperationType == OperationType.Delete)
{
string resourceId;
if (request.IsNameBased)
{
resourceId = responseHeaders[HttpConstants.HttpHeaders.OwnerId];
}
else
{
resourceId = request.ResourceId;
}
this.sessionContainer.ClearTokenByResourceId(resourceId);
}
else
{
this.sessionContainer.SetSessionToken(request, responseHeaders);
}
}