public ICompensateActivityScopeContext <TActivity, TLog> GetScope(CompensateContext <TLog> context)
{
if (context.TryGetPayload <INestedContainer>(out var existingContainer))
{
existingContainer.Inject <ConsumeContext>(context);
var activity = existingContainer.GetInstance <TActivity>();
var activityContext = context.CreateActivityContext(activity);
return(new ExistingCompensateActivityScopeContext <TActivity, TLog>(activityContext));
}
var nestedContainer = _container.GetNestedContainer(context);
try
{
var activity = nestedContainer.GetInstance <TActivity>();
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
activityContext.UpdatePayload(nestedContainer);
return(new CreatedCompensateActivityScopeContext <INestedContainer, TActivity, TLog>(nestedContainer, activityContext));
}
catch
{
nestedContainer.Dispose();
throw;
}
}
public ICompensateActivityScopeContext <TActivity, TLog> GetScope(CompensateContext <TLog> context)
{
if (context.TryGetPayload <IUnityContainer>(out var existingContainer))
{
var activity = existingContainer.Resolve <TActivity>(new DependencyOverride(typeof(TLog), new InjectionParameter(context.Log)));
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
return(new ExistingCompensateActivityScopeContext <TActivity, TLog>(activityContext));
}
var scope = _container.CreateChildContainer();
try
{
var activity = scope.Resolve <TActivity>(new DependencyOverride(typeof(TLog), new InjectionParameter(context.Log)));
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
var contextScope = scope;
activityContext.GetOrAddPayload(() => contextScope);
return(new CreatedCompensateActivityScopeContext <IUnityContainer, TActivity, TLog>(contextScope, activityContext));
}
catch
{
scope.Dispose();
throw;
}
}
public ICompensateActivityScopeContext <TActivity, TLog> GetScope(CompensateContext <TLog> context)
{
if (context.TryGetPayload <IKernel>(out var kernel))
{
kernel.UpdateScope(context);
var activity = kernel.Resolve <TActivity>(new Arguments().AddTyped(context.Log));
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
return(new ExistingCompensateActivityScopeContext <TActivity, TLog>(activityContext, ReleaseComponent));
}
var scope = _kernel.CreateNewOrUseExistingMessageScope(context);
try
{
var activity = _kernel.Resolve <TActivity>(new Arguments().AddTyped(context.Log));
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
activityContext.UpdatePayload(_kernel);
return(new CreatedCompensateActivityScopeContext <IDisposable, TActivity, TLog>(scope, activityContext, ReleaseComponent));
}
catch
{
scope.Dispose();
throw;
}
}
public async Task Compensate(CompensateContext <TLog> context, IPipe <CompensateActivityContext <TActivity, TLog> > next)
{
TActivity activity = null;
try
{
activity = _compensateFactory(context.Log);
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
await next.Send(activityContext).ConfigureAwait(false);
}
finally
{
switch (activity)
{
case IAsyncDisposable asyncDisposable:
await asyncDisposable.DisposeAsync().ConfigureAwait(false);
break;
case IDisposable disposable:
disposable.Dispose();
break;
}
}
}
public ICompensateActivityScopeContext <TActivity, TLog> GetScope(CompensateContext <TLog> context)
{
if (context.TryGetPayload <IServiceScope>(out var existingServiceScope))
{
existingServiceScope.UpdateScope(context);
var activity = existingServiceScope.ServiceProvider.GetService <TActivity>();
if (activity == null)
{
throw new ConsumerException($"Unable to resolve activity type '{TypeMetadataCache<TActivity>.ShortName}'.");
}
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
return(new ExistingCompensateActivityScopeContext <TActivity, TLog>(activityContext));
}
if (!context.TryGetPayload(out IServiceProvider serviceProvider))
{
serviceProvider = _serviceProvider;
}
var serviceScope = serviceProvider.GetRequiredService <IServiceScopeFactory>().CreateScope();
try
{
serviceScope.UpdateScope(context);
var activity = serviceScope.ServiceProvider.GetService <TActivity>();
if (activity == null)
{
throw new ConsumerException($"Unable to resolve activity type '{TypeMetadataCache<TActivity>.ShortName}'.");
}
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
activityContext.UpdatePayload(serviceScope);
return(new CreatedCompensateActivityScopeContext <IServiceScope, TActivity, TLog>(serviceScope, activityContext));
}
catch
{
serviceScope.Dispose();
throw;
}
}
public async ValueTask <ICompensateActivityScopeContext <TActivity, TLog> > GetScope(CompensateContext <TLog> context)
{
if (context.TryGetPayload <IServiceScope>(out var existingServiceScope))
{
existingServiceScope.SetCurrentConsumeContext(context);
var activity = existingServiceScope.ServiceProvider.GetService <TActivity>();
if (activity == null)
{
throw new ConsumerException($"Unable to resolve activity type '{TypeMetadataCache<TActivity>.ShortName}'.");
}
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
return(new ExistingCompensateActivityScopeContext <TActivity, TLog>(activityContext));
}
if (!context.TryGetPayload(out IServiceProvider serviceProvider))
{
serviceProvider = _serviceProvider;
}
var serviceScope = serviceProvider.CreateScope();
try
{
CompensateContext <TLog> scopeContext = new CompensateContextScope <TLog>(context, serviceScope, serviceScope.ServiceProvider);
serviceScope.SetCurrentConsumeContext(scopeContext);
var activity = serviceScope.ServiceProvider.GetService <TActivity>();
if (activity == null)
{
throw new ConsumerException($"Unable to resolve activity type '{TypeMetadataCache<TActivity>.ShortName}'.");
}
CompensateActivityContext <TActivity, TLog> activityContext = scopeContext.CreateActivityContext(activity);
return(new CreatedCompensateActivityScopeContext <IServiceScope, TActivity, TLog>(serviceScope, activityContext));
}
catch
{
if (serviceScope is IAsyncDisposable asyncDisposable)
{
await asyncDisposable.DisposeAsync().ConfigureAwait(false);
}
else
{
serviceScope.Dispose();
}
throw;
}
}
public ICompensateActivityScopeContext <TActivity, TLog> GetScope(CompensateContext <TLog> context)
{
if (context.TryGetPayload <Scope>(out var existingScope))
{
existingScope.UpdateScope(context);
var activity = existingScope
.Container
.GetInstance <TActivity>();
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
return(new ExistingCompensateActivityScopeContext <TActivity, TLog>(activityContext));
}
var scope = AsyncScopedLifestyle.BeginScope(_container);
try
{
scope.UpdateScope(context);
var activity = scope
.Container
.GetInstance <TActivity>();
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
var contextScope = scope;
activityContext.UpdatePayload(scope);
return(new CreatedCompensateActivityScopeContext <Scope, TActivity, TLog>(contextScope, activityContext));
}
catch
{
scope.Dispose();
throw;
}
}
public ICompensateActivityScopeContext <TActivity, TLog> GetScope(CompensateContext <TLog> context)
{
if (context.TryGetPayload <ILifetimeScope>(out var existingLifetimeScope))
{
var activity = existingLifetimeScope.Resolve <TActivity>(TypedParameter.From(context.Log));
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
return(new ExistingCompensateActivityScopeContext <TActivity, TLog>(activityContext));
}
var parentLifetimeScope = _scopeProvider.GetLifetimeScope(context);
var lifetimeScope = parentLifetimeScope.BeginLifetimeScope(_name, builder =>
{
builder.ConfigureScope(context);
_configureScope?.Invoke(builder, context);
});
try
{
var activity = lifetimeScope.Resolve <TActivity>(TypedParameter.From(context.Log));
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
activityContext.UpdatePayload(lifetimeScope);
return(new CreatedCompensateActivityScopeContext <ILifetimeScope, TActivity, TLog>(lifetimeScope, activityContext));
}
catch
{
lifetimeScope.Dispose();
throw;
}
}
public async ValueTask <ICompensateActivityScopeContext <TActivity, TLog> > GetScope(CompensateContext <TLog> context)
{
if (context.TryGetPayload <Scope>(out var existingScope))
{
existingScope.UpdateScope(context);
var activity = existingScope
.Container
.GetInstance <TActivity>();
CompensateActivityContext <TActivity, TLog> activityContext = context.CreateActivityContext(activity);
return(new ExistingCompensateActivityScopeContext <TActivity, TLog>(activityContext));
}
var scope = AsyncScopedLifestyle.BeginScope(_container);
try
{
CompensateContext <TLog> scopeContext = new CompensateContextScope <TLog>(context, scope);
scope.UpdateScope(scopeContext);
var activity = scope.Container.GetInstance <TActivity>();
CompensateActivityContext <TActivity, TLog> activityContext = scopeContext.CreateActivityContext(activity);
return(new CreatedCompensateActivityScopeContext <Scope, TActivity, TLog>(scope, activityContext));
}
catch
{
await scope.DisposeScopeAsync().ConfigureAwait(false);
throw;
}
}