Merge pull request #124 from tsf9810292/feature/support-object-type

Support for passing in types that do not match exact entity type
artiomchi · Nov 20, 2023 · 5501125 · 5501125
2 parents 473d273 + 29f265a
commit 5501125
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/FlexLabs.EntityFrameworkCore.Upsert/UpsertCommandBuilder.cs b/src/FlexLabs.EntityFrameworkCore.Upsert/UpsertCommandBuilder.cs
@@ -1,4 +1,4 @@
-using System;
+using System;
 using System.Collections.Generic;
 using System.Globalization;
 using System.Linq;

diff --git a/src/FlexLabs.EntityFrameworkCore.Upsert/UpsertExtensions.cs b/src/FlexLabs.EntityFrameworkCore.Upsert/UpsertExtensions.cs
@@ -48,6 +48,7 @@ public static UpsertCommandBuilder<TEntity> UpsertRange<TEntity>(this DbContext
                 throw new ArgumentNullException(nameof(entities));
 
             var entityType = dbContext.GetService<IModel>().FindEntityType(typeof(TEntity))
+                ?? (entities.Length == 0 ? null : dbContext.GetService<IModel>().FindEntityType(entities.First().GetType()))
                 ?? throw new InvalidOperationException(Resources.EntityTypeMustBeMappedInDbContext);
             return new UpsertCommandBuilder<TEntity>(dbContext, entityType, entities);
         }
@@ -67,13 +68,14 @@ public static UpsertCommandBuilder<TEntity> UpsertRange<TEntity>(this DbContext
             if (entities == null)
                 throw new ArgumentNullException(nameof(entities));
 
-            var entityType = dbContext.GetService<IModel>().FindEntityType(typeof(TEntity))
-                ?? throw new InvalidOperationException(Resources.EntityTypeMustBeMappedInDbContext);
             var collection = entities switch
             {
                 ICollection<TEntity> entityCollection => entityCollection,
                 _ => entities.ToArray()
             };
+            var entityType = dbContext.GetService<IModel>().FindEntityType(typeof(TEntity))
+                ?? (collection.Count > 0 ? null : dbContext.GetService<IModel>().FindEntityType(collection.First().GetType()))
+                ?? throw new InvalidOperationException(Resources.EntityTypeMustBeMappedInDbContext);
             return new UpsertCommandBuilder<TEntity>(dbContext, entityType, collection);
         }