Entity Framework带条件映射实现方法？基于Student、Teacher、Document类

First, let's finalize the Document class to track which parent entity it belongs to, using the ParentType values you specified (1 for Student, 2 for Teacher):

public class Document : BaseEntity 
{
    public string DocumentName { get; set; } // Adjust this to match your actual document properties
    public int ParentId { get; set; }
    public int ParentType { get; set; } // 1 = Student, 2 = Teacher

    // Navigation properties to link back to parent entities
    public Student? Student { get; set; }
    public Teacher? Teacher { get; set; }
}

Next, configure the relationships in your DbContext using EF's fluent API. The key is using HasFilter to enforce that each parent's Documents collection only includes records matching their parent type:

protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    // Configure Student ↔ Document relationship
    modelBuilder.Entity<Student>()
        .HasMany(s => s.Documents)
        .WithOne(d => d.Student)
        .HasForeignKey(d => d.ParentId)
        .HasPrincipalKey(s => s.ItemId)
        .HasFilter("ParentType = 1"); // Only fetch documents linked to Students

    // Configure Teacher ↔ Document relationship
    modelBuilder.Entity<Teacher>()
        .HasMany(t => t.Documents)
        .WithOne(d => d.Teacher)
        .HasForeignKey(d => d.ParentId)
        .HasPrincipalKey(t => t.ItemId)
        .HasFilter("ParentType = 2"); // Only fetch documents linked to Teachers

    // Add a check constraint to ensure valid ParentType values (data integrity)
    modelBuilder.Entity<Document>()
        .HasCheckConstraint("CK_Document_ParentType", "ParentType IN (1, 2)");

    base.OnModelCreating(modelBuilder);
}

Key Details: #

• The HasFilter clause injects a conditional check directly into the SQL query EF generates when loading the Documents collection. So when you access student.Documents, EF automatically filters for ParentType = 1—no manual Where clauses needed.
• The check constraint on Document prevents invalid parent types from being saved, keeping your database consistent.

Pro Tip: Automate ParentType Setup #

Avoid manual errors by adding helper methods to your parent classes to set ParentType and ParentId automatically when adding documents:

public class Student : BaseEntity 
{
    public string Name { get; set; }
    public string Surname { get; set; }
    public ICollection<Document> Documents { get; set; } = new List<Document>();

    public void AddDocument(Document document)
    {
        document.ParentType = 1;
        document.ParentId = this.ItemId;
        Documents.Add(document);
    }
}

public class Teacher : BaseEntity 
{
    public string Name { get; set; }
    public string Surname { get; set; }
    public ICollection<Document> Documents { get; set; } = new List<Document>();

    public void AddDocument(Document document)
    {
        document.ParentType = 2;
        document.ParentId = this.ItemId;
        Documents.Add(document);
    }
}

Now calling student.AddDocument(new Document { DocumentName = "Transcript.pdf" }) will handle all the parent linking automatically.

内容的提问来源于stack exchange，提问作者user3351644