If relationship PrimaryProperties aren't primary key, generate HasPri…

…ncipalKey on relationshipo
loresoft · Nov 14, 2024 · 7ce664a · 7ce664a
1 parent 296e690
commit 7ce664a
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/EntityFrameworkCore.Generator.Core/Templates/MappingClassTemplate.cs b/src/EntityFrameworkCore.Generator.Core/Templates/MappingClassTemplate.cs
@@ -224,6 +224,30 @@ private void GenerateRelationshipMapping(Relationship relationship)
         }
         CodeBuilder.Append(")");
 
+        var primaryKeys = relationship.PrimaryProperties;
+        var nonPrimaryPrincipalKey = !primaryKeys
+            .Select(pp => relationship.PrimaryEntity.Properties.ByProperty(pp.PropertyName))
+            .All(p => p.IsPrimaryKey ?? true);
+
+        if (nonPrimaryPrincipalKey)
+        {
+            CodeBuilder.AppendLine();
+
+            CodeBuilder.Append(".HasPrincipalKey(t => ");
+            if (primaryKeys.Count > 1)
+            {
+                CodeBuilder.Append("new { ");
+                CodeBuilder.Append(string.Join(", ", primaryKeys.Select(pp => $"t.{pp.PropertyName.ToSafeName()}")));
+                CodeBuilder.Append(" }");
+            }
+            else
+            {
+                var propertyName = primaryKeys.First().PropertyName.ToSafeName();
+                CodeBuilder.Append($"t.{propertyName}");
+            }
+            CodeBuilder.Append(")");
+        }
+
         if (!string.IsNullOrEmpty(relationship.RelationshipName))
         {
             CodeBuilder.AppendLine();