dan/Toolkit2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Pipelines

Browse Source
This commit is contained in:
Dan Clark
2024-11-25 09:44:12 +00:00
parent 93e2be4eb0
commit d1ee8add0c
9 changed files with 197 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Toolkit.Foundation/AsyncHandlerKeyedInitialization.cs
+41 -10
View File
@@ -5,20 +5,35 @@ namespace Toolkit.Foundation;
public class AsyncHandlerKeyedInitialization<TMessage, THandler>(string key, IServiceProvider provider) :
IInitialization where THandler : class, IAsyncHandler<TMessage>
where TMessage : class
where TMessage : class
{
public void Initialize()
{
if (!StrongReferenceMessenger.Default.IsRegistered<AsyncResponseEventArgs<TMessage, Unit>, string>(provider, key))
{
StrongReferenceMessenger.Default.Register<IServiceProvider, AsyncResponseEventArgs<TMessage, Unit>, string>(provider, key,
(provider, args) =>
async (provider, args) =>
{
foreach (IAsyncHandler<TMessage> handler in provider.GetKeyedServices<IAsyncHandler<TMessage>>(key))
IEnumerable<IAsyncHandler<TMessage>> handlers = provider.GetKeyedServices<IAsyncHandler<TMessage>>(key);
IEnumerable<IAsyncPipelineBehavior<TMessage, Unit>> behaviors = provider.GetServices<IAsyncPipelineBehavior<TMessage, Unit>>();
AsyncHandlerDelegate<Unit> handlerDelegate = async () =>
{
handler.Handle(args.Message, args.CancellationToken);
args.Reply(Unit.Value);
foreach (IAsyncHandler<TMessage> handler in handlers)
{
await handler.Handle(args.Message, args.CancellationToken);
}
return Unit.Value;
};
foreach (IAsyncPipelineBehavior<TMessage, Unit> behavior in behaviors.Reverse())
{
AsyncHandlerDelegate<Unit> next = handlerDelegate;
handlerDelegate = () => behavior.Handle(args.Message, next);
}
await handlerDelegate();
args.Reply(Unit.Value);
});
}
}
@@ -26,20 +41,36 @@ public class AsyncHandlerKeyedInitialization<TMessage, THandler>(string key, ISe
public class AsyncHandlerKeyedInitialization<TMessage, TResponse, THandler>(string key, IServiceProvider provider) :
IInitialization where THandler : class, IAsyncHandler<TMessage, TResponse>
where TMessage : class
where TMessage : class
{
public void Initialize()
{
if (!StrongReferenceMessenger.Default.IsRegistered<AsyncResponseEventArgs<TMessage, TResponse>, string>(provider, key))
{
StrongReferenceMessenger.Default.Register<IServiceProvider, AsyncResponseEventArgs<TMessage, TResponse>, string>(provider, key,
(provider, args) =>
async (provider, args) =>
{
foreach (IAsyncHandler<TMessage, TResponse> handler in provider.GetKeyedServices<IAsyncHandler<TMessage, TResponse>>(key))
IEnumerable<IAsyncHandler<TMessage, TResponse>> handlers = provider.GetKeyedServices<IAsyncHandler<TMessage, TResponse>>(key);
IEnumerable<IAsyncPipelineBehavior<TMessage, TResponse>> behaviors = provider.GetServices<IAsyncPipelineBehavior<TMessage, TResponse>>();
AsyncHandlerDelegate<TResponse> handlerDelegate = async () =>
{
args.Reply(handler.Handle(args.Message, args.CancellationToken));
TResponse response = default!;
foreach (IAsyncHandler<TMessage, TResponse> handler in handlers)
{
response = await handler.Handle(args.Message, args.CancellationToken);
}
return response;
};
foreach (IAsyncPipelineBehavior<TMessage, TResponse> behavior in behaviors.Reverse())
{
AsyncHandlerDelegate<TResponse> next = handlerDelegate;
handlerDelegate = () => behavior.Handle(args.Message, next);
}
args.Reply(await handlerDelegate());
});
}
}
}
}