Example 1 with OrderBy
use of io.trino.sql.tree.OrderBy in project trino by trinodb.
the class FunctionCallProvider method getExpectedValue.
@Override
public FunctionCall getExpectedValue(SymbolAliases aliases) {
List<Expression> symbolReferences = toSymbolReferences(args, aliases);
if (isWindowFunction) {
verify(!distinct, "window does not support distinct");
verify(orderBy.isEmpty(), "window does not support order by");
return new ExpectedWindowFunctionCall(symbolReferences);
}
Optional<OrderBy> orderByClause = Optional.empty();
if (!orderBy.isEmpty()) {
orderByClause = Optional.of(new OrderBy(orderBy.stream().map(item -> new SortItem(Symbol.from(aliases.get(item.getField())).toSymbolReference(), item.getOrdering(), item.getNullOrdering())).collect(Collectors.toList())));
}
return new FunctionCall(Optional.empty(), name, Optional.empty(), filter.map(symbol -> symbol.toSymbol(aliases).toSymbolReference()), orderByClause, distinct, Optional.empty(), Optional.empty(), symbolReferences);
}
Also used :
OrderBy(io.trino.sql.tree.OrderBy)
Symbol(io.trino.sql.planner.Symbol)
SortItem(io.trino.sql.tree.SortItem)
Collectors(java.util.stream.Collectors)
String.format(java.lang.String.format)
QualifiedName(io.trino.sql.tree.QualifiedName)
Objects(java.util.Objects)
PlanMatchPattern.toSymbolReferences(io.trino.sql.planner.assertions.PlanMatchPattern.toSymbolReferences)
List(java.util.List)
ImmutableList(com.google.common.collect.ImmutableList)
Verify.verify(com.google.common.base.Verify.verify)
Objects.requireNonNull(java.util.Objects.requireNonNull)
WindowFrame(io.trino.sql.tree.WindowFrame)
Optional(java.util.Optional)
Expression(io.trino.sql.tree.Expression)
FunctionCall(io.trino.sql.tree.FunctionCall)
OrderBy(io.trino.sql.tree.OrderBy)
Joiner(com.google.common.base.Joiner)
WindowSpecification(io.trino.sql.tree.WindowSpecification)
SortItem(io.trino.sql.tree.SortItem)
Expression(io.trino.sql.tree.Expression)
FunctionCall(io.trino.sql.tree.FunctionCall)
Example 2 with OrderBy
use of io.trino.sql.tree.OrderBy in project trino by trinodb.
the class AstBuilder method visitQueryNoWith.
@Override
public Node visitQueryNoWith(SqlBaseParser.QueryNoWithContext context) {
QueryBody term = (QueryBody) visit(context.queryTerm());
Optional<OrderBy> orderBy = Optional.empty();
if (context.ORDER() != null) {
orderBy = Optional.of(new OrderBy(getLocation(context.ORDER()), visit(context.sortItem(), SortItem.class)));
}
Optional<Offset> offset = Optional.empty();
if (context.OFFSET() != null) {
Expression rowCount;
if (context.offset.INTEGER_VALUE() != null) {
rowCount = new LongLiteral(getLocation(context.offset.INTEGER_VALUE()), context.offset.getText());
} else {
rowCount = new Parameter(getLocation(context.offset.QUESTION_MARK()), parameterPosition);
parameterPosition++;
}
offset = Optional.of(new Offset(Optional.of(getLocation(context.OFFSET())), rowCount));
}
Optional<Node> limit = Optional.empty();
if (context.FETCH() != null) {
Optional<Expression> rowCount = Optional.empty();
if (context.fetchFirst != null) {
if (context.fetchFirst.INTEGER_VALUE() != null) {
rowCount = Optional.of(new LongLiteral(getLocation(context.fetchFirst.INTEGER_VALUE()), context.fetchFirst.getText()));
} else {
rowCount = Optional.of(new Parameter(getLocation(context.fetchFirst.QUESTION_MARK()), parameterPosition));
parameterPosition++;
}
}
limit = Optional.of(new FetchFirst(Optional.of(getLocation(context.FETCH())), rowCount, context.TIES() != null));
} else if (context.LIMIT() != null) {
if (context.limit == null) {
throw new IllegalStateException("Missing LIMIT value");
}
Expression rowCount;
if (context.limit.ALL() != null) {
rowCount = new AllRows(getLocation(context.limit.ALL()));
} else if (context.limit.rowCount().INTEGER_VALUE() != null) {
rowCount = new LongLiteral(getLocation(context.limit.rowCount().INTEGER_VALUE()), context.limit.getText());
} else {
rowCount = new Parameter(getLocation(context.limit.rowCount().QUESTION_MARK()), parameterPosition);
parameterPosition++;
}
limit = Optional.of(new Limit(Optional.of(getLocation(context.LIMIT())), rowCount));
}
if (term instanceof QuerySpecification) {
// When we have a simple query specification
// followed by order by, offset, limit or fetch,
// fold the order by, limit, offset or fetch clauses
// into the query specification (analyzer/planner
// expects this structure to resolve references with respect
// to columns defined in the query specification)
QuerySpecification query = (QuerySpecification) term;
return new Query(getLocation(context), Optional.empty(), new QuerySpecification(getLocation(context), query.getSelect(), query.getFrom(), query.getWhere(), query.getGroupBy(), query.getHaving(), query.getWindows(), orderBy, offset, limit), Optional.empty(), Optional.empty(), Optional.empty());
}
return new Query(getLocation(context), Optional.empty(), term, orderBy, offset, limit);
}
Also used :
OrderBy(io.trino.sql.tree.OrderBy)
AllRows(io.trino.sql.tree.AllRows)
Query(io.trino.sql.tree.Query)
WithQuery(io.trino.sql.tree.WithQuery)
LongLiteral(io.trino.sql.tree.LongLiteral)
Node(io.trino.sql.tree.Node)
TerminalNode(org.antlr.v4.runtime.tree.TerminalNode)
Offset(io.trino.sql.tree.Offset)
QuerySpecification(io.trino.sql.tree.QuerySpecification)
SortItem(io.trino.sql.tree.SortItem)
DereferenceExpression(io.trino.sql.tree.DereferenceExpression)
LogicalExpression(io.trino.sql.tree.LogicalExpression)
SearchedCaseExpression(io.trino.sql.tree.SearchedCaseExpression)
BindExpression(io.trino.sql.tree.BindExpression)
CoalesceExpression(io.trino.sql.tree.CoalesceExpression)
QuantifiedComparisonExpression(io.trino.sql.tree.QuantifiedComparisonExpression)
SimpleCaseExpression(io.trino.sql.tree.SimpleCaseExpression)
SubqueryExpression(io.trino.sql.tree.SubqueryExpression)
LambdaExpression(io.trino.sql.tree.LambdaExpression)
SubscriptExpression(io.trino.sql.tree.SubscriptExpression)
NullIfExpression(io.trino.sql.tree.NullIfExpression)
ArithmeticUnaryExpression(io.trino.sql.tree.ArithmeticUnaryExpression)
InListExpression(io.trino.sql.tree.InListExpression)
NotExpression(io.trino.sql.tree.NotExpression)
ArithmeticBinaryExpression(io.trino.sql.tree.ArithmeticBinaryExpression)
TryExpression(io.trino.sql.tree.TryExpression)
ComparisonExpression(io.trino.sql.tree.ComparisonExpression)
IfExpression(io.trino.sql.tree.IfExpression)
Expression(io.trino.sql.tree.Expression)
NumericParameter(io.trino.sql.tree.NumericParameter)
DataTypeParameter(io.trino.sql.tree.DataTypeParameter)
TypeParameter(io.trino.sql.tree.TypeParameter)
Parameter(io.trino.sql.tree.Parameter)
Limit(io.trino.sql.tree.Limit)
QueryBody(io.trino.sql.tree.QueryBody)
FetchFirst(io.trino.sql.tree.FetchFirst)
Example 3 with OrderBy
use of io.trino.sql.tree.OrderBy in project trino by trinodb.
the class TestSqlParser method testListagg.
@Test
public void testListagg() {
assertExpression("LISTAGG(x) WITHIN GROUP (ORDER BY x)", new FunctionCall(Optional.empty(), QualifiedName.of("LISTAGG"), Optional.empty(), Optional.empty(), Optional.of(new OrderBy(ImmutableList.of(new SortItem(new Identifier("x", false), ASCENDING, UNDEFINED)))), false, Optional.empty(), Optional.empty(), ImmutableList.of(identifier("x"), new StringLiteral(""), new BooleanLiteral("true"), new StringLiteral("..."), new BooleanLiteral("false"))));
assertExpression("LISTAGG( DISTINCT x) WITHIN GROUP (ORDER BY x)", new FunctionCall(Optional.empty(), QualifiedName.of("LISTAGG"), Optional.empty(), Optional.empty(), Optional.of(new OrderBy(ImmutableList.of(new SortItem(new Identifier("x", false), ASCENDING, UNDEFINED)))), true, Optional.empty(), Optional.empty(), ImmutableList.of(identifier("x"), new StringLiteral(""), new BooleanLiteral("true"), new StringLiteral("..."), new BooleanLiteral("false"))));
assertExpression("LISTAGG(x, ',') WITHIN GROUP (ORDER BY y)", new FunctionCall(Optional.empty(), QualifiedName.of("LISTAGG"), Optional.empty(), Optional.empty(), Optional.of(new OrderBy(ImmutableList.of(new SortItem(new Identifier("y", false), ASCENDING, UNDEFINED)))), false, Optional.empty(), Optional.empty(), ImmutableList.of(identifier("x"), new StringLiteral(","), new BooleanLiteral("true"), new StringLiteral("..."), new BooleanLiteral("false"))));
assertExpression("LISTAGG(x, ',' ON OVERFLOW ERROR) WITHIN GROUP (ORDER BY x)", new FunctionCall(Optional.empty(), QualifiedName.of("LISTAGG"), Optional.empty(), Optional.empty(), Optional.of(new OrderBy(ImmutableList.of(new SortItem(new Identifier("x", false), ASCENDING, UNDEFINED)))), false, Optional.empty(), Optional.empty(), ImmutableList.of(identifier("x"), new StringLiteral(","), new BooleanLiteral("true"), new StringLiteral("..."), new BooleanLiteral("false"))));
assertExpression("LISTAGG(x, ',' ON OVERFLOW TRUNCATE WITH COUNT) WITHIN GROUP (ORDER BY x)", new FunctionCall(Optional.empty(), QualifiedName.of("LISTAGG"), Optional.empty(), Optional.empty(), Optional.of(new OrderBy(ImmutableList.of(new SortItem(new Identifier("x", false), ASCENDING, UNDEFINED)))), false, Optional.empty(), Optional.empty(), ImmutableList.of(identifier("x"), new StringLiteral(","), new BooleanLiteral("false"), new StringLiteral("..."), new BooleanLiteral("true"))));
assertExpression("LISTAGG(x, ',' ON OVERFLOW TRUNCATE 'HIDDEN' WITHOUT COUNT) WITHIN GROUP (ORDER BY x)", new FunctionCall(Optional.empty(), QualifiedName.of("LISTAGG"), Optional.empty(), Optional.empty(), Optional.of(new OrderBy(ImmutableList.of(new SortItem(new Identifier("x", false), ASCENDING, UNDEFINED)))), false, Optional.empty(), Optional.empty(), ImmutableList.of(identifier("x"), new StringLiteral(","), new BooleanLiteral("false"), new StringLiteral("HIDDEN"), new BooleanLiteral("false"))));
}
Also used :
OrderBy(io.trino.sql.tree.OrderBy)
SortItem(io.trino.sql.tree.SortItem)
QueryUtil.quotedIdentifier(io.trino.sql.QueryUtil.quotedIdentifier)
Identifier(io.trino.sql.tree.Identifier)
StringLiteral(io.trino.sql.tree.StringLiteral)
BooleanLiteral(io.trino.sql.tree.BooleanLiteral)
FunctionCall(io.trino.sql.tree.FunctionCall)
Test(org.junit.jupiter.api.Test)
Example 4 with OrderBy
use of io.trino.sql.tree.OrderBy in project trino by trinodb.
the class TestSqlParser method testShowStatsForQuery.
@Test
public void testShowStatsForQuery() {
String[] tableNames = { "t", "s.t", "c.s.t" };
for (String fullName : tableNames) {
QualifiedName qualifiedName = makeQualifiedName(fullName);
// Simple SELECT
assertStatement(format("SHOW STATS FOR (SELECT * FROM %s)", qualifiedName), createShowStats(qualifiedName, ImmutableList.of(new AllColumns()), Optional.empty()));
// SELECT with predicate
assertStatement(format("SHOW STATS FOR (SELECT * FROM %s WHERE field > 0)", qualifiedName), createShowStats(qualifiedName, ImmutableList.of(new AllColumns()), Optional.of(new ComparisonExpression(ComparisonExpression.Operator.GREATER_THAN, new Identifier("field"), new LongLiteral("0")))));
// SELECT with more complex predicate
assertStatement(format("SHOW STATS FOR (SELECT * FROM %s WHERE field > 0 or field < 0)", qualifiedName), createShowStats(qualifiedName, ImmutableList.of(new AllColumns()), Optional.of(LogicalExpression.or(new ComparisonExpression(ComparisonExpression.Operator.GREATER_THAN, new Identifier("field"), new LongLiteral("0")), new ComparisonExpression(ComparisonExpression.Operator.LESS_THAN, new Identifier("field"), new LongLiteral("0"))))));
}
// SELECT with LIMIT
assertThat(statement("SHOW STATS FOR (SELECT * FROM t LIMIT 10)")).isEqualTo(new ShowStats(Optional.of(location(1, 1)), new TableSubquery(new Query(location(1, 17), Optional.empty(), new QuerySpecification(location(1, 17), new Select(location(1, 17), false, ImmutableList.of(new AllColumns(location(1, 24), Optional.empty(), ImmutableList.of()))), Optional.of(new Table(location(1, 31), QualifiedName.of(ImmutableList.of(new Identifier(location(1, 31), "t", false))))), Optional.empty(), Optional.empty(), Optional.empty(), ImmutableList.of(), Optional.empty(), Optional.empty(), Optional.of(new Limit(location(1, 33), new LongLiteral(location(1, 39), "10")))), Optional.empty(), Optional.empty(), Optional.empty()))));
// SELECT with ORDER BY ... LIMIT
assertThat(statement("SHOW STATS FOR (SELECT * FROM t ORDER BY field LIMIT 10)")).isEqualTo(new ShowStats(Optional.of(location(1, 1)), new TableSubquery(new Query(location(1, 17), Optional.empty(), new QuerySpecification(location(1, 17), new Select(location(1, 17), false, ImmutableList.of(new AllColumns(location(1, 24), Optional.empty(), ImmutableList.of()))), Optional.of(new Table(location(1, 31), QualifiedName.of(ImmutableList.of(new Identifier(location(1, 31), "t", false))))), Optional.empty(), Optional.empty(), Optional.empty(), ImmutableList.of(), Optional.of(new OrderBy(location(1, 33), ImmutableList.of(new SortItem(location(1, 42), new Identifier(location(1, 42), "field", false), ASCENDING, UNDEFINED)))), Optional.empty(), Optional.of(new Limit(location(1, 48), new LongLiteral(location(1, 54), "10")))), Optional.empty(), Optional.empty(), Optional.empty()))));
// SELECT with WITH
assertThat(statement("SHOW STATS FOR (\n" + " WITH t AS (SELECT 1 )\n" + " SELECT * FROM t)")).isEqualTo(new ShowStats(Optional.of(location(1, 1)), new TableSubquery(new Query(location(2, 4), Optional.of(new With(location(2, 4), false, ImmutableList.of(new WithQuery(location(2, 9), new Identifier(location(2, 9), "t", false), new Query(location(2, 15), Optional.empty(), new QuerySpecification(location(2, 15), new Select(location(2, 15), false, ImmutableList.of(new SingleColumn(location(2, 22), new LongLiteral(location(2, 22), "1"), Optional.empty()))), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), ImmutableList.of(), Optional.empty(), Optional.empty(), Optional.empty()), Optional.empty(), Optional.empty(), Optional.empty()), Optional.empty())))), new QuerySpecification(location(3, 4), new Select(location(3, 4), false, ImmutableList.of(new AllColumns(location(3, 11), Optional.empty(), ImmutableList.of()))), Optional.of(new Table(location(3, 18), QualifiedName.of(ImmutableList.of(new Identifier(location(3, 18), "t", false))))), Optional.empty(), Optional.empty(), Optional.empty(), ImmutableList.of(), Optional.empty(), Optional.empty(), Optional.empty()), Optional.empty(), Optional.empty(), Optional.empty()))));
}
Also used :
OrderBy(io.trino.sql.tree.OrderBy)
CreateTable(io.trino.sql.tree.CreateTable)
DropTable(io.trino.sql.tree.DropTable)
Table(io.trino.sql.tree.Table)
TruncateTable(io.trino.sql.tree.TruncateTable)
RenameTable(io.trino.sql.tree.RenameTable)
QueryUtil.simpleQuery(io.trino.sql.QueryUtil.simpleQuery)
Query(io.trino.sql.tree.Query)
WithQuery(io.trino.sql.tree.WithQuery)
LongLiteral(io.trino.sql.tree.LongLiteral)
QualifiedName(io.trino.sql.tree.QualifiedName)
AllColumns(io.trino.sql.tree.AllColumns)
SingleColumn(io.trino.sql.tree.SingleColumn)
TableSubquery(io.trino.sql.tree.TableSubquery)
With(io.trino.sql.tree.With)
QuantifiedComparisonExpression(io.trino.sql.tree.QuantifiedComparisonExpression)
ComparisonExpression(io.trino.sql.tree.ComparisonExpression)
QuerySpecification(io.trino.sql.tree.QuerySpecification)
SortItem(io.trino.sql.tree.SortItem)
QueryUtil.quotedIdentifier(io.trino.sql.QueryUtil.quotedIdentifier)
Identifier(io.trino.sql.tree.Identifier)
ShowStats(io.trino.sql.tree.ShowStats)
WithQuery(io.trino.sql.tree.WithQuery)
CreateTableAsSelect(io.trino.sql.tree.CreateTableAsSelect)
Select(io.trino.sql.tree.Select)
Limit(io.trino.sql.tree.Limit)
Test(org.junit.jupiter.api.Test)
Example 5 with OrderBy
use of io.trino.sql.tree.OrderBy in project trino by trinodb.
the class TestSqlParser method testWindowSpecification.
@Test
public void testWindowSpecification() {
assertExpression("rank() OVER someWindow", new FunctionCall(Optional.empty(), QualifiedName.of("rank"), Optional.of(new WindowReference(new Identifier("someWindow"))), Optional.empty(), Optional.empty(), false, Optional.empty(), Optional.empty(), ImmutableList.of()));
assertExpression("rank() OVER (someWindow PARTITION BY x ORDER BY y ROWS CURRENT ROW)", new FunctionCall(Optional.empty(), QualifiedName.of("rank"), Optional.of(new WindowSpecification(Optional.of(new Identifier("someWindow")), ImmutableList.of(new Identifier("x")), Optional.of(new OrderBy(ImmutableList.of(new SortItem(new Identifier("y"), ASCENDING, UNDEFINED)))), Optional.of(new WindowFrame(ROWS, new FrameBound(CURRENT_ROW), Optional.empty(), ImmutableList.of(), Optional.empty(), Optional.empty(), Optional.empty(), ImmutableList.of(), ImmutableList.of())))), Optional.empty(), Optional.empty(), false, Optional.empty(), Optional.empty(), ImmutableList.of()));
assertExpression("rank() OVER (PARTITION BY x ORDER BY y ROWS CURRENT ROW)", new FunctionCall(Optional.empty(), QualifiedName.of("rank"), Optional.of(new WindowSpecification(Optional.empty(), ImmutableList.of(new Identifier("x")), Optional.of(new OrderBy(ImmutableList.of(new SortItem(new Identifier("y"), ASCENDING, UNDEFINED)))), Optional.of(new WindowFrame(ROWS, new FrameBound(CURRENT_ROW), Optional.empty(), ImmutableList.of(), Optional.empty(), Optional.empty(), Optional.empty(), ImmutableList.of(), ImmutableList.of())))), Optional.empty(), Optional.empty(), false, Optional.empty(), Optional.empty(), ImmutableList.of()));
}
Also used :
OrderBy(io.trino.sql.tree.OrderBy)
SortItem(io.trino.sql.tree.SortItem)
QueryUtil.quotedIdentifier(io.trino.sql.QueryUtil.quotedIdentifier)
Identifier(io.trino.sql.tree.Identifier)
WindowFrame(io.trino.sql.tree.WindowFrame)
FrameBound(io.trino.sql.tree.FrameBound)
WindowSpecification(io.trino.sql.tree.WindowSpecification)
FunctionCall(io.trino.sql.tree.FunctionCall)
WindowReference(io.trino.sql.tree.WindowReference)
Test(org.junit.jupiter.api.Test)
Aggregations
OrderBy (io.trino.sql.tree.OrderBy)11
SortItem (io.trino.sql.tree.SortItem)11
FunctionCall (io.trino.sql.tree.FunctionCall)8
ComparisonExpression (io.trino.sql.tree.ComparisonExpression)7
Expression (io.trino.sql.tree.Expression)7
DereferenceExpression (io.trino.sql.tree.DereferenceExpression)6
Identifier (io.trino.sql.tree.Identifier)6
ArithmeticBinaryExpression (io.trino.sql.tree.ArithmeticBinaryExpression)5
IfExpression (io.trino.sql.tree.IfExpression)5
LambdaExpression (io.trino.sql.tree.LambdaExpression)5
QualifiedName (io.trino.sql.tree.QualifiedName)5
QuantifiedComparisonExpression (io.trino.sql.tree.QuantifiedComparisonExpression)5
QueryUtil.quotedIdentifier (io.trino.sql.QueryUtil.quotedIdentifier)4
ArithmeticUnaryExpression (io.trino.sql.tree.ArithmeticUnaryExpression)4
BindExpression (io.trino.sql.tree.BindExpression)4
CoalesceExpression (io.trino.sql.tree.CoalesceExpression)4
InListExpression (io.trino.sql.tree.InListExpression)4
LogicalExpression (io.trino.sql.tree.LogicalExpression)4
LongLiteral (io.trino.sql.tree.LongLiteral)4
Query (io.trino.sql.tree.Query)4
