private IGraphQueryStep BuildQueryPlanForRecursiveEdge(IGraphQueryStep left, int index, PatternMatchElementExpression patternExpression)
{
var recursive = patternExpression.Path[index].Recursive.Value;
var pattern = recursive.Pattern;
var steps = new List <SingleEdgeMatcher>((pattern.Count + 1) / 2);
for (int i = 0; i < pattern.Count; i += 2)
{
if (GraphQuery.WithEdgePredicates.TryGetValue(pattern[i].Alias, out var recursiveEdge) == false)
{
throw new InvalidOperationException($"BuildQueryPlanForEdge was invoked for recursive alias='{pattern[i].Alias}' which suppose to be an edge but no corresponding WITH EDGE clause was found.");
}
steps.Add(new SingleEdgeMatcher
{
IncludedEdges = new Dictionary <string, Sparrow.Json.BlittableJsonReaderObject>(StringComparer.OrdinalIgnoreCase),
QueryParameters = _query.QueryParameters,
Edge = recursiveEdge,
Results = new List <Match>(),
Right = i + 1 < pattern.Count ? BuildQueryPlanForMatchNode(pattern[i + 1]) : null,
EdgeAlias = pattern[i].Alias
});
}
var recursiveStep = new RecursionQueryStep(left, steps, recursive, recursive.GetOptions(_query.Metadata, _query.QueryParameters), _token)
{
CollectIntermediateResults = CollectIntermediateResults
};
if (index + 1 < patternExpression.Path.Length)
{
if (patternExpression.Path[index + 1].Recursive.HasValue)
{
throw new InvalidQueryException("Two adjacent 'recursive' queries are not allowed", GraphQuery.QueryText);
}
if (patternExpression.Path[index + 1].IsEdge)
{
var nextPlan = BuildQueryPlanForPattern(patternExpression, index + 2);
nextPlan = BuildQueryPlanForEdge(recursiveStep, nextPlan, patternExpression.Path[index + 1]);
recursiveStep.SetNext(nextPlan.GetSingleGraphStepExecution());
}
else
{
var nextPlan = BuildQueryPlanForPattern(patternExpression, index + 1);
recursiveStep.SetNext(nextPlan.GetSingleGraphStepExecution());
}
}
return(recursiveStep);
}
public override IGraphQueryStep VisitRecursionQueryStep(RecursionQueryStep rqs)
{
_token.ThrowIfCancellationRequested();
var left = Visit(rqs.Left);
bool modified = ReferenceEquals(left, rqs.Left) == false;
var steps = new List <SingleEdgeMatcher>();
foreach (var step in rqs.Steps)
{
_isVisitingRight = true;
var right = Visit(step.Right);
_isVisitingRight = false;
if (ReferenceEquals(right, step.Right) == false)
{
modified = true;
steps.Add(new SingleEdgeMatcher(step, right, _documentsStorage.DocumentDatabase.IdentityPartsSeparator));
}
else
{
steps.Add(step);
}
}
var next = rqs.GetNextStep();
Visit(next);
if (modified == false)
{
return(rqs);
}
var result = new RecursionQueryStep(rqs, left, steps, _token);
if (next != null)
{
next.SetPrev(result);
result.SetNext(next);
result.SetAliases(rqs.GetAllAliases());
}
return(result);
}