private async Task <TableResult> HandleInsertAsync(TableOperation operation, CloudTableClient client, CloudTable table, TableRequestOptions options, OperationContext context)
{
if (operation.IsTableEntity)
{
await client.DocumentClient.CreateDatabaseIfNotExistsAsync(new Database { Id = StellarConstants.TableDatabaseName });
string collectionName = ((DynamicTableEntity)operation.Entity).Properties[TableConstants.TableName].StringValue;
RequestOptions requestOptions = null;
string collectionRUConfig = ConfigurationManager.AppSettings["DocumentDbCollectionRU"];
int collectionRU = 0;
if (!string.IsNullOrEmpty(collectionRUConfig) && int.TryParse(collectionRUConfig, out collectionRU))
{
requestOptions = new RequestOptions {
OfferThroughput = collectionRU
};
}
ResourceResponse <DocumentCollection> response
= await client.DocumentClient.CreateDocumentCollectionAsync(
UriFactory.CreateDatabaseUri(StellarConstants.TableDatabaseName),
new DocumentCollection
{
Id = collectionName,
PartitionKey = new PartitionKeyDefinition()
{
Paths = { "/" + StellarConstants.PartitionKeyPropertyName }
},
}, requestOptions);
context.RequestResults.Add(response.ToRequestResult());
return(EntityHelpers.GetTableResultFromResponse(response, context));
}
else
{
Document document = EntityHelpers.GetDocumentFromEntity(operation.Entity, context, options);
ResourceResponse <Document> response = await client.DocumentClient.CreateDocumentAsync(table.GetCollectionUri(), document);
context.RequestResults.Add(response.ToRequestResult());
return(this.GetTableResultFromResponse(operation, response, context, options));
}
}
private async Task <TableResult> HandleDeleteAsync(TableOperation operation, CloudTableClient client, CloudTable table, TableRequestOptions options, OperationContext context)
{
if (operation.IsTableEntity)
{
string collectionName = ((DynamicTableEntity)operation.Entity).Properties[TableConstants.TableName].StringValue;
Uri documentCollectionUri = UriFactory.CreateDocumentCollectionUri(StellarConstants.TableDatabaseName, collectionName);
ResourceResponse <DocumentCollection> response = await client.DocumentClient.DeleteDocumentCollectionAsync(documentCollectionUri);
context.RequestResults.Add(response.ToRequestResult());
return(EntityHelpers.GetTableResultFromResponse(response, context));
}
else
{
RequestOptions requestOptions;
Uri documentUri = this.GetDocumentUri(operation, table, out requestOptions);
ResourceResponse <Document> response = await client.DocumentClient.DeleteDocumentAsync(documentUri, requestOptions);
context.RequestResults.Add(response.ToRequestResult());
return(this.GetTableResultFromResponse(operation, response, context, options));
}
}
private async Task <TableResult> HandleReadAsync(TableOperation operation, CloudTableClient client, CloudTable table, TableRequestOptions options, OperationContext context)
{
try
{
if (operation.IsTableEntity)
{
//TODO: Why do we get the collection name from the Entity instead of the table
string collectionName = ((DynamicTableEntity)operation.Entity).Properties[TableConstants.TableName].StringValue;
Uri documentCollectionUri = UriFactory.CreateDocumentCollectionUri(StellarConstants.TableDatabaseName, collectionName);
ResourceResponse <DocumentCollection> response = await client.DocumentClient.ReadDocumentCollectionAsync(documentCollectionUri);
context.RequestResults.Add(response.ToRequestResult());
return(EntityHelpers.GetTableResultFromResponse(response, context));
}
else
{
RequestOptions requestOptions;
Uri documentUri = this.GetDocumentUri(operation, table, out requestOptions);
ResourceResponse <Document> response = await client.DocumentClient.ReadDocumentAsync(documentUri, requestOptions);
context.RequestResults.Add(response.ToRequestResult());
return(this.GetTableResultFromResponse(operation, response, context, options));
}
}
catch (DocumentClientException exception)
{
if (exception.StatusCode == HttpStatusCode.NotFound)
{
return(new TableResult
{
HttpStatusCode = (int)HttpStatusCode.NotFound
});
}
throw;
}
}