private static void LiftOrderBy(
SelectExpression innerJoinSelectExpression,
SelectExpression targetSelectExpression,
TableExpressionBase innerJoinExpression)
{
foreach (var ordering in innerJoinSelectExpression.OrderBy)
{
var orderingExpression = ordering.Expression;
var aliasExpression = ordering.Expression as AliasExpression;
if (aliasExpression?.Alias != null)
{
var columnExpression = aliasExpression.TryGetColumnExpression();
if (columnExpression != null)
{
orderingExpression
= new ColumnExpression(
aliasExpression.Alias,
columnExpression.Property,
columnExpression.Table);
}
}
var index = innerJoinSelectExpression.AddToProjection(orderingExpression);
var expression = innerJoinSelectExpression.Projection[index];
var newExpression
= targetSelectExpression.UpdateColumnExpression(expression, innerJoinExpression);
targetSelectExpression.AddToOrderBy(new Ordering(newExpression, ordering.OrderingDirection));
}
innerJoinSelectExpression.ClearOrderBy();
}
private Expression VisitSelectExpression(SelectExpression selectExpression)
{
base.Visit(selectExpression);
if (!RequiresRowNumberPaging(selectExpression))
{
return selectExpression;
}
var subQuery = selectExpression.PushDownSubquery();
foreach (var projection in subQuery.Projection)
{
var alias = projection as AliasExpression;
var column = projection as ColumnExpression;
if (column != null)
{
column = new ColumnExpression(column.Name, column.Property, subQuery);
selectExpression.AddToProjection(column);
continue;
}
column = alias?.TryGetColumnExpression();
if (column != null)
{
column = new ColumnExpression(column.Name, column.Property, subQuery);
alias = new AliasExpression(alias.Alias, column);
selectExpression.AddToProjection(alias);
}
else
{
column = new ColumnExpression(alias.Alias, alias.Expression.Type, subQuery);
selectExpression.AddToProjection(column);
}
}
if (subQuery.OrderBy.Count == 0)
{
subQuery.AddToOrderBy(
new Ordering(new SqlFunctionExpression("@@RowCount", typeof(int)), OrderingDirection.Asc));
}
var columnExpression = new ColumnExpression(RowNumberColumnName, typeof(int), subQuery);
var rowNumber = new RowNumberExpression(columnExpression, subQuery.OrderBy);
subQuery.ClearOrderBy();
subQuery.AddToProjection(rowNumber, false);
Expression predicate = null;
var offset = subQuery.Offset ?? 0;
if (subQuery.Offset.HasValue)
{
predicate = Expression.GreaterThan(columnExpression, Expression.Constant(offset));
}
if (subQuery.Limit.HasValue)
{
var exp = Expression.LessThanOrEqual(columnExpression, Expression.Constant(offset + subQuery.Limit.Value));
if (predicate != null)
{
exp = Expression.AndAlso(predicate, exp);
}
predicate = exp;
}
selectExpression.Predicate = predicate;
return selectExpression;
}