Fixed type in mongo saga state

snatch-dev · Aug 20, 2020 · fc1ae7f · fc1ae7f
1 parent 5e94162
commit fc1ae7f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/src/Chronicle.Integrations.MongoDB/src/Chronicle.Integrations.MongoDB.csproj b/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>

diff --git a/src/Chronicle.Integrations.MongoDB/src/Persistence/MongoSagaState.cs b/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)
         {