private (NavigationExpansionExpression navigationExpression, Expression nullKeyExpression) CreateNullComparisonArguments( NavigationBindingExpression navigationBindingExpression, NavigationExpansionExpression navigationExpansionExpression) { var navigationKeyAccessExpression = NavigationExpansionHelpers.CreateKeyAccessExpression( navigationBindingExpression, new[] { navigationBindingExpression.EntityType.FindPrimaryKey().Properties.First() }, addNullCheck: true); var nullKeyExpression = NavigationExpansionHelpers.CreateNullKeyExpression( navigationKeyAccessExpression.Type, keyCount: 1); var newNavigationExpansionExpressionState = new NavigationExpansionExpressionState( navigationExpansionExpression.State.CurrentParameter, navigationExpansionExpression.State.SourceMappings, Expression.Lambda(navigationKeyAccessExpression, navigationExpansionExpression.State.PendingSelector.Parameters[0]), applyPendingSelector: true, navigationExpansionExpression.State.PendingOrderings, navigationExpansionExpression.State.PendingIncludeChain, // we need to remap cardinality reducing operator since it's source type has now changed navigationExpansionExpression.State.PendingCardinalityReducingOperator?.GetGenericMethodDefinition().MakeGenericMethod(navigationKeyAccessExpression.Type), navigationExpansionExpression.State.CustomRootMappings, navigationExpansionExpression.State.MaterializeCollectionNavigation); var navigationExpression = new NavigationExpansionExpression( navigationExpansionExpression.Operand, newNavigationExpansionExpressionState, navigationKeyAccessExpression.Type); return(navigationExpression, nullKeyExpression); }
private (Expression Operand, NavigationExpansionExpressionState State) ApplyIncludes( NavigationExpansionExpression navigationExpansionExpression) { var includeVisitor = new PendingSelectorIncludeVisitor(); var rewrittenBody = includeVisitor.Visit(navigationExpansionExpression.State.PendingSelector.Body); if (navigationExpansionExpression.State.PendingSelector.Body != rewrittenBody) { navigationExpansionExpression.State.PendingSelector = Expression.Lambda(rewrittenBody, navigationExpansionExpression.State.PendingSelector.Parameters[0]); navigationExpansionExpression.State.ApplyPendingSelector = true; } if (includeVisitor.PendingIncludes.Count > 0) { var result = (Source : navigationExpansionExpression.Operand, Parameter : navigationExpansionExpression.State.CurrentParameter); foreach (var pendingIncludeNode in includeVisitor.PendingIncludes) { result = NavigationExpansionHelpers.AddNavigationJoin( result.Source, result.Parameter, pendingIncludeNode.SourceMapping, pendingIncludeNode.NavTreeNode, navigationExpansionExpression.State, new List <INavigation>(), include: true); } var pendingSelector = navigationExpansionExpression.State.PendingSelector; if (navigationExpansionExpression.State.CurrentParameter != result.Parameter) { var pendingSelectorBody = new ExpressionReplacingVisitor(navigationExpansionExpression.State.CurrentParameter, result.Parameter).Visit(navigationExpansionExpression.State.PendingSelector.Body); pendingSelector = Expression.Lambda(pendingSelectorBody, result.Parameter); } var newState = new NavigationExpansionExpressionState( result.Parameter, navigationExpansionExpression.State.SourceMappings, pendingSelector, applyPendingSelector: true, navigationExpansionExpression.State.PendingOrderings, navigationExpansionExpression.State.PendingIncludeChain, navigationExpansionExpression.State.PendingCardinalityReducingOperator, navigationExpansionExpression.State.CustomRootMappings, navigationExpansionExpression.State.MaterializeCollectionNavigation); return(Operand : result.Source, newState); } return(navigationExpansionExpression.Operand, navigationExpansionExpression.State); }
protected override Expression VisitExtension(Expression extensionExpression) { if (extensionExpression is NavigationBindingExpression navigationBindingExpression) { if (navigationBindingExpression.RootParameter == _sourceParameter && navigationBindingExpression.NavigationTreeNode.Parent != null && navigationBindingExpression.NavigationTreeNode.IsCollection) { var lastNavigation = navigationBindingExpression.NavigationTreeNode.Navigation; return(lastNavigation.ForeignKey.IsOwnership ? NavigationExpansionHelpers.CreateNavigationExpansionRoot(navigationBindingExpression, lastNavigation.GetTargetType(), lastNavigation) : CreateCollectionNavigationExpression(navigationBindingExpression.NavigationTreeNode, navigationBindingExpression.RootParameter, navigationBindingExpression.SourceMapping)); } else { return(extensionExpression); } } return(base.VisitExtension(extensionExpression)); }
private Expression ProcessMemberPushdown( Expression source, NavigationExpansionExpression navigationExpansionExpression, bool efProperty, MemberInfo memberInfo, string propertyName, Type resultType) { // in case of nested FirstOrDefaults, we need to dig into the inner most - that's where the member finally gets pushed down to if (navigationExpansionExpression.State.PendingSelector.Body is NavigationExpansionExpression navigationExpansionPendingSelector && navigationExpansionPendingSelector.State.PendingCardinalityReducingOperator != null) { var newPendingSelector = (NavigationExpansionExpression)ProcessMemberPushdown(source, navigationExpansionPendingSelector, efProperty, memberInfo, propertyName, resultType); var newStateNested = new NavigationExpansionExpressionState( navigationExpansionExpression.State.CurrentParameter, navigationExpansionExpression.State.SourceMappings, Expression.Lambda(newPendingSelector, navigationExpansionExpression.State.CurrentParameter), applyPendingSelector: true, navigationExpansionExpression.State.PendingOrderings, navigationExpansionExpression.State.PendingIncludeChain, // we need to remap cardinality reducing operator since it's source type has now changed navigationExpansionExpression.State.PendingCardinalityReducingOperator.GetGenericMethodDefinition().MakeGenericMethod(newPendingSelector.Type), navigationExpansionExpression.State.CustomRootMappings, navigationExpansionExpression.State.MaterializeCollectionNavigation); return(new NavigationExpansionExpression( navigationExpansionExpression.Operand, newStateNested, resultType)); } var selectorParameter = Expression.Parameter(source.Type, navigationExpansionExpression.State.CurrentParameter.Name); var selectorBody = efProperty ? (Expression)Expression.Call(EF.PropertyMethod.MakeGenericMethod(resultType), selectorParameter, Expression.Constant(propertyName)) : Expression.MakeMemberAccess(selectorParameter, memberInfo); if (navigationExpansionExpression.State.PendingCardinalityReducingOperator.MethodIsClosedFormOf(LinqMethodHelpers.QueryableFirstOrDefaultMethodInfo) || navigationExpansionExpression.State.PendingCardinalityReducingOperator.MethodIsClosedFormOf(LinqMethodHelpers.QueryableFirstOrDefaultPredicateMethodInfo) || navigationExpansionExpression.State.PendingCardinalityReducingOperator.MethodIsClosedFormOf(LinqMethodHelpers.QueryableSingleOrDefaultMethodInfo) || navigationExpansionExpression.State.PendingCardinalityReducingOperator.MethodIsClosedFormOf(LinqMethodHelpers.QueryableSingleOrDefaultPredicateMethodInfo) || navigationExpansionExpression.State.PendingCardinalityReducingOperator.MethodIsClosedFormOf(LinqMethodHelpers.EnumerableFirstOrDefaultMethodInfo) || navigationExpansionExpression.State.PendingCardinalityReducingOperator.MethodIsClosedFormOf(LinqMethodHelpers.EnumerableFirstOrDefaultPredicateMethodInfo) || navigationExpansionExpression.State.PendingCardinalityReducingOperator.MethodIsClosedFormOf(LinqMethodHelpers.EnumerableSingleOrDefaultMethodInfo) || navigationExpansionExpression.State.PendingCardinalityReducingOperator.MethodIsClosedFormOf(LinqMethodHelpers.EnumerableSingleOrDefaultPredicateMethodInfo)) { if (!selectorBody.Type.IsNullableType()) { selectorBody = Expression.Convert(selectorBody, selectorBody.Type.MakeNullable()); } } var selector = Expression.Lambda(selectorBody, selectorParameter); var remappedSelectorBody = ReplacingExpressionVisitor.Replace( selectorParameter, navigationExpansionExpression.State.PendingSelector.Body, selector.Body); var binder = new NavigationPropertyBindingVisitor( navigationExpansionExpression.State.CurrentParameter, navigationExpansionExpression.State.SourceMappings); var boundSelectorBody = binder.Visit(remappedSelectorBody); if (boundSelectorBody is NavigationBindingExpression navigationBindingExpression && navigationBindingExpression.NavigationTreeNode.Navigation != null) { if (navigationBindingExpression.NavigationTreeNode.IsCollection) { var lastNavigation = navigationBindingExpression.NavigationTreeNode.Navigation; var collectionNavigationElementType = lastNavigation.ForeignKey.DeclaringEntityType.ClrType; var entityQueryable = NullAsyncQueryProvider.Instance.CreateEntityQueryableExpression(collectionNavigationElementType); var outerParameter = Expression.Parameter(collectionNavigationElementType, collectionNavigationElementType.GenerateParameterName()); var outerKeyAccess = NavigationExpansionHelpers.CreateKeyAccessExpression( outerParameter, lastNavigation.ForeignKey.Properties); var innerParameter = Expression.Parameter(navigationExpansionExpression.Type); var innerKeyAccessLambda = Expression.Lambda( NavigationExpansionHelpers.CreateKeyAccessExpression( innerParameter, lastNavigation.ForeignKey.PrincipalKey.Properties), innerParameter); var combinedKeySelectorBody = ReplacingExpressionVisitor.Replace( innerKeyAccessLambda.Parameters[0], navigationExpansionExpression.State.PendingSelector.Body, innerKeyAccessLambda.Body); if (outerKeyAccess.Type != combinedKeySelectorBody.Type) { if (combinedKeySelectorBody.Type.IsNullableType()) { outerKeyAccess = Expression.Convert(outerKeyAccess, combinedKeySelectorBody.Type); } else { combinedKeySelectorBody = Expression.Convert(combinedKeySelectorBody, outerKeyAccess.Type); } } var rewrittenState = new NavigationExpansionExpressionState( navigationExpansionExpression.State.CurrentParameter, navigationExpansionExpression.State.SourceMappings, Expression.Lambda(combinedKeySelectorBody, navigationExpansionExpression.State.CurrentParameter), applyPendingSelector: true, navigationExpansionExpression.State.PendingOrderings, navigationExpansionExpression.State.PendingIncludeChain, // we need to remap cardinality reducing operator since it's source type has now changed navigationExpansionExpression.State.PendingCardinalityReducingOperator.GetGenericMethodDefinition().MakeGenericMethod(combinedKeySelectorBody.Type), navigationExpansionExpression.State.CustomRootMappings, materializeCollectionNavigation: null); var rewrittenNavigationExpansionExpression = new NavigationExpansionExpression(navigationExpansionExpression.Operand, rewrittenState, combinedKeySelectorBody.Type); var inner = new NavigationExpansionReducingVisitor().Visit(rewrittenNavigationExpansionExpression); var predicate = Expression.Lambda( Expression.Equal(outerKeyAccess, inner), outerParameter); var whereMethodInfo = LinqMethodHelpers.QueryableWhereMethodInfo.MakeGenericMethod(collectionNavigationElementType); var rewritten = Expression.Call( whereMethodInfo, entityQueryable, predicate); var entityType = lastNavigation.ForeignKey.DeclaringEntityType; return(NavigationExpansionHelpers.CreateNavigationExpansionRoot(rewritten, entityType, materializeCollectionNavigation: null)); } else { return(ProcessSelectCore( navigationExpansionExpression.Operand, navigationExpansionExpression.State, selector, selectorBody.Type)); } } var newState = new NavigationExpansionExpressionState( navigationExpansionExpression.State.CurrentParameter, navigationExpansionExpression.State.SourceMappings, Expression.Lambda(boundSelectorBody, navigationExpansionExpression.State.CurrentParameter), applyPendingSelector: true, navigationExpansionExpression.State.PendingOrderings, navigationExpansionExpression.State.PendingIncludeChain, // we need to remap cardinality reducing operator since it's source type has now changed navigationExpansionExpression.State.PendingCardinalityReducingOperator.GetGenericMethodDefinition().MakeGenericMethod(boundSelectorBody.Type), navigationExpansionExpression.State.CustomRootMappings, navigationExpansionExpression.State.MaterializeCollectionNavigation); // TODO: expand navigations var result = new NavigationExpansionExpression( navigationExpansionExpression.Operand, newState, selectorBody.Type); return(resultType != result.Type ? (Expression)Expression.Convert(result, resultType) : result); }
private Expression TryRewriteNavigationComparison(Expression left, Expression right, bool equality) { var leftBinding = left as NavigationBindingExpression; var rightBinding = right as NavigationBindingExpression; var leftNullConstant = left.IsNullConstantExpression(); var rightNullConstant = right.IsNullConstantExpression(); Expression newLeft = null; Expression newRight = null; // comparing two different collection navigations always returns false if (leftBinding != null && rightBinding != null && leftBinding.NavigationTreeNode.Navigation != rightBinding.NavigationTreeNode.Navigation && (leftBinding.NavigationTreeNode.IsCollection || rightBinding.NavigationTreeNode.IsCollection)) { if (leftBinding.NavigationTreeNode.IsCollection) { var parentTreeNode = leftBinding.NavigationTreeNode.Parent; parentTreeNode.Children.Remove(leftBinding.NavigationTreeNode); } if (rightBinding.NavigationTreeNode.IsCollection) { var parentTreeNode = rightBinding.NavigationTreeNode.Parent; parentTreeNode.Children.Remove(rightBinding.NavigationTreeNode); } return(Expression.Constant(false)); } if (leftBinding != null && rightBinding != null && leftBinding.EntityType == rightBinding.EntityType) { if (leftBinding.NavigationTreeNode.Navigation == rightBinding.NavigationTreeNode.Navigation && leftBinding.NavigationTreeNode.IsCollection) { leftBinding = CreateParentBindingExpression(leftBinding); rightBinding = CreateParentBindingExpression(rightBinding); } // TODO: what about entities without PKs? var primaryKeyProperties = leftBinding.EntityType.FindPrimaryKey().Properties; newLeft = NavigationExpansionHelpers.CreateKeyAccessExpression(leftBinding, primaryKeyProperties, addNullCheck: leftBinding.NavigationTreeNode.Optional); newRight = NavigationExpansionHelpers.CreateKeyAccessExpression(rightBinding, primaryKeyProperties, addNullCheck: rightBinding.NavigationTreeNode.Optional); } if (leftBinding != null && rightNullConstant) { if (leftBinding.NavigationTreeNode.IsCollection) { leftBinding = CreateParentBindingExpression(leftBinding); } // TODO: what about entities without PKs? var primaryKeyProperties = leftBinding.EntityType.FindPrimaryKey().Properties; newLeft = NavigationExpansionHelpers.CreateKeyAccessExpression(leftBinding, primaryKeyProperties, addNullCheck: leftBinding.NavigationTreeNode.Optional); newRight = NavigationExpansionHelpers.CreateNullKeyExpression(newLeft.Type, primaryKeyProperties.Count); } if (rightBinding != null && leftNullConstant) { if (rightBinding.NavigationTreeNode.IsCollection) { rightBinding = CreateParentBindingExpression(rightBinding); } // TODO: what about entities without PKs? var primaryKeyProperties = rightBinding.EntityType.FindPrimaryKey().Properties; newRight = NavigationExpansionHelpers.CreateKeyAccessExpression(rightBinding, primaryKeyProperties, addNullCheck: rightBinding.NavigationTreeNode.Optional); newLeft = NavigationExpansionHelpers.CreateNullKeyExpression(newRight.Type, primaryKeyProperties.Count); } if (newLeft == null || newRight == null) { return(null); } if (newLeft.Type != newRight.Type) { if (newLeft.Type.IsNullableType()) { newRight = Expression.Convert(newRight, newLeft.Type); } else { newLeft = Expression.Convert(newLeft, newRight.Type); } } return(equality ? Expression.Equal(newLeft, newRight) : Expression.NotEqual(newLeft, newRight)); }
public static Expression CreateCollectionNavigationExpression( NavigationTreeNode navigationTreeNode, ParameterExpression rootParameter, SourceMapping sourceMapping) { var collectionEntityType = navigationTreeNode.Navigation.ForeignKey.DeclaringEntityType; Expression operand; if (navigationTreeNode.IncludeState == NavigationState.Pending || navigationTreeNode.ExpansionState == NavigationState.Pending) { var entityQueryable = (Expression)NullAsyncQueryProvider.Instance.CreateEntityQueryableExpression(collectionEntityType.ClrType); var outerBinding = new NavigationBindingExpression( rootParameter, navigationTreeNode.Parent, navigationTreeNode.Navigation.DeclaringEntityType, sourceMapping, navigationTreeNode.Navigation.DeclaringEntityType.ClrType); var outerKeyAccess = NavigationExpansionHelpers.CreateKeyAccessExpression( outerBinding, navigationTreeNode.Navigation.ForeignKey.PrincipalKey.Properties, addNullCheck: outerBinding.NavigationTreeNode.Optional); var collectionCurrentParameter = Expression.Parameter( collectionEntityType.ClrType, collectionEntityType.ClrType.GenerateParameterName()); var innerKeyAccess = NavigationExpansionHelpers.CreateKeyAccessExpression( collectionCurrentParameter, navigationTreeNode.Navigation.ForeignKey.Properties); var predicate = Expression.Lambda( CreateKeyComparisonExpressionForCollectionNavigationSubquery( outerKeyAccess, innerKeyAccess, outerBinding), collectionCurrentParameter); operand = Expression.Call( LinqMethodHelpers.QueryableWhereMethodInfo.MakeGenericMethod(collectionEntityType.ClrType), entityQueryable, predicate); } else { operand = new NavigationBindingExpression( rootParameter, navigationTreeNode, collectionEntityType, sourceMapping, collectionEntityType.ClrType); } var result = NavigationExpansionHelpers.CreateNavigationExpansionRoot( operand, collectionEntityType, navigationTreeNode.Navigation); // this is needed for cases like: root.Include(r => r.Collection).ThenInclude(c => c.Reference).Select(r => r.Collection) // result should be elements of the collection navigation with their 'Reference' included var newSourceMapping = result.State.SourceMappings.Single(); IncludeHelpers.CopyIncludeInformation(navigationTreeNode, newSourceMapping.NavigationTree, newSourceMapping); return(result); }