-
Notifications
You must be signed in to change notification settings - Fork 16
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Implementation for usable order by clause.
- Loading branch information
Showing
30 changed files
with
1,036 additions
and
734 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
17 changes: 17 additions & 0 deletions
17
Musoq.Evaluator/Visitors/RewriteFieldOrderedWithGroupMethodCall.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,17 @@ | ||
using Musoq.Parser.Nodes; | ||
|
||
namespace Musoq.Evaluator.Visitors; | ||
|
||
public class RewriteFieldOrderedWithGroupMethodCall : RewriteFieldWithGroupMethodCallBase<FieldOrderedNode> | ||
{ | ||
public RewriteFieldOrderedWithGroupMethodCall(FieldNode[] nodes) | ||
: base(nodes) | ||
{ | ||
} | ||
|
||
public override void Visit(FieldOrderedNode node) | ||
{ | ||
base.Visit(node); | ||
Expression = Nodes.Pop() as FieldOrderedNode; | ||
} | ||
} |
79 changes: 5 additions & 74 deletions
79
Musoq.Evaluator/Visitors/RewriteFieldWithGroupMethodCall.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,87 +1,18 @@ | ||
using System.Collections.Generic; | ||
using System.Linq; | ||
using Musoq.Evaluator.Helpers; | ||
using Musoq.Parser; | ||
using Musoq.Parser.Nodes; | ||
using Musoq.Parser.Tokens; | ||
using Musoq.Plugins; | ||
using Musoq.Parser.Nodes; | ||
|
||
namespace Musoq.Evaluator.Visitors | ||
{ | ||
public class RewriteFieldWithGroupMethodCall : CloneQueryVisitor | ||
public class RewriteFieldWithGroupMethodCall : RewriteFieldWithGroupMethodCallBase<FieldNode> | ||
{ | ||
private readonly FieldNode[] _fields; | ||
|
||
public RewriteFieldWithGroupMethodCall(FieldNode[] fields) | ||
public RewriteFieldWithGroupMethodCall(FieldNode[] nodes) | ||
: base(nodes) | ||
{ | ||
_fields = fields; | ||
} | ||
|
||
public FieldNode Expression { get; private set; } | ||
|
||
|
||
public override void Visit(FieldNode node) | ||
{ | ||
base.Visit(node); | ||
Expression = Nodes.Pop() as FieldNode; | ||
} | ||
|
||
public override void Visit(AccessColumnNode node) | ||
{ | ||
Nodes.Push(new AccessColumnNode(NamingHelper.ToColumnName(node.Alias, node.Name), string.Empty, | ||
node.ReturnType, TextSpan.Empty)); | ||
} | ||
|
||
public override void Visit(DotNode node) | ||
{ | ||
if (!(node.Root is DotNode) && node.Root is AccessColumnNode column) | ||
{ | ||
Nodes.Pop(); | ||
Nodes.Pop(); | ||
|
||
var name = $"{NamingHelper.ToColumnName(column.Alias, column.Name)}.{node.Expression.ToString()}"; | ||
Nodes.Push(new AccessColumnNode(name, string.Empty, node.ReturnType, TextSpan.Empty)); | ||
return; | ||
} | ||
|
||
base.Visit(node); | ||
} | ||
|
||
public override void Visit(AccessMethodNode node) | ||
{ | ||
if (node.IsAggregateMethod()) | ||
{ | ||
Nodes.Pop(); | ||
|
||
var wordNode = node.Arguments.Args[0] as WordNode; | ||
var accessGroup = new AccessColumnNode("none", string.Empty, typeof(Group), TextSpan.Empty); | ||
var args = new List<Node> {accessGroup, wordNode}; | ||
args.AddRange(node.Arguments.Args.Skip(1)); | ||
var extractFromGroup = new AccessMethodNode( | ||
new FunctionToken(node.Method.Name, TextSpan.Empty), | ||
new ArgsListNode(args.ToArray()), node.ExtraAggregateArguments, node.CanSkipInjectSource, node.Method, node.Alias); | ||
Nodes.Push(extractFromGroup); | ||
} | ||
else if (_fields.Select(f => f.Expression.ToString()).Contains(node.ToString())) | ||
{ | ||
Nodes.Push(new AccessColumnNode(node.ToString(), string.Empty, node.ReturnType, TextSpan.Empty)); | ||
} | ||
else | ||
{ | ||
base.Visit(node); | ||
} | ||
} | ||
|
||
public override void Visit(AccessCallChainNode node) | ||
{ | ||
Nodes.Push(new AccessColumnNode(node.ToString(), string.Empty, node.ReturnType, TextSpan.Empty)); | ||
} | ||
|
||
public override void Visit(CaseNode node) | ||
{ | ||
if (_fields.Select(f => f.Expression.ToString()).Contains(node.ToString())) | ||
Nodes.Push(new AccessColumnNode(node.ToString(), string.Empty, node.ReturnType, TextSpan.Empty)); | ||
else | ||
base.Visit(node); | ||
} | ||
} | ||
} |
Oops, something went wrong.