/// <summary>
/// This is an internal API that supports the Entity Framework Core infrastructure and not subject to
/// the same compatibility standards as public APIs. It may be changed or removed without notice in
/// any release. You should only use it directly in your code with extreme caution and knowing that
/// doing so can result in application failures when updating to a new Entity Framework Core release.
/// </summary>
protected override Expression VisitShapedQuery(ShapedQueryExpression shapedQueryExpression)
{
Check.NotNull(shapedQueryExpression, nameof(shapedQueryExpression));
var jObjectParameter = Expression.Parameter(typeof(JObject), "jObject");
var shaperBody = shapedQueryExpression.ShaperExpression;
shaperBody = new JObjectInjectingExpressionVisitor().Visit(shaperBody);
shaperBody = InjectEntityMaterializers(shaperBody);
switch (shapedQueryExpression.QueryExpression)
{
case SelectExpression selectExpression:
selectExpression.ApplyProjection();
shaperBody = new CosmosProjectionBindingRemovingExpressionVisitor(selectExpression, jObjectParameter, IsTracking)
.Visit(shaperBody);
var shaperLambda = Expression.Lambda(
shaperBody,
QueryCompilationContext.QueryContextParameter,
jObjectParameter);
return(Expression.New(
typeof(QueryingEnumerable <>).MakeGenericType(shaperLambda.ReturnType).GetConstructors()[0],
Expression.Convert(
QueryCompilationContext.QueryContextParameter,
typeof(CosmosQueryContext)),
Expression.Constant(_sqlExpressionFactory),
Expression.Constant(_querySqlGeneratorFactory),
Expression.Constant(selectExpression),
Expression.Constant(shaperLambda.Compile()),
Expression.Constant(_contextType),
Expression.Constant(_partitionKeyFromExtension, typeof(string)),
Expression.Constant(_logger)));
case ReadItemExpression readItemExpression:
shaperBody =
new CosmosProjectionBindingRemovingReadItemExpressionVisitor(readItemExpression, jObjectParameter, IsTracking)
.Visit(shaperBody);
var shaperReadItemLambda = Expression.Lambda(
shaperBody,
QueryCompilationContext.QueryContextParameter,
jObjectParameter);
return(Expression.New(
typeof(ReadItemQueryingEnumerable <>).MakeGenericType(shaperReadItemLambda.ReturnType).GetConstructors()[0],
Expression.Convert(
QueryCompilationContext.QueryContextParameter,
typeof(CosmosQueryContext)),
Expression.Constant(readItemExpression),
Expression.Constant(shaperReadItemLambda.Compile()),
Expression.Constant(_contextType),
Expression.Constant(_logger)));
default:
throw new NotImplementedException();
}
}
protected override Expression VisitShapedQuery(ShapedQueryExpression shapedQueryExpression)
{
var jObjectParameter = Expression.Parameter(typeof(JObject), "jObject");
var shaperBody = shapedQueryExpression.ShaperExpression;
shaperBody = new JObjectInjectingExpressionVisitor().Visit(shaperBody);
shaperBody = InjectEntityMaterializers(shaperBody);
switch (shapedQueryExpression.QueryExpression)
{
case SelectExpression selectExpression:
shaperBody = new CosmosProjectionBindingRemovingExpressionVisitor(
selectExpression, jObjectParameter,
QueryCompilationContext.QueryTrackingBehavior == QueryTrackingBehavior.TrackAll)
.Visit(shaperBody);
var shaperLambda = Expression.Lambda(
shaperBody,
QueryCompilationContext.QueryContextParameter,
jObjectParameter);
return(Expression.New(
typeof(QueryingEnumerable <>).MakeGenericType(shaperLambda.ReturnType).GetConstructors()[0],
Expression.Convert(
QueryCompilationContext.QueryContextParameter,
typeof(CosmosQueryContext)),
Expression.Constant(_sqlExpressionFactory),
Expression.Constant(_querySqlGeneratorFactory),
Expression.Constant(selectExpression),
Expression.Constant(shaperLambda.Compile()),
Expression.Constant(_contextType),
Expression.Constant(_partitionKeyFromExtension, typeof(string)),
Expression.Constant(
QueryCompilationContext.QueryTrackingBehavior == QueryTrackingBehavior.NoTrackingWithIdentityResolution),
Expression.Constant(_threadSafetyChecksEnabled)));
case ReadItemExpression readItemExpression:
shaperBody = new CosmosProjectionBindingRemovingReadItemExpressionVisitor(
readItemExpression, jObjectParameter,
QueryCompilationContext.QueryTrackingBehavior == QueryTrackingBehavior.TrackAll)
.Visit(shaperBody);
var shaperReadItemLambda = Expression.Lambda(
shaperBody,
QueryCompilationContext.QueryContextParameter,
jObjectParameter);
return(Expression.New(
typeof(ReadItemQueryingEnumerable <>).MakeGenericType(shaperReadItemLambda.ReturnType).GetConstructors()[0],
Expression.Convert(
QueryCompilationContext.QueryContextParameter,
typeof(CosmosQueryContext)),
Expression.Constant(readItemExpression),
Expression.Constant(shaperReadItemLambda.Compile()),
Expression.Constant(_contextType),
Expression.Constant(
QueryCompilationContext.QueryTrackingBehavior == QueryTrackingBehavior.NoTrackingWithIdentityResolution),
Expression.Constant(_threadSafetyChecksEnabled)));
default:
throw new NotSupportedException(CoreStrings.UnhandledExpressionNode(shapedQueryExpression.QueryExpression));
}
}
