public static async Task WithContextAsync(this IDbContextFactory contextFactory, Func <IDbContext, Task> action, CancellationToken cancellationToken = default)
{
if (contextFactory == null)
{
throw new ArgumentNullException(nameof(contextFactory));
}
if (action == null)
{
throw new ArgumentNullException(nameof(action));
}
var retryPolicy = contextFactory.Configuration?.RetryPolicy;
if (retryPolicy != null)
{
await DbRetryer.RetryAsync(retryPolicy, PerformAction, cancellationToken);
}
else
{
await PerformAction();
}
async Task PerformAction()
{
using (var context = await contextFactory.OpenContextAsync(cancellationToken))
{
await action(context);
if (context.CommitState != DbContextCommitState.Rollback)
{
context.Commit();
}
}
}
}
public static void WithContext(this IDbContextFactory contextFactory, Action <IDbContext> action)
{
if (contextFactory == null)
{
throw new ArgumentNullException(nameof(contextFactory));
}
if (action == null)
{
throw new ArgumentNullException(nameof(action));
}
var retryPolicy = contextFactory.Configuration?.RetryPolicy;
if (retryPolicy != null)
{
DbRetryer.Retry(retryPolicy, PerformAction);
}
else
{
PerformAction();
}
void PerformAction()
{
using (var context = contextFactory.OpenContext())
{
action(context);
if (context.CommitState != DbContextCommitState.Rollback)
{
context.Commit();
}
}
}
}
public static T WithContext <T>(this IDbContextFactory contextFactory, Func <IDbContext, T> action)
{
if (contextFactory == null)
{
throw new ArgumentNullException(nameof(contextFactory));
}
if (action == null)
{
throw new ArgumentNullException(nameof(action));
}
var retryPolicy = contextFactory.Configuration?.RetryPolicy;
if (retryPolicy != null)
{
return(DbRetryer.Retry(retryPolicy, PerformAction));
}
else
{
return(PerformAction());
}
T PerformAction()
{
T result;
using (var context = contextFactory.OpenContext())
{
result = action(context);
if (context.CommitState != DbContextCommitState.Rollback)
{
context.Commit();
}
}
return(result);
}
}
