Ensure entity type is registered for EntityEntry

Fixes #261
TurnerSoftware · Oct 31, 2021 · a163b11 · a163b11
1 parent bd0cd9b
commit a163b11
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/MongoFramework/Infrastructure/EntityEntry.cs b/src/MongoFramework/Infrastructure/EntityEntry.cs
@@ -1,5 +1,6 @@
 using System;
 using MongoDB.Bson;
+using MongoFramework.Infrastructure.Mapping;
 
 namespace MongoFramework.Infrastructure
 {
@@ -33,6 +34,9 @@ public EntityEntry(object entity, EntityEntryState state)
 			Entity = entity;
 			EntityType = entity.GetType();
 
+			// Ensure the entity type is registered before we potentially serialize the entity to a BSON document
+			EntityMapping.TryRegisterType(EntityType, out _);
+
 			if (state == EntityEntryState.NoChanges)
 			{
 				OriginalValues = Entity.ToBsonDocument();