EF Core中删除并重建关联实体的代码简化方案咨询

我有一个带有关联实体的EF Core实体，代码如下：

public sealed class MyEntity
{
    public Guid Id { get; init; }

    public string Description { get; set; }

    public int? Age { get; set; }

    public RelatadEntity? RelatadEntity { get; set; }
}

对应的更新代码如下：

public async Task Update(Guid id, UpdateEntity update)
{
    var entity = (await context.TableA.FindAsync(id))!;
    var entityFromModel = mapper.Map<MyEntity>(update);

    entity.Description = entityFromModel.Description;
    entity.Age = entityFromModel.Age;
        
    if (entity.RelatedEntity is not null)
        context.TableB.Remove(entity.RelatedEntity);
    entity.RelatedEntity = entityFromModel.RelatedEntity is not null ? (await context.TableB.AddAsync(entityFromModel.RelatedEntity)).Entity : null;
        
    await context.SaveChangesAsync();
}

这段代码会删除原有关联实体（若存在）并创建新关联实体，同时更新主实体，但代码量较大。我对EF Core不太熟悉，想了解是否有更简洁的实现方式，能否使用context.Update自动重建关联实体？

1. 利用AutoMapper直接映射到已跟踪实体 #

你可以用AutoMapper的Map重载，直接把UpdateEntity映射到从数据库查询出的已跟踪实体上，省去手动赋值的步骤：

public async Task Update(Guid id, UpdateEntity update)
{
    // 提前Include关联实体，确保能获取到旧关联
    var entity = await context.TableA.Include(e => e.RelatedEntity).FirstAsync(e => e.Id == id);
    
    // 先处理旧关联：存在则删除
    if (entity.RelatedEntity != null)
    {
        context.TableB.Remove(entity.RelatedEntity);
    }
    
    // 直接映射主实体字段和新关联实体
    mapper.Map(update, entity);
    
    // 新关联实体不为空时，EF会自动跟踪并标记为新增
    if (entity.RelatedEntity != null)
    {
        context.TableB.Add(entity.RelatedEntity);
    }
    
    await context.SaveChangesAsync();
}

需要确保你的AutoMapper配置正确，能将UpdateEntity中的关联字段映射到MyEntity.RelatedEntity上。

2. context.Update无法自动替换关联实体 #

context.Update会把实体及其所有关联实体标记为Modified，但它不会自动删除原有的关联实体，也无法区分关联实体是需要更新还是完全替换。如果你的业务逻辑是每次更新都要替换关联实体（删旧加新），直接用Update达不到预期效果，它只会标记现有关联为修改状态，不会处理旧关联的删除。

3. 封装关联实体替换逻辑进一步简化 #

可以把关联实体的替换逻辑封装成扩展方法，减少重复代码：

public static void ReplaceRelatedEntity<TMain, TRelated>(this DbContext context, TMain mainEntity, Expression<Func<TMain, TRelated?>> property, TRelated? newRelated)
    where TRelated : class
{
    var oldRelated = property.Compile()(mainEntity);
    if (oldRelated != null)
    {
        context.Remove(oldRelated);
    }
    property.SetValue(mainEntity, newRelated);
    if (newRelated != null)
    {
        context.Add(newRelated);
    }
}

之后在更新方法中调用即可：

public async Task Update(Guid id, UpdateEntity update)
{
    var entity = await context.TableA.Include(e => e.RelatedEntity).FirstAsync(e => e.Id == id);
    mapper.Map(update, entity);
    
    context.ReplaceRelatedEntity(entity, e => e.RelatedEntity, entity.RelatedEntity);
    
    await context.SaveChangesAsync();
}

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