public async Task <T> Update <T>(IAsyncTableStorageRepository <T> table, string paritionKey, string rowKey, Action <T> update) where T : ITableEntity, new()
{
_logger?.LogTrace("TableStorageConcurrencyManager: Update - commencing concurrency checked update operation");
bool concurrencyException = true;
T entity = default(T);
while (concurrencyException)
{
entity = rowKey == null ? (await table.GetAsync(paritionKey)).Single() : await table.GetAsync(paritionKey, rowKey);
update(entity);
try
{
await table.UpdateAsync(entity);
concurrencyException = false;
}
catch (StorageException storageException)
{
if (storageException.RequestInformation.HttpStatusCode != 412)
{
throw;
}
_logger?.LogTrace("TableStorageConcurrencyManager: Update - retrying concurrency checked update operation due to concurrency failure");
}
}
_logger?.LogTrace("TableStorageConcurrencyManager: Update - completed concurrent update");
return(entity);
}
public async Task <T> InsertOrUpdate <T>(IAsyncTableStorageRepository <T> table, string partitionKey, string rowKey, Action <T> update, Func <T> insert) where T : ITableEntity, new()
{
_logger?.LogTrace("TableStorageConcurrencyManager: InsertOrUpdate - commencing concurrency checked update operation");
bool concurrencyException = true;
T entity = default(T);
while (concurrencyException)
{
entity = rowKey == null ? (await table.GetAsync(partitionKey)).SingleOrDefault() : await table.GetAsync(partitionKey, rowKey);
if (entity == null)
{
try
{
entity = insert();
await table.InsertAsync(entity);
concurrencyException = false;
}
catch (UniqueKeyViolationException)
{
// someone has inserted in the meantime so we now need to update
_logger?.LogTrace("TableStorageConcurrencyManager: InsertOrUpdate - failed as insert has occurred in the interim, now attempting update");
}
}
else
{
update(entity);
try
{
await table.UpdateAsync(entity);
concurrencyException = false;
}
catch (StorageException storageException)
{
if (storageException.RequestInformation.HttpStatusCode != 412)
{
throw;
}
_logger?.LogTrace("TableStorageConcurrencyManager: InsertOrUpdate - retrying concurrency checked update operation due to concurrency failure");
}
}
}
return(entity);
}