DbContext vs ObjectContext — изменение поведения свойства навигации

Я переношу большой проект из ObjectContext в DbContext, EF6.1.3. Просто столкнулся с проблемой, которую будет трудно надежно отследить в исходном коде, и мне интересно, может ли быть подход, с помощью которого я могу имитировать поведение ObjectContext.

Рассмотрим два класса, Parent и Child. Родитель имеет ноль или более дочерних объектов. На уровне таблицы у Child есть столбец ParentID, который находится в отношении FK со столбцом ID в родительском объекте. Вот два класса POCO, которые я создал, чтобы проиллюстрировать проблему:

Partial Public Class Parent
    Public Property ID As Integer

    Public Overridable Property Children As ICollection(Of Child) = New HashSet(Of Child)

End Class

Partial Public Class Child
    Public Property ID As Integer
    Public Property ParentID As Integer

    Public Overridable Property Parent As Parent

End Class

и вот небольшая программа для иллюстрации проблемы:

  Sub Main()
    Using session As New testEntities
      Dim parent = session.Parents.Add(session.Parents.Create)
      Dim child = session.Children.Create
      parent.Children.Add(child)
      Console.WriteLine(child.Parent Is Nothing)
      session.SaveChanges()
      Console.WriteLine(child.Parent Is Nothing)
    End Using

В реализации ObjectContext добавление дочернего элемента к родительскому также задает свойство родительского элемента дочернего элемента. С DbContext этого не происходит до тех пор, пока сеанс не будет зафиксирован.

В коде, который я переношу, есть несколько мест (которые мы нашли до сих пор), где коду будет передан эквивалент объекта Child, который был добавлен к Parent, а затем попытка сослаться на объект Parent через свойство Parent ребенка . Они компилируются правильно, но поведение во время выполнения «сломано» с DbContext. Поиск всех таких случаев, когда используется этот шаблон, будет дорогостоящим, и будет очень легко пропустить случаи, которые затем вызовут проблемы во время выполнения. Может ли кто-нибудь предложить обходной путь, который позволит коду работать как есть? Я предполагаю, что мы могли бы изменить файл TT, чтобы сгенерировать наш собственный класс вместо HashSet для свойства Children, реализовать конструктор, который принимает ссылку на зависимое свойство, и метод Add, который обновляет зависимое свойство. Прежде чем мы пойдем по этому пути, есть ли что-то более простое, что мы могли упустить?