/
CommandExecutor.cs
148 lines (128 loc) · 4.88 KB
/
CommandExecutor.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Threading;
using System.Threading.Tasks;
using IOKode.OpinionatedFramework.Commands;
using IOKode.OpinionatedFramework.ConfigureApplication;
using Command = IOKode.OpinionatedFramework.Commands.Command;
namespace IOKode.OpinionatedFramework.ContractImplementations.CommandExecutor;
public class CommandExecutor : ICommandExecutor
{
private readonly CommandMiddleware[] middlewares;
private Dictionary<string, object?>? sharedData;
public CommandExecutor(CommandMiddleware[] middlewares,
IEnumerable<KeyValuePair<string, object?>>? sharedData = null)
{
this.middlewares = middlewares;
if (sharedData is Dictionary<string, object?> sharedDataAsDict)
{
this.sharedData = sharedDataAsDict;
}
else
{
this.sharedData = sharedData?.ToDictionary(kvp => kvp.Key, kvp => kvp.Value) ?? new Dictionary<string, object?>();
}
}
public async Task InvokeAsync<TCommand>(TCommand command, CancellationToken cancellationToken)
where TCommand : Command
{
using (Container.CreateScope())
{
var context = SettableCommandContext.Create(command.GetType(), this.sharedData, cancellationToken);
await InvokeMiddlewarePipeline(command, context, 0);
}
}
public async Task<TResult> InvokeAsync<TCommand, TResult>(TCommand command, CancellationToken cancellationToken)
where TCommand : Command<TResult>
{
using (Container.CreateScope())
{
var context = SettableCommandContext.Create(command.GetType(), sharedData, cancellationToken);
return await InvokeMiddlewarePipeline<TCommand, TResult>(command, context, 0);
}
}
private static MethodInfo GetExecuteMethod<TCommand>()
{
var executeMethod = typeof(TCommand).GetMethod("ExecuteAsync", BindingFlags.Instance | BindingFlags.NonPublic)!;
return executeMethod;
}
private static MethodInfo GetPrepareMethod<TCommand>()
{
var executeMethod = typeof(TCommand).GetMethod("PrepareAsync", BindingFlags.Instance | BindingFlags.NonPublic)!;
return executeMethod;
}
/// <summary>
/// Prepare and execute command that DO NOT return any result.
/// </summary>
private async Task PrepareAndExecuteAsync<TCommand>(TCommand command, SettableCommandContext context)
{
var prepare = GetPrepareMethod<TCommand>();
var execute = GetExecuteMethod<TCommand>();
var parameters = new object?[] { context };
try
{
await (Task)prepare.Invoke(command, parameters)!;
await (Task)execute.Invoke(command, parameters)!;
}
catch (TargetInvocationException ex)
{
throw ex.InnerException!;
}
context.SetAsExecuted();
this.sharedData = context.ShareData;
}
/// <summary>
/// Prepare and execute command that DO return a result.
/// </summary>
private async Task<TResult> PrepareAndExecuteAsync<TCommand, TResult>(TCommand command,
SettableCommandContext context)
{
var prepare = GetPrepareMethod<TCommand>();
var execute = GetExecuteMethod<TCommand>();
var parameters = new object?[] { context };
TResult result;
try
{
await (Task)prepare.Invoke(command, parameters)!;
result = await (Task<TResult>)execute.Invoke(command, parameters)!;
}
catch (TargetInvocationException ex)
{
throw ex.InnerException!;
}
context.SetAsExecuted();
context.SetResult(result);
this.sharedData = context.ShareData;
return result;
}
private async Task InvokeMiddlewarePipeline<TCommand>(TCommand command, SettableCommandContext context, int index)
{
if (index >= middlewares.Length)
{
await PrepareAndExecuteAsync(command, context);
return;
}
var middleware = middlewares[index];
await middleware.ExecuteAsync(context,
ctx => InvokeMiddlewarePipeline(command, (SettableCommandContext)ctx, index + 1));
}
private async Task<TResult> InvokeMiddlewarePipeline<TCommand, TResult>(TCommand command,
SettableCommandContext context, int index)
where TCommand : Command<TResult>
{
TResult result = default!;
if (index >= middlewares.Length)
{
return await PrepareAndExecuteAsync<TCommand, TResult>(command, context);
}
var middleware = middlewares[index];
await middleware.ExecuteAsync(context,
async ctx =>
{
result = await InvokeMiddlewarePipeline<TCommand, TResult>(command, (SettableCommandContext)ctx,
index + 1);
});
return result;
}
}