protected BatchExpression UpdateBatch(BatchExpression batch, Expression input, LambdaExpression operation, Expression batchSize, Expression stream)
{
if (input != batch.Input || operation != batch.Operation || batchSize != batch.BatchSize || stream != batch.Stream)
{
return(new BatchExpression(input, operation, batchSize, stream));
}
return(batch);
}
protected virtual Expression VisitBatch(BatchExpression batch)
{
var operation = (LambdaExpression)Visit(batch.Operation);
var batchSize = Visit(batch.BatchSize);
var stream = Visit(batch.Stream);
return(UpdateBatch(batch, batch.Input, operation, batchSize, stream));
}
protected virtual Expression BuildExecuteBatch(BatchExpression batch)
{
// parameterize query
Expression operation = this.Parameterize(batch.Operation.Body);
string commandText = this.linguist.Format(operation);
var namedValues = NamedValueGatherer.Gather(operation);
QueryCommand command = new QueryCommand(commandText, namedValues.Select(v => new QueryParameter(v.Name, v.Type, v.QueryType)));
Expression[] values = namedValues.Select(v => Expression.Convert(this.Visit(v.Value), typeof(object))).ToArray();
Expression paramSets = Expression.Call(typeof(Enumerable), "Select", new Type[] { batch.Operation.Parameters[1].Type, typeof(object[]) },
batch.Input,
Expression.Lambda(Expression.NewArrayInit(typeof(object), values), new[] { batch.Operation.Parameters[1] })
);
Expression plan = null;
ProjectionExpression projection = ProjectionFinder.FindProjection(operation);
if (projection != null)
{
var saveScope = this.scope;
ParameterExpression reader = Expression.Parameter(typeof(FieldReader), "r" + nReaders++);
this.scope = new Scope(this.scope, reader, projection.Select.Alias, projection.Select.Columns);
LambdaExpression projector = Expression.Lambda(this.Visit(projection.Projector), reader);
this.scope = saveScope;
var entity = EntityFinder.Find(projection.Projector);
command = new QueryCommand(command.CommandText, command.Parameters);
plan = Expression.Call(this.executor, "ExecuteBatch", new Type[] { projector.Body.Type },
Expression.Constant(command),
paramSets,
projector,
Expression.Constant(entity, typeof(MappingEntity)),
batch.BatchSize,
batch.Stream
);
}
else
{
plan = Expression.Call(this.executor, "ExecuteBatch", null,
Expression.Constant(command),
paramSets,
batch.BatchSize,
batch.Stream
);
}
return(plan);
}
protected override Expression VisitBatch(BatchExpression batch)
{
if (this.linguist.Language.AllowsMultipleCommands || !IsMultipleCommands(batch.Operation.Body as CommandExpression))
{
return(this.BuildExecuteBatch(batch));
}
else
{
var source = this.Visit(batch.Input);
var op = this.Visit(batch.Operation.Body);
var fn = Expression.Lambda(op, batch.Operation.Parameters[1]);
return(Expression.Call(this.GetType(), "Batch", new Type[] { TypeHelper.GetElementType(source.Type), batch.Operation.Body.Type }, source, fn, batch.Stream));
}
}
protected virtual Expression VisitBatch(BatchExpression batch)
{
this.Write("Batch(");
this.WriteLine(Indentation.Inner);
this.Visit(batch.Input);
this.Write(",");
this.WriteLine(Indentation.Same);
this.Visit(batch.Operation);
this.Write(",");
this.WriteLine(Indentation.Same);
this.Visit(batch.BatchSize);
this.Write(", ");
this.Visit(batch.Stream);
this.WriteLine(Indentation.Outer);
this.Write(")");
return(batch);
}
protected virtual bool CompareBatch(BatchExpression x, BatchExpression y)
{
return(this.Compare(x.Input, y.Input) && this.Compare(x.Operation, y.Operation) &&
this.Compare(x.BatchSize, y.BatchSize) && this.Compare(x.Stream, y.Stream));
}
