Example 1 with Offset
use of io.trino.sql.tree.Offset 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 2 with Offset
use of io.trino.sql.tree.Offset in project trino by trinodb.
the class TestSqlParser method testPrepareWithParameters.
@Test
public void testPrepareWithParameters() {
assertStatement("PREPARE myquery FROM SELECT ?, ? FROM foo", new Prepare(identifier("myquery"), simpleQuery(selectList(new Parameter(0), new Parameter(1)), table(QualifiedName.of("foo")))));
assertStatement("PREPARE myquery FROM SELECT * FROM foo LIMIT ?", new Prepare(identifier("myquery"), simpleQuery(selectList(new AllColumns()), table(QualifiedName.of("foo")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Limit(new Parameter(0))))));
assertStatement("PREPARE myquery FROM SELECT ?, ? FROM foo LIMIT ?", new Prepare(identifier("myquery"), simpleQuery(selectList(new Parameter(0), new Parameter(1)), table(QualifiedName.of("foo")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Limit(new Parameter(2))))));
assertStatement("PREPARE myquery FROM SELECT ? FROM foo FETCH FIRST ? ROWS ONLY", new Prepare(identifier("myquery"), simpleQuery(selectList(new Parameter(0)), table(QualifiedName.of("foo")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new FetchFirst(new Parameter(1))))));
assertStatement("PREPARE myquery FROM SELECT ?, ? FROM foo FETCH NEXT ? ROWS WITH TIES", new Prepare(identifier("myquery"), simpleQuery(selectList(new Parameter(0), new Parameter(1)), table(QualifiedName.of("foo")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new FetchFirst(new Parameter(2), true)))));
assertStatement("PREPARE myquery FROM SELECT ?, ? FROM foo OFFSET ? ROWS", new Prepare(identifier("myquery"), simpleQuery(selectList(new Parameter(0), new Parameter(1)), table(QualifiedName.of("foo")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Offset(new Parameter(2))), Optional.empty())));
assertStatement("PREPARE myquery FROM SELECT ? FROM foo OFFSET ? ROWS LIMIT ?", new Prepare(identifier("myquery"), simpleQuery(selectList(new Parameter(0)), table(QualifiedName.of("foo")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Offset(new Parameter(1))), Optional.of(new Limit(new Parameter(2))))));
assertStatement("PREPARE myquery FROM SELECT ? FROM foo OFFSET ? ROWS FETCH FIRST ? ROWS WITH TIES", new Prepare(identifier("myquery"), simpleQuery(selectList(new Parameter(0)), table(QualifiedName.of("foo")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Offset(new Parameter(1))), Optional.of(new FetchFirst(new Parameter(2), true)))));
}
Also used :
Prepare(io.trino.sql.tree.Prepare)
Parameter(io.trino.sql.tree.Parameter)
AllColumns(io.trino.sql.tree.AllColumns)
Limit(io.trino.sql.tree.Limit)
FetchFirst(io.trino.sql.tree.FetchFirst)
Offset(io.trino.sql.tree.Offset)
Test(org.junit.jupiter.api.Test)
Example 3 with Offset
use of io.trino.sql.tree.Offset in project trino by trinodb.
the class TestSqlParser method testSelectWithOffset.
@Test
public void testSelectWithOffset() {
assertStatement("SELECT * FROM table1 OFFSET 2 ROWS", simpleQuery(selectList(new AllColumns()), new Table(QualifiedName.of("table1")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Offset(new LongLiteral("2"))), Optional.empty()));
assertStatement("SELECT * FROM table1 OFFSET 2", simpleQuery(selectList(new AllColumns()), new Table(QualifiedName.of("table1")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Offset(new LongLiteral("2"))), Optional.empty()));
Query valuesQuery = query(values(row(new LongLiteral("1"), new StringLiteral("1")), row(new LongLiteral("2"), new StringLiteral("2"))));
assertStatement("SELECT * FROM (VALUES (1, '1'), (2, '2')) OFFSET 2 ROWS", simpleQuery(selectList(new AllColumns()), subquery(valuesQuery), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Offset(new LongLiteral("2"))), Optional.empty()));
assertStatement("SELECT * FROM (VALUES (1, '1'), (2, '2')) OFFSET 2", simpleQuery(selectList(new AllColumns()), subquery(valuesQuery), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Offset(new LongLiteral("2"))), Optional.empty()));
}
Also used :
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)
StringLiteral(io.trino.sql.tree.StringLiteral)
LongLiteral(io.trino.sql.tree.LongLiteral)
AllColumns(io.trino.sql.tree.AllColumns)
Offset(io.trino.sql.tree.Offset)
Test(org.junit.jupiter.api.Test)
Aggregations
Offset (io.trino.sql.tree.Offset)3
AllColumns (io.trino.sql.tree.AllColumns)2
FetchFirst (io.trino.sql.tree.FetchFirst)2
Limit (io.trino.sql.tree.Limit)2
LongLiteral (io.trino.sql.tree.LongLiteral)2
Parameter (io.trino.sql.tree.Parameter)2
Query (io.trino.sql.tree.Query)2
WithQuery (io.trino.sql.tree.WithQuery)2
Test (org.junit.jupiter.api.Test)2
QueryUtil.simpleQuery (io.trino.sql.QueryUtil.simpleQuery)1
AllRows (io.trino.sql.tree.AllRows)1
ArithmeticBinaryExpression (io.trino.sql.tree.ArithmeticBinaryExpression)1
ArithmeticUnaryExpression (io.trino.sql.tree.ArithmeticUnaryExpression)1
BindExpression (io.trino.sql.tree.BindExpression)1
CoalesceExpression (io.trino.sql.tree.CoalesceExpression)1
ComparisonExpression (io.trino.sql.tree.ComparisonExpression)1
CreateTable (io.trino.sql.tree.CreateTable)1
DataTypeParameter (io.trino.sql.tree.DataTypeParameter)1
DereferenceExpression (io.trino.sql.tree.DereferenceExpression)1
DropTable (io.trino.sql.tree.DropTable)1
