Cast based on StreamIdentity

JasperFx · Feb 6, 2025 · 11d8dc6 · 11d8dc6
1 parent 7f92687
commit 11d8dc6
Showing 1 changed file with 3 additions and 13 deletions.
diff --git a/src/Marten/Events/Aggregation/CustomProjection.cs b/src/Marten/Events/Aggregation/CustomProjection.cs
@@ -376,7 +376,7 @@ async ValueTask<TDoc> ILiveAggregator<TDoc>.BuildAsync(IReadOnlyList<IEvent> eve
         var documentSessionBase = session as DocumentSessionBase ?? (DocumentSessionBase)session.DocumentStore.LightweightSession();
 
         var latestEvent = events.Last();
-        var streamId = IdentityFromEvent(latestEvent);
+        var streamId = IdentityFromEvent(documentSessionBase.Options.EventGraph.StreamIdentity, latestEvent);
         var slice = new EventSlice<TDoc, TId>(streamId, session, events);
         if (Lifecycle == ProjectionLifecycle.Live)
         {
@@ -420,16 +420,6 @@ public IAggregateCache<TId, TDoc> CacheFor(Tenant tenant)
         }
     }
 
-    public TId IdentityFromEvent(IEvent e)
-    {
-        if (typeof(TId) == typeof(Guid))
-        {
-            return e.StreamId.To<TId>();
-        }
-
-        return e.StreamKey.To<TId>();
-    }
+    public TId IdentityFromEvent(StreamIdentity streamIdentity, IEvent e) =>
+        streamIdentity == StreamIdentity.AsGuid ? e.StreamId.To<TId>() : e.StreamKey.To<TId>();
 }
-
-
-