private async Task <ResponseMessage> ProcessResourceOperationAsBulkStreamAsync(
OperationType operationType,
RequestOptions requestOptions,
ContainerInternal cosmosContainerCore,
PartitionKey partitionKey,
string itemId,
Stream streamPayload,
CancellationToken cancellationToken)
{
this.ThrowIfDisposed();
ItemRequestOptions itemRequestOptions = requestOptions as ItemRequestOptions;
TransactionalBatchItemRequestOptions batchItemRequestOptions = TransactionalBatchItemRequestOptions.FromItemRequestOptions(itemRequestOptions);
ItemBatchOperation itemBatchOperation = new ItemBatchOperation(
operationType: operationType,
operationIndex: 0,
partitionKey: partitionKey,
id: itemId,
resourceStream: streamPayload,
requestOptions: batchItemRequestOptions,
cosmosClientContext: this);
TransactionalBatchOperationResult batchOperationResult = await cosmosContainerCore.BatchExecutor.AddAsync(
itemBatchOperation,
itemRequestOptions,
cancellationToken);
return(batchOperationResult.ToResponseMessage());
}
private async Task <ResponseMessage> ProcessResourceOperationAsBulkStreamAsync(
Uri resourceUri,
ResourceType resourceType,
OperationType operationType,
RequestOptions requestOptions,
ContainerCore cosmosContainerCore,
PartitionKey partitionKey,
string itemId,
Stream streamPayload,
CosmosDiagnosticsContext diagnosticsContext,
CancellationToken cancellationToken)
{
ItemRequestOptions itemRequestOptions = requestOptions as ItemRequestOptions;
TransactionalBatchItemRequestOptions batchItemRequestOptions = TransactionalBatchItemRequestOptions.FromItemRequestOptions(itemRequestOptions);
ItemBatchOperation itemBatchOperation = new ItemBatchOperation(
operationType: operationType,
operationIndex: 0,
partitionKey: partitionKey,
id: itemId,
resourceStream: streamPayload,
requestOptions: batchItemRequestOptions,
diagnosticsContext: diagnosticsContext);
TransactionalBatchOperationResult batchOperationResult = await cosmosContainerCore.BatchExecutor.AddAsync(itemBatchOperation, itemRequestOptions, cancellationToken);
return(batchOperationResult.ToResponseMessage());
}
public void ToResponseMessageHasPropertiesMapped()
{
TransactionalBatchOperationResult result = CreateTestResult();
ResponseMessage response = result.ToResponseMessage();
Assert.AreEqual(result.StatusCode, response.StatusCode);
Assert.AreEqual(result.SubStatusCode, response.Headers.SubStatusCode);
Assert.AreEqual(result.ETag, response.Headers.ETag);
Assert.AreEqual(result.RequestCharge, response.Headers.RequestCharge);
Assert.AreEqual(result.RetryAfter, response.Headers.RetryAfter);
Assert.AreSame(result.ResourceStream, response.Content);
Assert.AreSame(result.Diagnostics, response.Diagnostics);
}
/// <summary>
/// Based on the Retry Policy, if a failed response should retry.
/// </summary>
public Task <ShouldRetryResult> ShouldRetryAsync(
TransactionalBatchOperationResult batchOperationResult,
CancellationToken cancellationToken)
{
if (this.retryPolicy == null ||
batchOperationResult.IsSuccessStatusCode)
{
return(Task.FromResult(ShouldRetryResult.NoRetry()));
}
ResponseMessage responseMessage = batchOperationResult.ToResponseMessage();
return(this.retryPolicy.ShouldRetryAsync(responseMessage, cancellationToken));
}
private async Task <ResponseMessage> ProcessResourceOperationAsBulkStreamAsync(
Uri resourceUri,
ResourceType resourceType,
OperationType operationType,
RequestOptions requestOptions,
ContainerCore cosmosContainerCore,
PartitionKey partitionKey,
string itemId,
Stream streamPayload,
Action <RequestMessage> requestEnricher,
CancellationToken cancellationToken)
{
ItemRequestOptions itemRequestOptions = requestOptions as ItemRequestOptions;
TransactionalBatchItemRequestOptions batchItemRequestOptions = TransactionalBatchItemRequestOptions.FromItemRequestOptions(itemRequestOptions);
ItemBatchOperation itemBatchOperation = new ItemBatchOperation(operationType, /* index */ 0, partitionKey, itemId, streamPayload, batchItemRequestOptions);
TransactionalBatchOperationResult batchOperationResult = await cosmosContainerCore.BatchExecutor.AddAsync(itemBatchOperation, itemRequestOptions, cancellationToken);
return(batchOperationResult.ToResponseMessage());
}
/// <summary>
/// Based on the Retry Policy, if a failed response should retry.
/// </summary>
public async Task <ShouldRetryResult> ShouldRetryAsync(
TransactionalBatchOperationResult batchOperationResult,
CancellationToken cancellationToken)
{
if (this.retryPolicy == null ||
batchOperationResult.IsSuccessStatusCode)
{
return(ShouldRetryResult.NoRetry());
}
ResponseMessage responseMessage = batchOperationResult.ToResponseMessage();
ShouldRetryResult shouldRetry = await this.retryPolicy.ShouldRetryAsync(responseMessage, cancellationToken);
if (shouldRetry.ShouldRetry)
{
this.initialTrace.AddChild(batchOperationResult.Trace);
}
return(shouldRetry);
}
