Merge pull request #541 from json-api-dotnet/feat/fluent-get-affected

wisepotato · web-flow · commit c6762b399bca · 2019-07-18T09:13:24.000+02:00
GetAffected syntax in resource hook helpers
diff --git a/src/Examples/GettingStarted/Properties/launchSettings.json b/src/Examples/GettingStarted/Properties/launchSettings.json
@@ -17,7 +17,8 @@
     },
     "GettingStarted": {
       "commandName": "Project",
-      "launchBrowser": true
+      "launchBrowser": true,
+      "environmentVariables": {}
     }
   }
-}
+}
diff --git a/src/Examples/JsonApiDotNetCoreExample/Properties/launchSettings.json b/src/Examples/JsonApiDotNetCoreExample/Properties/launchSettings.json
@@ -25,4 +25,4 @@
       }
     }
   }
-}
+}
diff --git a/src/Examples/NoEntityFrameworkExample/Properties/launchSettings.json b/src/Examples/NoEntityFrameworkExample/Properties/launchSettings.json
@@ -16,7 +16,8 @@
       }
     },
     "NoEntityFrameworkExample": {
-      "commandName": "Project"
+      "commandName": "Project",
+      "environmentVariables": {}
     }
   }
 }
diff --git a/src/Examples/ResourceEntitySeparationExample/Properties/launchSettings.json b/src/Examples/ResourceEntitySeparationExample/Properties/launchSettings.json
@@ -1,23 +1,24 @@
 {
-    "iisSettings": {
-        "windowsAuthentication": false,
-        "anonymousAuthentication": true,
-        "iisExpress": {
-            "applicationUrl": "http://localhost:57181/",
-            "sslPort": 0
-        }
+  "iisSettings": {
+    "windowsAuthentication": false,
+    "anonymousAuthentication": true,
+    "iisExpress": {
+      "applicationUrl": "http://localhost:57181/",
+      "sslPort": 0
+    }
+  },
+  "profiles": {
+    "IIS Express": {
+      "commandName": "IISExpress",
+      "launchBrowser": true,
+      "launchUrl": "api/v1/students",
+      "environmentVariables": {
+        "ASPNETCORE_ENVIRONMENT": "Development"
+      }
     },
-    "profiles": {
-        "IIS Express": {
-            "commandName": "IISExpress",
-            "launchBrowser": true,
-            "launchUrl": "api/v1/students",
-            "environmentVariables": {
-                "ASPNETCORE_ENVIRONMENT": "Development"
-            }
-        },
-        "ResourceEntitySeparationExample": {
-            "commandName": "Project"
-        }
+    "ResourceEntitySeparationExample": {
+      "commandName": "Project",
+      "environmentVariables": {}
     }
-}
+  }
+}
diff --git a/src/JsonApiDotNetCore/Extensions/IServiceCollectionExtensions.cs b/src/JsonApiDotNetCore/Extensions/IServiceCollectionExtensions.cs
@@ -64,7 +64,7 @@ public static IServiceCollection AddJsonApi(
             var config = new JsonApiOptions();
             configureOptions(config);
 
-            if(autoDiscover != null)
+            if (autoDiscover != null)
             {
                 var facade = new ServiceDiscoveryFacade(services, config.ResourceGraphBuilder);
                 autoDiscover(facade);
@@ -159,7 +159,7 @@ public static void AddJsonApiInternals(
             services.AddScoped<IControllerContext, Services.ControllerContext>();
             services.AddScoped<IDocumentBuilderOptionsProvider, DocumentBuilderOptionsProvider>();
 
-           
+
             if (jsonApiOptions.EnableResourceHooks)
             {
                 services.AddSingleton(typeof(IHooksDiscovery<>), typeof(HooksDiscovery<>));
@@ -204,7 +204,7 @@ public static void SerializeAsJsonApi(this MvcOptions options, JsonApiOptions js
         /// Adds all required registrations for the service to the container
         /// </summary>
         /// <exception cref="JsonApiSetupException"/>
-        public static IServiceCollection AddResourceService<T>(this IServiceCollection services) 
+        public static IServiceCollection AddResourceService<T>(this IServiceCollection services)
         {
             var typeImplementsAnExpectedInterface = false;
 
@@ -213,28 +213,29 @@ public static IServiceCollection AddResourceService<T>(this IServiceCollection s
             // it is _possible_ that a single concrete type could be used for multiple resources...
             var resourceDescriptors = GetResourceTypesFromServiceImplementation(serviceImplementationType);
 
-            foreach(var resourceDescriptor in resourceDescriptors)
+            foreach (var resourceDescriptor in resourceDescriptors)
             {
-                foreach(var openGenericType in ServiceDiscoveryFacade.ServiceInterfaces)
+                foreach (var openGenericType in ServiceDiscoveryFacade.ServiceInterfaces)
                 {
                     // A shorthand interface is one where the id type is ommitted
                     // e.g. IResourceService<T> is the shorthand for IResourceService<T, TId>
                     var isShorthandInterface = (openGenericType.GetTypeInfo().GenericTypeParameters.Length == 1);
-                    if(isShorthandInterface && resourceDescriptor.IdType != typeof(int))
+                    if (isShorthandInterface && resourceDescriptor.IdType != typeof(int))
                         continue; // we can't create a shorthand for id types other than int
 
                     var concreteGenericType = isShorthandInterface
                         ? openGenericType.MakeGenericType(resourceDescriptor.ResourceType)
                         : openGenericType.MakeGenericType(resourceDescriptor.ResourceType, resourceDescriptor.IdType);
 
-                    if(concreteGenericType.IsAssignableFrom(serviceImplementationType)) {
+                    if (concreteGenericType.IsAssignableFrom(serviceImplementationType))
+                    {
                         services.AddScoped(concreteGenericType, serviceImplementationType);
                         typeImplementsAnExpectedInterface = true;
                     }
                 }
             }
 
-            if(typeImplementsAnExpectedInterface == false)
+            if (typeImplementsAnExpectedInterface == false)
                 throw new JsonApiSetupException($"{serviceImplementationType} does not implement any of the expected JsonApiDotNetCore interfaces.");
 
             return services;
@@ -244,12 +245,12 @@ private static HashSet<ResourceDescriptor> GetResourceTypesFromServiceImplementa
         {
             var resourceDecriptors = new HashSet<ResourceDescriptor>();
             var interfaces = type.GetInterfaces();
-            foreach(var i in interfaces)
+            foreach (var i in interfaces)
             {
-                if(i.IsGenericType)
+                if (i.IsGenericType)
                 {
                     var firstGenericArgument = i.GenericTypeArguments.FirstOrDefault();
-                    if(TypeLocator.TryGetResourceDescriptor(firstGenericArgument, out var resourceDescriptor) == true)
+                    if (TypeLocator.TryGetResourceDescriptor(firstGenericArgument, out var resourceDescriptor) == true)
                     {
                         resourceDecriptors.Add(resourceDescriptor);
                     }
diff --git a/src/JsonApiDotNetCore/Hooks/Execution/DiffableEntityHashSet.cs b/src/JsonApiDotNetCore/Hooks/Execution/DiffableEntityHashSet.cs
@@ -2,8 +2,12 @@
 using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
+using System.Linq.Expressions;
+using System.Reflection;
+using JsonApiDotNetCore.Extensions;
 using JsonApiDotNetCore.Internal;
 using JsonApiDotNetCore.Models;
+using JsonApiDotNetCore.Services;
 
 namespace JsonApiDotNetCore.Hooks
 {
@@ -21,31 +25,37 @@ public interface IDiffableEntityHashSet<TResource> : IEntityHashSet<TResource> w
         ///  with their associated current value from the database.
         /// </summary>
         IEnumerable<EntityDiffPair<TResource>> GetDiffs();
-            
+
     }
 
     /// <inheritdoc />
     public class DiffableEntityHashSet<TResource> : EntityHashSet<TResource>, IDiffableEntityHashSet<TResource> where TResource : class, IIdentifiable
     {
         private readonly HashSet<TResource> _databaseValues;
         private readonly bool _databaseValuesLoaded;
+        private Dictionary<PropertyInfo, HashSet<TResource>> _updatedAttributes;
 
         public DiffableEntityHashSet(HashSet<TResource> requestEntities,
                           HashSet<TResource> databaseEntities,
-                          Dictionary<RelationshipAttribute, HashSet<TResource>> relationships) 
+                          Dictionary<RelationshipAttribute, HashSet<TResource>> relationships,
+                          Dictionary<PropertyInfo, HashSet<TResource>> updatedAttributes)
             : base(requestEntities, relationships)
         {
             _databaseValues = databaseEntities;
             _databaseValuesLoaded |= _databaseValues != null;
+            _updatedAttributes = updatedAttributes;
         }
 
         /// <summary>
         /// Used internally by the ResourceHookExecutor to make live a bit easier with generics
         /// </summary>
         internal DiffableEntityHashSet(IEnumerable requestEntities,
                   IEnumerable databaseEntities,
-                  Dictionary<RelationshipAttribute, IEnumerable> relationships)
-            : this((HashSet<TResource>)requestEntities, (HashSet<TResource>)databaseEntities, TypeHelper.ConvertRelationshipDictionary<TResource>(relationships)) { }
+                  Dictionary<RelationshipAttribute, IEnumerable> relationships,
+                  IJsonApiContext jsonApiContext)
+            : this((HashSet<TResource>)requestEntities, (HashSet<TResource>)databaseEntities, TypeHelper.ConvertRelationshipDictionary<TResource>(relationships),
+              TypeHelper.ConvertAttributeDictionary(jsonApiContext.AttributesToUpdate, (HashSet<TResource>)requestEntities))
+        { }
 
 
         /// <inheritdoc />
@@ -60,6 +70,24 @@ public IEnumerable<EntityDiffPair<TResource>> GetDiffs()
             }
         }
 
+        /// <inheritdoc />
+        public new HashSet<TResource> GetAffected(Expression<Func<TResource, object>> NavigationAction)
+        {
+            var propertyInfo = TypeHelper.ParseNavigationExpression(NavigationAction);
+            var propertyType = propertyInfo.PropertyType;
+            if (propertyType.Inherits(typeof(IEnumerable))) propertyType = TypeHelper.GetTypeOfList(propertyType);
+            if (propertyType.Implements<IIdentifiable>())
+            {
+                // the navigation action references a relationship. Redirect the call to the relationship dictionary. 
+                return base.GetAffected(NavigationAction);
+            }
+            else if (_updatedAttributes.TryGetValue(propertyInfo, out HashSet<TResource> entities))
+            {
+                return entities;
+            }
+            return new HashSet<TResource>();
+        }
+
         private HashSet<TResource> ThrowNoDbValuesError()
         {
             throw new MemberAccessException("Cannot iterate over the diffs if the LoadDatabaseValues option is set to false");
diff --git a/src/JsonApiDotNetCore/Hooks/Execution/EntityHashSet.cs b/src/JsonApiDotNetCore/Hooks/Execution/EntityHashSet.cs
@@ -5,6 +5,7 @@
 using System;
 using System.Collections.ObjectModel;
 using System.Collections.Immutable;
+using System.Linq.Expressions;
 
 namespace JsonApiDotNetCore.Hooks
 {
@@ -26,7 +27,7 @@ public interface IEntityHashSet<TResource> : IByAffectedRelationships<TResource>
     /// </summary>
     public class EntityHashSet<TResource> : HashSet<TResource>, IEntityHashSet<TResource> where TResource : class, IIdentifiable
     {
-    
+
 
         /// <inheritdoc />
         public Dictionary<RelationshipAttribute, HashSet<TResource>> AffectedRelationships { get => _relationships; }
@@ -53,9 +54,15 @@ public Dictionary<RelationshipAttribute, HashSet<TResource>> GetByRelationship(T
         }
 
         /// <inheritdoc />
-        public Dictionary<RelationshipAttribute, HashSet<TResource>> GetByRelationship<TRelatedResource>()  where TRelatedResource : class, IIdentifiable
+        public Dictionary<RelationshipAttribute, HashSet<TResource>> GetByRelationship<TRelatedResource>() where TRelatedResource : class, IIdentifiable
         {
             return GetByRelationship(typeof(TRelatedResource));
         }
+
+        /// <inheritdoc />
+        public HashSet<TResource> GetAffected(Expression<Func<TResource, object>> NavigationAction)
+        {
+            return _relationships.GetAffected(NavigationAction);
+        }
     }
 }
diff --git a/src/JsonApiDotNetCore/Hooks/Execution/RelationshipsDictionary.cs b/src/JsonApiDotNetCore/Hooks/Execution/RelationshipsDictionary.cs
@@ -2,6 +2,8 @@
 using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
+using System.Linq.Expressions;
+using System.Reflection;
 using JsonApiDotNetCore.Internal;
 using JsonApiDotNetCore.Models;
 
@@ -15,13 +17,9 @@ public interface IRelationshipsDictionary { }
     /// <summary>
     /// An interface that is implemented to expose a relationship dictionary on another class.
     /// </summary>
-    public interface IByAffectedRelationships<TDependentResource> : 
+    public interface IByAffectedRelationships<TDependentResource> :
         IRelationshipGetters<TDependentResource> where TDependentResource : class, IIdentifiable
     {
-        /// todo: expose getters that behave something like this:
-        /// relationshipDictionary.GetAffected( entity => entity.NavigationProperty ).
-        /// see https://stackoverflow.com/a/17116267/4441216
-
         /// <summary>
         /// Gets a dictionary of affected resources grouped by affected relationships.
         /// </summary>
@@ -31,10 +29,11 @@ public interface IByAffectedRelationships<TDependentResource> :
     /// <summary>
     /// A helper class that provides insights in which relationships have been updated for which entities.
     /// </summary>
-    public interface IRelationshipsDictionary<TDependentResource> : 
-        IRelationshipGetters<TDependentResource>, 
-        IReadOnlyDictionary<RelationshipAttribute, HashSet<TDependentResource>>, 
-        IRelationshipsDictionary where TDependentResource : class, IIdentifiable { }
+    public interface IRelationshipsDictionary<TDependentResource> :
+        IRelationshipGetters<TDependentResource>,
+        IReadOnlyDictionary<RelationshipAttribute, HashSet<TDependentResource>>,
+        IRelationshipsDictionary where TDependentResource : class, IIdentifiable
+    { }
 
     /// <summary>
     /// A helper class that provides insights in which relationships have been updated for which entities.
@@ -49,16 +48,24 @@ public interface IRelationshipGetters<TResource> where TResource : class, IIdent
         /// Gets a dictionary of all entities that have an affected relationship to type <paramref name="principalType"/>
         /// </summary>
         Dictionary<RelationshipAttribute, HashSet<TResource>> GetByRelationship(Type relatedResourceType);
+
+        /// <summary>
+        /// Gets a collection of all the entities for the property within <paramref name="NavigationAction"/>
+        /// has been affected by the request
+        /// </summary>
+        /// <param name="NavigationAction"></param>
+        HashSet<TResource> GetAffected(Expression<Func<TResource, object>> NavigationAction);
     }
 
+
     /// <summary>
     /// Implementation of IAffectedRelationships{TDependentResource}
     /// 
     /// It is practically a ReadOnlyDictionary{RelationshipAttribute, HashSet{TDependentResource}} dictionary
     /// with the two helper methods defined on IAffectedRelationships{TDependentResource}.
     /// </summary>
     public class RelationshipsDictionary<TResource> :
-        Dictionary<RelationshipAttribute, HashSet<TResource>>, 
+        Dictionary<RelationshipAttribute, HashSet<TResource>>,
         IRelationshipsDictionary<TResource> where TResource : class, IIdentifiable
     {
         /// <summary>
@@ -70,7 +77,7 @@ public RelationshipsDictionary(Dictionary<RelationshipAttribute, HashSet<TResour
         /// <summary>
         /// Used internally by the ResourceHookExecutor to make live a bit easier with generics
         /// </summary>
-        internal RelationshipsDictionary(Dictionary<RelationshipAttribute, IEnumerable> relationships) 
+        internal RelationshipsDictionary(Dictionary<RelationshipAttribute, IEnumerable> relationships)
             : this(TypeHelper.ConvertRelationshipDictionary<TResource>(relationships)) { }
 
         /// <inheritdoc />
@@ -84,5 +91,12 @@ public Dictionary<RelationshipAttribute, HashSet<TResource>> GetByRelationship(T
         {
             return this.Where(p => p.Key.DependentType == relatedType).ToDictionary(p => p.Key, p => p.Value);
         }
+
+        /// <inheritdoc />
+        public HashSet<TResource> GetAffected(Expression<Func<TResource, object>> NavigationAction)
+        {
+            var property = TypeHelper.ParseNavigationExpression(NavigationAction);
+            return this.Where(p => p.Key.InternalRelationshipName == property.Name).Select(p => p.Value).SingleOrDefault();
+        }
     }
 }
diff --git a/src/JsonApiDotNetCore/Hooks/ResourceHookExecutor.cs b/src/JsonApiDotNetCore/Hooks/ResourceHookExecutor.cs
@@ -49,7 +49,7 @@ public virtual IEnumerable<TEntity> BeforeUpdate<TEntity>(IEnumerable<TEntity> e
             {
                 var relationships = node.RelationshipsToNextLayer.Select(p => p.Attribute).ToArray();
                 var dbValues = LoadDbValues(typeof(TEntity), (IEnumerable<TEntity>)node.UniqueEntities, ResourceHook.BeforeUpdate, relationships);
-                var diff = new DiffableEntityHashSet<TEntity>(node.UniqueEntities, dbValues, node.PrincipalsToNextLayer());
+                var diff = new DiffableEntityHashSet<TEntity>(node.UniqueEntities, dbValues, node.PrincipalsToNextLayer(), _context);
                 IEnumerable<TEntity> updated = container.BeforeUpdate(diff, pipeline);
                 node.UpdateUnique(updated);
                 node.Reassign(entities);
diff --git a/src/JsonApiDotNetCore/Internal/TypeHelper.cs b/src/JsonApiDotNetCore/Internal/TypeHelper.cs
diff --git a/test/UnitTests/ResourceHooks/AffectedEntitiesHelperTests.cs b/test/UnitTests/ResourceHooks/AffectedEntitiesHelperTests.cs
diff --git a/test/UnitTests/ResourceHooks/ResourceHookExecutor/Update/BeforeUpdate_WithDbValues_Tests.cs b/test/UnitTests/ResourceHooks/ResourceHookExecutor/Update/BeforeUpdate_WithDbValues_Tests.cs

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,8 @@`
`17`	`17`	`},`
`18`	`18`	`"GettingStarted": {`
`19`	`19`	`"commandName": "Project",`
`20`		`- "launchBrowser": true`
	`20`	`+ "launchBrowser": true,`
	`21`	`+ "environmentVariables": {}`
`21`	`22`	`}`
`22`	`23`	`}`
`23`		`-}`
	`24`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -25,4 +25,4 @@`
`25`	`25`	`}`
`26`	`26`	`}`
`27`	`27`	`}`
`28`		`-}`
	`28`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,8 @@`
`16`	`16`	`}`
`17`	`17`	`},`
`18`	`18`	`"NoEntityFrameworkExample": {`
`19`		`- "commandName": "Project"`
	`19`	`+ "commandName": "Project",`
	`20`	`+ "environmentVariables": {}`
`20`	`21`	`}`
`21`	`22`	`}`
`22`	`23`	`}`
Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ public static IServiceCollection AddJsonApi(`
`64`	`64`	`var config = new JsonApiOptions();`
`65`	`65`	`configureOptions(config);`
`66`	`66`
`67`		`- if(autoDiscover != null)`
	`67`	`+ if (autoDiscover != null)`
`68`	`68`	`{`
`69`	`69`	`var facade = new ServiceDiscoveryFacade(services, config.ResourceGraphBuilder);`
`70`	`70`	`autoDiscover(facade);`
`@@ -159,7 +159,7 @@ public static void AddJsonApiInternals(`
`159`	`159`	`services.AddScoped<IControllerContext, Services.ControllerContext>();`
`160`	`160`	`services.AddScoped<IDocumentBuilderOptionsProvider, DocumentBuilderOptionsProvider>();`
`161`	`161`
`162`		`-`
	`162`	`+`
`163`	`163`	`if (jsonApiOptions.EnableResourceHooks)`
`164`	`164`	`{`
`165`	`165`	`services.AddSingleton(typeof(IHooksDiscovery<>), typeof(HooksDiscovery<>));`
`@@ -204,7 +204,7 @@ public static void SerializeAsJsonApi(this MvcOptions options, JsonApiOptions js`
`204`	`204`	`/// Adds all required registrations for the service to the container`
`205`	`205`	`/// </summary>`
`206`	`206`	`/// <exception cref="JsonApiSetupException"/>`
`207`		`- public static IServiceCollection AddResourceService<T>(this IServiceCollection services)`
	`207`	`+ public static IServiceCollection AddResourceService<T>(this IServiceCollection services)`
`208`	`208`	`{`
`209`	`209`	`var typeImplementsAnExpectedInterface = false;`
`210`	`210`
`@@ -213,28 +213,29 @@ public static IServiceCollection AddResourceService<T>(this IServiceCollection s`
`213`	`213`	`// it is _possible_ that a single concrete type could be used for multiple resources...`
`214`	`214`	`var resourceDescriptors = GetResourceTypesFromServiceImplementation(serviceImplementationType);`
`215`	`215`
`216`		`- foreach(var resourceDescriptor in resourceDescriptors)`
	`216`	`+ foreach (var resourceDescriptor in resourceDescriptors)`
`217`	`217`	`{`
`218`		`- foreach(var openGenericType in ServiceDiscoveryFacade.ServiceInterfaces)`
	`218`	`+ foreach (var openGenericType in ServiceDiscoveryFacade.ServiceInterfaces)`
`219`	`219`	`{`
`220`	`220`	`// A shorthand interface is one where the id type is ommitted`
`221`	`221`	`// e.g. IResourceService<T> is the shorthand for IResourceService<T, TId>`
`222`	`222`	`var isShorthandInterface = (openGenericType.GetTypeInfo().GenericTypeParameters.Length == 1);`
`223`		`- if(isShorthandInterface && resourceDescriptor.IdType != typeof(int))`
	`223`	`+ if (isShorthandInterface && resourceDescriptor.IdType != typeof(int))`
`224`	`224`	`continue; // we can't create a shorthand for id types other than int`
`225`	`225`
`226`	`226`	`var concreteGenericType = isShorthandInterface`
`227`	`227`	`? openGenericType.MakeGenericType(resourceDescriptor.ResourceType)`
`228`	`228`	`: openGenericType.MakeGenericType(resourceDescriptor.ResourceType, resourceDescriptor.IdType);`
`229`	`229`
`230`		`- if(concreteGenericType.IsAssignableFrom(serviceImplementationType)) {`
	`230`	`+ if (concreteGenericType.IsAssignableFrom(serviceImplementationType))`
	`231`	`+ {`
`231`	`232`	`services.AddScoped(concreteGenericType, serviceImplementationType);`
`232`	`233`	`typeImplementsAnExpectedInterface = true;`
`233`	`234`	`}`
`234`	`235`	`}`
`235`	`236`	`}`
`236`	`237`
`237`		`- if(typeImplementsAnExpectedInterface == false)`
	`238`	`+ if (typeImplementsAnExpectedInterface == false)`
`238`	`239`	`throw new JsonApiSetupException($"{serviceImplementationType} does not implement any of the expected JsonApiDotNetCore interfaces.");`
`239`	`240`
`240`	`241`	`return services;`
`@@ -244,12 +245,12 @@ private static HashSet<ResourceDescriptor> GetResourceTypesFromServiceImplementa`
`244`	`245`	`{`
`245`	`246`	`var resourceDecriptors = new HashSet<ResourceDescriptor>();`
`246`	`247`	`var interfaces = type.GetInterfaces();`
`247`		`- foreach(var i in interfaces)`
	`248`	`+ foreach (var i in interfaces)`
`248`	`249`	`{`
`249`		`- if(i.IsGenericType)`
	`250`	`+ if (i.IsGenericType)`
`250`	`251`	`{`
`251`	`252`	`var firstGenericArgument = i.GenericTypeArguments.FirstOrDefault();`
`252`		`- if(TypeLocator.TryGetResourceDescriptor(firstGenericArgument, out var resourceDescriptor) == true)`
	`253`	`+ if (TypeLocator.TryGetResourceDescriptor(firstGenericArgument, out var resourceDescriptor) == true)`
`253`	`254`	`{`
`254`	`255`	`resourceDecriptors.Add(resourceDescriptor);`
`255`	`256`	`}`
Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@`
`5`	`5`	`using System;`
`6`	`6`	`using System.Collections.ObjectModel;`
`7`	`7`	`using System.Collections.Immutable;`
	`8`	`+using System.Linq.Expressions;`
`8`	`9`
`9`	`10`	`namespace JsonApiDotNetCore.Hooks`
`10`	`11`	`{`
`@@ -26,7 +27,7 @@ public interface IEntityHashSet<TResource> : IByAffectedRelationships<TResource>`
`26`	`27`	`/// </summary>`
`27`	`28`	`public class EntityHashSet<TResource> : HashSet<TResource>, IEntityHashSet<TResource> where TResource : class, IIdentifiable`
`28`	`29`	`{`
`29`		`-`
	`30`	`+`
`30`	`31`
`31`	`32`	`/// <inheritdoc />`
`32`	`33`	`public Dictionary<RelationshipAttribute, HashSet<TResource>> AffectedRelationships { get => _relationships; }`
`@@ -53,9 +54,15 @@ public Dictionary<RelationshipAttribute, HashSet<TResource>> GetByRelationship(T`
`53`	`54`	`}`
`54`	`55`
`55`	`56`	`/// <inheritdoc />`
`56`		`- public Dictionary<RelationshipAttribute, HashSet<TResource>> GetByRelationship<TRelatedResource>() where TRelatedResource : class, IIdentifiable`
	`57`	`+ public Dictionary<RelationshipAttribute, HashSet<TResource>> GetByRelationship<TRelatedResource>() where TRelatedResource : class, IIdentifiable`
`57`	`58`	`{`
`58`	`59`	`return GetByRelationship(typeof(TRelatedResource));`
`59`	`60`	`}`
	`61`	`+`
	`62`	`+ /// <inheritdoc />`
	`63`	`+ public HashSet<TResource> GetAffected(Expression<Func<TResource, object>> NavigationAction)`
	`64`	`+ {`
	`65`	`+ return _relationships.GetAffected(NavigationAction);`
	`66`	`+ }`
`60`	`67`	`}`
`61`	`68`	`}`
Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,7 @@ public virtual IEnumerable<TEntity> BeforeUpdate<TEntity>(IEnumerable<TEntity> e`
`49`	`49`	`{`
`50`	`50`	`var relationships = node.RelationshipsToNextLayer.Select(p => p.Attribute).ToArray();`
`51`	`51`	`var dbValues = LoadDbValues(typeof(TEntity), (IEnumerable<TEntity>)node.UniqueEntities, ResourceHook.BeforeUpdate, relationships);`
`52`		`- var diff = new DiffableEntityHashSet<TEntity>(node.UniqueEntities, dbValues, node.PrincipalsToNextLayer());`
	`52`	`+ var diff = new DiffableEntityHashSet<TEntity>(node.UniqueEntities, dbValues, node.PrincipalsToNextLayer(), _context);`
`53`	`53`	`IEnumerable<TEntity> updated = container.BeforeUpdate(diff, pipeline);`
`54`	`54`	`node.UpdateUnique(updated);`
`55`	`55`	`node.Reassign(entities);`