Fixes #57

src/AutoMapper.Collection.EntityFrameworkCore/GenerateEntityFrameworkCorePrimaryKeyPropertyMaps.cs

@@ -7,8 +7,7 @@
 
 namespace AutoMapper.EntityFrameworkCore
 {
-    public class GenerateEntityFrameworkCorePrimaryKeyPropertyMaps<TDatabaseContext> : IGeneratePropertyMaps
-        where TDatabaseContext : DbContext
+    public class GenerateEntityFrameworkCorePrimaryKeyPropertyMaps : IGeneratePropertyMaps
     {
         private readonly IModel _model;
 
@@ -20,7 +19,7 @@ public IEnumerable<PropertyMap> GeneratePropertyMaps(TypeMap typeMap)
         {
             var propertyMaps = typeMap.PropertyMaps;
             var keyMembers = _model.FindEntityType(typeMap.DestinationType)?.FindPrimaryKey().Properties ?? new List<IProperty>();
-            return keyMembers.Select(m => propertyMaps.FirstOrDefault(p => p.DestinationMember.Name == m.Name));
+            return propertyMaps.Where(p => keyMembers.Any(m => m.Name == p.DestinationMember.Name));
         }
     }
 }

src/AutoMapper.Collection.EntityFrameworkCore/MapperConfigurationExpressionExtensions.cs

@@ -18,7 +18,7 @@ public static void UseEntityFrameworkCoreModel<TContext>(this IMapperConfigurati
         {
             using (var context = new TContext())
             {
-                config.UseEntityFrameworkCoreModel<TContext>(context.Model);
+                config.UseEntityFrameworkCoreModel(context.Model);
             }
         }
 
@@ -47,15 +47,15 @@ public static void UseEntityFrameworkCoreModel<TContext>(this IMapperConfigurati
             using (var scope = serviceProvider.CreateScope())
             {
                 var context = scope.ServiceProvider.GetRequiredService<TContext>();
-                config.UseEntityFrameworkCoreModel<TContext>(context.Model);
+                config.UseEntityFrameworkCoreModel(context.Model);
             }
         }
 
         /// <summary>
         /// Generates and adds property maps based on the primary keys for the given <see cref="DbContext"/>. This method is generally
         /// only used if you are using <see cref="DbContextOptionsBuilder.UseModel(IModel)"/>.
         /// </summary>
-        public static void UseEntityFrameworkCoreModel<TContext>(this IMapperConfigurationExpression config, IModel model)
-            where TContext : DbContext => config.SetGeneratePropertyMaps(new GenerateEntityFrameworkCorePrimaryKeyPropertyMaps<TContext>(model));
+        public static void UseEntityFrameworkCoreModel(this IMapperConfigurationExpression config, IModel model)
+            => config.SetGeneratePropertyMaps(new GenerateEntityFrameworkCorePrimaryKeyPropertyMaps(model));
     }
 }