Example 1 with Limit
use of io.trino.sql.tree.Limit 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 Limit
use of io.trino.sql.tree.Limit 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 3 with Limit
use of io.trino.sql.tree.Limit in project trino by trinodb.
the class TestSqlParser method testSelectWithLimit.
@Test
public void testSelectWithLimit() {
assertStatement("SELECT * FROM table1 LIMIT 2", simpleQuery(selectList(new AllColumns()), new Table(QualifiedName.of("table1")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Limit(new LongLiteral("2")))));
assertStatement("SELECT * FROM table1 LIMIT ALL", simpleQuery(selectList(new AllColumns()), new Table(QualifiedName.of("table1")), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Limit(new AllRows()))));
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')) LIMIT ALL", simpleQuery(selectList(new AllColumns()), subquery(valuesQuery), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.of(new Limit(new AllRows()))));
}
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)
AllRows(io.trino.sql.tree.AllRows)
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)
Limit(io.trino.sql.tree.Limit)
Test(org.junit.jupiter.api.Test)
Example 4 with Limit
use of io.trino.sql.tree.Limit in project trino by trinodb.
the class QueryRewriter method getColumns.
private List<Column> getColumns(Connection connection, CreateTableAsSelect createTableAsSelect) throws SQLException {
io.trino.sql.tree.Query createSelectClause = createTableAsSelect.getQuery();
// Rewrite the query to select zero rows, so that we can get the column names and types
QueryBody innerQuery = createSelectClause.getQueryBody();
io.trino.sql.tree.Query zeroRowsQuery;
if (innerQuery instanceof QuerySpecification) {
QuerySpecification querySpecification = (QuerySpecification) innerQuery;
innerQuery = new QuerySpecification(querySpecification.getSelect(), querySpecification.getFrom(), querySpecification.getWhere(), querySpecification.getGroupBy(), querySpecification.getHaving(), querySpecification.getWindows(), querySpecification.getOrderBy(), querySpecification.getOffset(), Optional.of(new Limit(new LongLiteral("0"))));
zeroRowsQuery = new io.trino.sql.tree.Query(createSelectClause.getWith(), innerQuery, Optional.empty(), Optional.empty(), Optional.empty());
} else {
zeroRowsQuery = new io.trino.sql.tree.Query(createSelectClause.getWith(), innerQuery, Optional.empty(), Optional.empty(), Optional.of(new Limit(new LongLiteral("0"))));
}
ImmutableList.Builder<Column> columns = ImmutableList.builder();
try (java.sql.Statement jdbcStatement = connection.createStatement()) {
ExecutorService executor = newSingleThreadExecutor();
TimeLimiter limiter = SimpleTimeLimiter.create(executor);
java.sql.Statement limitedStatement = limiter.newProxy(jdbcStatement, java.sql.Statement.class, timeout.toMillis(), TimeUnit.MILLISECONDS);
try (ResultSet resultSet = limitedStatement.executeQuery(formatSql(zeroRowsQuery))) {
ResultSetMetaData metaData = resultSet.getMetaData();
for (int i = 1; i <= metaData.getColumnCount(); i++) {
String name = metaData.getColumnName(i);
int type = metaData.getColumnType(i);
columns.add(new Column(name, APPROXIMATE_TYPES.contains(type)));
}
} catch (UncheckedTimeoutException e) {
throw new SQLException("SQL statement execution timed out", e);
} finally {
executor.shutdownNow();
}
}
return columns.build();
}
Also used :
SimpleTimeLimiter(com.google.common.util.concurrent.SimpleTimeLimiter)
TimeLimiter(com.google.common.util.concurrent.TimeLimiter)
LongLiteral(io.trino.sql.tree.LongLiteral)
SQLException(java.sql.SQLException)
ImmutableList(com.google.common.collect.ImmutableList)
UncheckedTimeoutException(com.google.common.util.concurrent.UncheckedTimeoutException)
ResultSetMetaData(java.sql.ResultSetMetaData)
QuerySpecification(io.trino.sql.tree.QuerySpecification)
SingleColumn(io.trino.sql.tree.SingleColumn)
ExecutorService(java.util.concurrent.ExecutorService)
ResultSet(java.sql.ResultSet)
Limit(io.trino.sql.tree.Limit)
QueryBody(io.trino.sql.tree.QueryBody)
Example 5 with Limit
use of io.trino.sql.tree.Limit 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)
Aggregations
Limit (io.trino.sql.tree.Limit)5
LongLiteral (io.trino.sql.tree.LongLiteral)4
AllColumns (io.trino.sql.tree.AllColumns)3
Query (io.trino.sql.tree.Query)3
QuerySpecification (io.trino.sql.tree.QuerySpecification)3
WithQuery (io.trino.sql.tree.WithQuery)3
Test (org.junit.jupiter.api.Test)3
QueryUtil.simpleQuery (io.trino.sql.QueryUtil.simpleQuery)2
AllRows (io.trino.sql.tree.AllRows)2
ComparisonExpression (io.trino.sql.tree.ComparisonExpression)2
CreateTable (io.trino.sql.tree.CreateTable)2
DropTable (io.trino.sql.tree.DropTable)2
FetchFirst (io.trino.sql.tree.FetchFirst)2
OrderBy (io.trino.sql.tree.OrderBy)2
QuantifiedComparisonExpression (io.trino.sql.tree.QuantifiedComparisonExpression)2
QueryBody (io.trino.sql.tree.QueryBody)2
RenameTable (io.trino.sql.tree.RenameTable)2
SingleColumn (io.trino.sql.tree.SingleColumn)2
SortItem (io.trino.sql.tree.SortItem)2
Table (io.trino.sql.tree.Table)2
