/// <summary>
/// Prepares the join tree.
/// </summary>
private JoinTrie PrepareJoin()
{
// construct the join tree
var allPaths = new List <string[]>();
if (_select != null)
{
allPaths.AddRange(_select.ColumnAccesses().Select(e => e.Path));
}
if (_orderby != null)
{
allPaths.AddRange(_orderby.ColumnAccesses().Select(e => e.Path));
}
if (_filter != null)
{
allPaths.AddRange(_filter.ColumnAccesses().Select(e => e.Path));
}
// This will represent the mapping from paths to symbols
var joinTree = JoinTrie.Make(TypeDescriptor.Get <T>(), allPaths);
return(joinTree);
}
/// <summary>
/// Create a <see cref="SqlEntityStatement"/> that contains all the needed information to execute the query
/// as an INNER JOIN of any one of the other queries that uses it as a principal query
/// IMPORTANT: Calling this method will keep a permanent cache of some parts of the result, therefore
/// if the arguments need to change after that, a new <see cref="EntityQueryInternal"/> must be created
/// </summary>
private string PrepareStatementAsPrincipal(
Func <Type, string> sources,
SqlStatementVariables vars,
SqlStatementParameters ps,
bool isAncestorExpand,
ArraySegment <string> pathToCollectionProperty,
int userId,
DateTime?userToday,
DateTimeOffset?userNow)
{
// (1) Prepare the JOIN's clause
JoinTrie joinTrie = PrepareJoin(pathToCollectionProperty);
var joinSql = joinTrie.GetSql(sources);
// Compilation context
var today = userToday ?? DateTime.Today;
var now = userNow ?? DateTimeOffset.Now;
var ctx = new QxCompilationContext(joinTrie, sources, vars, ps, today, now, userId);
// (2) Prepare the SELECT clause
SqlSelectClause selectClause = PrepareSelectAsPrincipal(joinTrie, pathToCollectionProperty, isAncestorExpand);
var selectSql = selectClause.ToSql(IsAncestorExpand);
// (3) Prepare the inner join with the principal query (if any)
string principalQuerySql = PreparePrincipalQuerySql(ctx);
// (4) Prepare the WHERE clause
string whereSql = PrepareWhereSql(ctx);
// (5) Prepare the ORDERBY clause
string orderbySql = PrepareOrderBySql(ctx);
// (6) Prepare the OFFSET and FETCH clauses
string offsetFetchSql = PrepareOffsetFetch();
if (string.IsNullOrWhiteSpace(offsetFetchSql))
{
// In a principal query, order by is only added if there is an offset-fetch (usually in the root query)
orderbySql = "";
}
// (7) Finally put together the final SQL statement and return it
string sql = QueryTools.CombineSql(
selectSql: selectSql,
joinSql: joinSql,
principalQuerySql: principalQuerySql,
whereSql: whereSql,
orderbySql: orderbySql,
offsetFetchSql: offsetFetchSql,
groupbySql: null,
havingSql: null,
selectFromTempSql: null
);
// (8) Return the result
return(sql);
}
/// <summary>
/// Create the <see cref="JoinTrie"/> from the paths in all the arguments
/// </summary>
private JoinTrie PrepareJoin(ArraySegment <string>?pathToCollection = null)
{
// construct the join tree
var allPaths = new List <string[]>();
if (Select != null)
{
allPaths.AddRange(Select.Select(e => e.Path));
}
if (Expand != null)
{
allPaths.AddRange(Expand.Select(e => e.Path));
}
if (Filter != null)
{
allPaths.AddRange(Filter.ColumnAccesses().Select(e => e.Path));
}
if (OrderBy != null)
{
allPaths.AddRange(OrderBy.ColumnAccesses().Select(e => e.Path));
}
if (pathToCollection != null)
{
var pathToCollectionEntity = new ArraySegment <string>(
pathToCollection.Value.Array,
pathToCollection.Value.Offset,
pathToCollection.Value.Count - 1);
allPaths.Add(pathToCollectionEntity.ToArray());
}
// This will represent the mapping from paths to symbols
return(JoinTrie.Make(ResultDescriptor, allPaths));
}
/// <summary>
/// Prepares the SELECT statement and the column map, as it would appear in the INNER JOIN of another query that relies on this as its principal
/// </summary>
private static SqlSelectClause PrepareSelectAsPrincipal(JoinTrie joinTree, ArraySegment <string> pathToCollection, bool isAncestorExpand)
{
// Take the segment without the last item
var pathToCollectionEntity = new ArraySegment <string>(
pathToCollection.Array,
pathToCollection.Offset,
pathToCollection.Count - 1);
string symbol = joinTree[pathToCollectionEntity]?.Symbol;
if (string.IsNullOrWhiteSpace(symbol))
{
// Developer mistake
throw new InvalidOperationException($"Could not find the path {string.Join(".", pathToCollectionEntity)} in the joinTree");
}
var columns = new List <(string Symbol, ArraySegment <string> Path, string PropName)>
{
(symbol, pathToCollectionEntity, isAncestorExpand ? "Node" : "Id")
};
return(new SqlSelectClause(columns));
}
/// <summary>
/// Prepares the SELECT statement and the column map, using the <see cref="Select"/> argument
/// </summary>
private SqlSelectClause PrepareSelect(JoinTrie joinTree)
{
var selects = new HashSet <(string Symbol, string PropName)>();
var columns = new List <(string Symbol, ArraySegment <string> Path, string PropName)>();
void AddSelect(string symbol, ArraySegment <string> path, string propName)
{
// NULL happens when there is a select that has been segmented from the middle
// and the first section of the segment no longer terminates with a simple property
// propName = propName ?? "Id";
if (propName == null)
{
return;
}
if (selects.Add((symbol, propName)))
{
columns.Add((symbol, path, propName));
}
}
// Any path step that is touched by a select (which has a property) ignores the expand, the joinTree below
// allows us to efficiently check if any particular step is touched by a select
JoinTrie overridingSelectTree = Select == null ? null : JoinTrie.Make(ResultDescriptor, Select.Select(e => e.Path)); // Overriding select paths
// Optimization: remember the joins that have been selected and don't select them again
var selectedJoins = new HashSet <JoinTrie>();
// For every expanded entity that has not been tainted by a select argument, we add all its properties to the list of selects
Expand ??= ExpressionExpand.Empty;
foreach (var expand in Expand.Union(ExpressionExpand.RootSingleton))
{
string[] path = expand.Path;
for (int i = 0; i <= path.Length; i++)
{
var subpath = new ArraySegment <string>(path, 0, i);
var selectMatch = overridingSelectTree?[subpath];
if (selectMatch == null) // This expand is not overridden by a select
{
var join = joinTree[subpath];
if (join == null)
{
// Developer mistake
throw new InvalidOperationException($"The path '{string.Join('.', subpath)}' was not found in the joinTree");
}
else if (selectedJoins.Contains(join))
{
continue;
}
else
{
selectedJoins.Add(join);
}
foreach (var prop in join.EntityDescriptor.SimpleProperties)
{
AddSelect(join.Symbol, subpath, prop.Name);
}
}
}
}
if (Select != null)
{
foreach (var select in Select)
{
// Add the property
string[] path = select.Path;
{
var join = joinTree[path];
var propName = select.Property; // Can be null
AddSelect(join.Symbol, path, propName);
}
// In this loop we ensure all levels to the selected properties
// have their Ids and Foreign Keys added to the select collection
for (int i = 0; i <= path.Length; i++)
{
var subpath = new ArraySegment <string>(path, 0, i);
var join = joinTree[subpath];
if (join == null)
{
// Developer mistake
throw new InvalidOperationException($"The path '{string.Join('.', subpath)}' was not found in the joinTree");
}
else if (selectedJoins.Contains(join))
{
// All properties were added earlier in an expand
continue;
}
else
{
selectedJoins.Add(join);
}
// The Id is ALWAYS required in every EntityWithKey
if (join.EntityDescriptor.HasId)
{
AddSelect(join.Symbol, subpath, "Id");
}
// Add all the foreign keys to the next level down
foreach (var nextJoin in join.Values)
{
AddSelect(join.Symbol, subpath, nextJoin.ForeignKeyName);
}
}
}
}
// If the foreign key to the principal query is specified, then always include that
// otherwise there will be no way to link the collection to the principal query once we load the data
if (!string.IsNullOrWhiteSpace(ForeignKeyToPrincipalQuery))
{
var path = Array.Empty <string>();
AddSelect(joinTree.Symbol, path, ForeignKeyToPrincipalQuery);
}
// Deals with trees
foreach (var path in PathsToParentEntitiesWithExpandedAncestors)
{
var join = joinTree[path];
AddSelect(join.Symbol, path, "ParentId");
}
if (IsAncestorExpand)
{
var path = Array.Empty <string>();
AddSelect(joinTree.Symbol, path, "ParentId");
}
// Change the hash set to a list so that the order is well defined
return(new SqlSelectClause(columns));
}
public async Task <DynamicOutput> ToListAsync(QueryContext ctx, CancellationToken cancellation)
{
var queryArgs = await _factory(cancellation);
var sources = queryArgs.Sources;
var connString = queryArgs.ConnectionString;
var loader = queryArgs.Loader;
var userId = ctx.UserId;
var userToday = ctx.UserToday;
// ------------------------ Validation Step
// SELECT Validation
if (_select == null)
{
string message = $"The select argument is required";
throw new InvalidOperationException(message);
}
// Make sure that measures are well formed: every column access is wrapped inside an aggregation function
foreach (var exp in _select)
{
if (exp.ContainsAggregations) // This is a measure
{
// Every column access must descend from an aggregation function
var exposedColumnAccess = exp.UnaggregatedColumnAccesses().FirstOrDefault();
if (exposedColumnAccess != null)
{
throw new QueryException($"Select parameter contains a measure with a column access {exposedColumnAccess} that is not included within an aggregation.");
}
}
}
// ORDER BY Validation
if (_orderby != null)
{
foreach (var exp in _orderby)
{
// Order by cannot be a constant
if (!exp.ContainsAggregations && !exp.ContainsColumnAccesses)
{
throw new QueryException("OrderBy parameter cannot be a constant, every orderby expression must contain either an aggregation or a column access.");
}
}
}
// FILTER Validation
if (_filter != null)
{
var conditionWithAggregation = _filter.Expression.Aggregations().FirstOrDefault();
if (conditionWithAggregation != null)
{
throw new QueryException($"Filter contains a condition with an aggregation function: {conditionWithAggregation}");
}
}
// HAVING Validation
if (_having != null)
{
// Every column access must descend from an aggregation function
var exposedColumnAccess = _having.Expression.UnaggregatedColumnAccesses().FirstOrDefault();
if (exposedColumnAccess != null)
{
throw new QueryException($"Having parameter contains a column access {exposedColumnAccess} that is not included within an aggregation.");
}
}
// ------------------------ Preparation Step
// If all is good Prepare some universal variables and parameters
var vars = new SqlStatementVariables();
var ps = new SqlStatementParameters();
var today = userToday ?? DateTime.Today;
var now = DateTimeOffset.Now;
// ------------------------ Tree Analysis Step
// By convention if A.B.Id AND A.B.ParentId are both in the select expression,
// then this is a tree dimension and we return all the ancestors of A.B,
// What do we select for the ancestors? All non-aggregated expressions in
// the original select that contain column accesses exclusively starting with A.B
var additionalNodeSelects = new List <QueryexColumnAccess>();
var ancestorsStatements = new List <SqlDimensionAncestorsStatement>();
{
// Find all column access atoms that terminate with ParentId, those are the potential tree dimensions
var parentIdSelects = _select
.Where(e => e is QueryexColumnAccess ca && ca.Property == "ParentId")
.Cast <QueryexColumnAccess>();
foreach (var parentIdSelect in parentIdSelects)
{
var pathToTreeEntity = parentIdSelect.Path; // A.B
// Confirm it's a tree dimension
var idSelect = _select.FirstOrDefault(e => e is QueryexColumnAccess ca && ca.Property == "Id" && ca.PathEquals(pathToTreeEntity));
if (idSelect != null)
{
// Prepare the Join Trie
var treeType = TypeDescriptor.Get <T>();
foreach (var step in pathToTreeEntity)
{
treeType = treeType.NavigationProperty(step)?.TypeDescriptor ??
throw new QueryException($"Property {step} does not exist on type {treeType.Name}.");
}
// Create or Get the name of the Node column
string nodeColumnName = NodeColumnName(additionalNodeSelects.Count);
additionalNodeSelects.Add(new QueryexColumnAccess(pathToTreeEntity, "Node")); // Tell the principal query to include this node
// Get all atoms that contain column accesses exclusively starting with A.B
var principalSelectsWithMatchingPrefix = _select
.Where(exp => exp.ColumnAccesses().All(ca => ca.PathStartsWith(pathToTreeEntity)));
// Calculate the target indices
var targetIndices = principalSelectsWithMatchingPrefix
.Select(exp => SelectIndexDictionary[exp]);
// Remove the prefix from all column accesses
var ancestorSelects = principalSelectsWithMatchingPrefix
.Select(exp => exp.Clone(prefixToRemove: pathToTreeEntity));
var allPaths = ancestorSelects.SelectMany(e => e.ColumnAccesses()).Select(e => e.Path);
var joinTrie = JoinTrie.Make(treeType, allPaths);
var joinSql = joinTrie.GetSql(sources);
// Prepare the Context
var complicationCtx = new QxCompilationContext(joinTrie, sources, vars, ps, today, now, userId);
// Prepare the SQL components
var selectSql = PrepareAncestorSelectSql(complicationCtx, ancestorSelects);
var principalQuerySql = PreparePrincipalQuerySql(nodeColumnName);
// Combine the SQL components
string sql = QueryTools.CombineSql(
selectSql: selectSql,
joinSql: joinSql,
principalQuerySql: principalQuerySql,
whereSql: null,
orderbySql: null,
offsetFetchSql: null,
groupbySql: null,
havingSql: null,
selectFromTempSql: null);
// Get the index of the id select
int idIndex = SelectIndexDictionary[idSelect];
// Create and add the statement object
var statement = new SqlDimensionAncestorsStatement(idIndex, sql, targetIndices);
ancestorsStatements.Add(statement);
}
}
}
// ------------------------ The SQL Generation Step
// (1) Prepare the JOIN's clause
var principalJoinTrie = PreparePrincipalJoin();
var principalJoinSql = principalJoinTrie.GetSql(sources);
// Compilation context
var principalCtx = new QxCompilationContext(principalJoinTrie, sources, vars, ps, today, now, userId);
// (2) Prepare all the SQL clauses
var(principalSelectSql, principalGroupbySql, principalColumnCount) = PreparePrincipalSelectAndGroupBySql(principalCtx, additionalNodeSelects);
string principalWhereSql = PreparePrincipalWhereSql(principalCtx);
string principalHavingSql = PreparePrincipalHavingSql(principalCtx);
string principalOrderbySql = PreparePrincipalOrderBySql();
string principalSelectFromTempSql = PrepareSelectFromTempSql();
// (3) Put together the final SQL statement
string principalSql = QueryTools.CombineSql(
selectSql: principalSelectSql,
joinSql: principalJoinSql,
principalQuerySql: null,
whereSql: principalWhereSql,
orderbySql: principalOrderbySql,
offsetFetchSql: null,
groupbySql: principalGroupbySql,
havingSql: principalHavingSql,
selectFromTempSql: principalSelectFromTempSql
);
// ------------------------ Execute SQL and return Output
var principalStatement = new SqlDynamicStatement(principalSql, principalColumnCount);
var args = new DynamicLoaderArguments
{
CountSql = null, // No counting in aggregate functions
PrincipalStatement = principalStatement,
DimensionAncestorsStatements = ancestorsStatements,
Variables = vars,
Parameters = ps,
};
var output = await loader.LoadDynamic(connString, args, cancellation);
return(output);
}
