Correct TrackGraph nullability (#27261)

Fixes #27257
dotnet · Jan 25, 2022 · d95657a · d95657a
1 parent 9e8ec9f
commit d95657a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/EFCore/ChangeTracking/ChangeTracker.cs b/src/EFCore/ChangeTracking/ChangeTracker.cs
@@ -301,7 +301,7 @@ public virtual void TrackGraph(
     /// <typeparam name="TState">The type of the state object.</typeparam>
     public virtual void TrackGraph<TState>(
         object rootEntity,
-        TState? state,
+        TState state,
         Func<EntityEntryGraphNode<TState>, bool> callback)
     {
         Check.NotNull(rootEntity, nameof(rootEntity));

diff --git a/src/EFCore/ChangeTracking/EntityEntryGraphNode`.cs b/src/EFCore/ChangeTracking/EntityEntryGraphNode`.cs
@@ -24,7 +24,7 @@ public class EntityEntryGraphNode<TState> : EntityEntryGraphNode
     [EntityFrameworkInternal]
     public EntityEntryGraphNode(
         InternalEntityEntry entry,
-        TState? state,
+        TState state,
         InternalEntityEntry? sourceEntry,
         INavigationBase? inboundNavigation)
         : base(entry, sourceEntry, inboundNavigation)
@@ -35,7 +35,7 @@ public EntityEntryGraphNode(
     /// <summary>
     ///     Gets or sets state that will be available to all nodes that are visited after this node.
     /// </summary>
-    public virtual TState? NodeState { get; set; }
+    public virtual TState NodeState { get; set; }
 
     /// <summary>
     ///     Creates a new node for the entity that is being traversed next in the graph.