Examples with QueryType io.trino.spi.resourcegroups.QueryType used on opensource projects

Example 1 with QueryType

use of io.trino.spi.resourcegroups.QueryType in project trino by trinodb.

the class QueryPreparer method prepareQuery.

public PreparedQuery prepareQuery(Session session, Statement wrappedStatement) throws ParsingException, TrinoException {
    Statement statement = wrappedStatement;
    Optional<String> prepareSql = Optional.empty();
    if (statement instanceof Execute) {
        prepareSql = Optional.of(session.getPreparedStatementFromExecute((Execute) statement));
        statement = sqlParser.createStatement(prepareSql.get(), createParsingOptions(session));
    }
    if (statement instanceof ExplainAnalyze) {
        Statement innerStatement = ((ExplainAnalyze) statement).getStatement();
        Optional<QueryType> innerQueryType = getQueryType(innerStatement);
        if (innerQueryType.isEmpty() || innerQueryType.get() == QueryType.DATA_DEFINITION) {
            throw new TrinoException(NOT_SUPPORTED, "EXPLAIN ANALYZE doesn't support statement type: " + innerStatement.getClass().getSimpleName());
        }
    }
    List<Expression> parameters = ImmutableList.of();
    if (wrappedStatement instanceof Execute) {
        parameters = ((Execute) wrappedStatement).getParameters();
    }
    validateParameters(statement, parameters);
    return new PreparedQuery(statement, parameters, prepareSql);
}