Search in sources :

Example 1 with Intersect

use of io.trino.sql.tree.Intersect in project trino by trinodb.

the class AstBuilder method visitSetOperation.

@Override
public Node visitSetOperation(SqlBaseParser.SetOperationContext context) {
    QueryBody left = (QueryBody) visit(context.left);
    QueryBody right = (QueryBody) visit(context.right);
    boolean distinct = context.setQuantifier() == null || context.setQuantifier().DISTINCT() != null;
    switch(context.operator.getType()) {
        case SqlBaseLexer.UNION:
            return new Union(getLocation(context.UNION()), ImmutableList.of(left, right), distinct);
        case SqlBaseLexer.INTERSECT:
            return new Intersect(getLocation(context.INTERSECT()), ImmutableList.of(left, right), distinct);
        case SqlBaseLexer.EXCEPT:
            return new Except(getLocation(context.EXCEPT()), left, right, distinct);
    }
    throw new IllegalArgumentException("Unsupported set operation: " + context.operator.getText());
}
Also used : Intersect(io.trino.sql.tree.Intersect) QueryBody(io.trino.sql.tree.QueryBody) Union(io.trino.sql.tree.Union) Except(io.trino.sql.tree.Except)

Aggregations

Except (io.trino.sql.tree.Except)1 Intersect (io.trino.sql.tree.Intersect)1 QueryBody (io.trino.sql.tree.QueryBody)1 Union (io.trino.sql.tree.Union)1