Upsert родительский и дочерний элементы с EntityFramework

Это то, что я использую, чтобы получить то, что я хочу. Для моей цели это прекрасно работает:

foreach (var metric in metrics)
{
    var added = context.Set(metric.GetType()).Add(metric);

    foreach (var dbEntityEntry in context.ChangeTracker.Entries())
    {
        //If these entities exist, don't add them.
        var metricContext = dbEntityEntry.Entity as MetricContext;
        if (metricContext != null)
        {
            var found = context.MetricContext.FirstOrDefault(c => c.Context == metricContext.Context);
            if (found != default(MetricContext))
            {
                dbEntityEntry.State = EntityState.Detached;
            }
        }
    }
}

Это отсоединяет любые объекты, которые уже находятся в базе данных, то есть они не будут обрабатываться.

Это работает:

public void Upsert(ParentEntity parentEntity) {
    ParentEntity existing = DB.Parents.Find(parentEntity.Id);

    if (existing == null) {
        DB.Parents.Add(parentEntity);
    } else {
        var newChildKeys = new HashSet<int>(parentEntity.Children.Select(m => m.Id));

        var existingChildren = DB.Children.Where(m => m.ParentId == parentEntity.Id).ToArray();
        var existingChildrenKeys = new HashSet<int>(existingChildren.Select(m => m.Id));

        foreach (var existingChild in existingChildren) {
            if (newChildKeys.Contains(existingChild.Id)) {
                DB.Entry(existingChild).State = EntityState.Detached;
            } else {
                DB.Children.Remove(existingChild);
            }
        }

        foreach (var child in parentEntity.Children) {
            child.ParentId = parentEntity.Id;
            if (existingChildrenKeys.Contains((child.Id))) {
                DB.Children.Attach(child);
                DB.Entry(child).State = EntityState.Modified;
            } else {
                DB.Children.Add(child);
            }
        }

        DB.Entry(existing).State = EntityState.Detached;
        DB.Parents.Attach(parentEntity);
        DB.Entry(parentEntity).State = EntityState.Modified;
    }

    SaveChanges();
}

Возможно, его можно улучшить со стилистической точки зрения/сделать более общим. Основная идея

• Найдите общие ключи для сущностей (существующие в базе данных или заменяющие сущности)
• Для ключей общий отсоедините старый объект, присоедините новый объект и пометьте его как измененный.
• Для ключей, которые больше не существуют - удалить сущность
• Для новых ключей - добавить сущность