Develop

src/Chronicle.Integrations.MongoDB/src/Chronicle.Integrations.MongoDB.csproj

@@ -11,10 +11,10 @@
     <PackageProjectUrl>https://github.com/chronicle-stack/Chronicle.Integrations.MongoDB</PackageProjectUrl>
     <PackageLicenseUrl>https://github.com/chronicle-stack/Chronicle/blob/master/LICENSE</PackageLicenseUrl>
     <PackageIconUrl>https://avatars1.githubusercontent.com/u/42150754?s=200</PackageIconUrl>
-    <VersionPrefix>3.1.0</VersionPrefix>
-    <Version>3.1.0</Version>
-    <AssemblyVersion>3.1.0</AssemblyVersion>
-    <FileVersion>3.1.0</FileVersion>
+    <VersionPrefix>3.1.1</VersionPrefix>
+    <Version>3.1.1</Version>
+    <AssemblyVersion>3.1.1</AssemblyVersion>
+    <FileVersion>3.1.1</FileVersion>
   </PropertyGroup>
 
   <ItemGroup>

src/Chronicle.Integrations.MongoDB/src/Persistence/MongoSagaState.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Linq;
 using System.Reflection;
 using MongoDB.Bson.Serialization.Attributes;
 
@@ -15,7 +16,12 @@ internal class MongoSagaState : ISagaState
         public string SagaType { get; set; }
         public SagaStates State { get; set; }
         public object Data { get; set; }
-        Type ISagaState.Type => Assembly.GetEntryAssembly()?.GetType(SagaType);
+
+        Type ISagaState.Type => _type ??= AppDomain.CurrentDomain.GetAssemblies()
+                .Select(a => a.GetType(SagaType))
+                .FirstOrDefault(t => t is {});
+
+        private Type _type;
 
         public void Update(SagaStates state, object data = null)
         {