Can use aggregate loaded from [Aggregate] in middleware. Closes GH-628

JasperFx · Nov 13, 2023 · 21ced43 · 21ced43
1 parent ca0ea5a
commit 21ced43
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Persistence/Wolverine.Marten/AggregateHandlerAttribute.cs b/src/Persistence/Wolverine.Marten/AggregateHandlerAttribute.cs
@@ -142,8 +142,8 @@ internal static void DetermineEventCaptureHandling(IChain chain, MethodCall firs
 
     internal static Variable RelayAggregateToHandlerMethod(MethodCall loader, MethodCall firstCall, Type aggregateType)
     {
-        var aggregateVariable = new Variable(aggregateType!,
-            $"{loader.ReturnVariable!.Usage}.{nameof(IEventStream<string>.Aggregate)}");
+        var aggregateVariable = new MemberAccessVariable(loader.ReturnVariable,
+            typeof(IEventStream<>).MakeGenericType(aggregateType).GetProperty("Aggregate"));
 
         if (firstCall.HandlerType == aggregateType)
         {