Method toASTImpl

src/Analyzer/QueryNode.cpp:450–591

}

ASTPtr QueryNode::toASTImpl(const ConvertToASTOptions & options) const
{
    auto select_query = make_intrusive<ASTSelectQuery>();
    /// Preserve the parser invariant `recursive_with -> with() != nullptr`.
    select_query->recursive_with = is_recursive_with && hasWith();
    select_query->distinct = is_distinct;
    select_query->limit_with_ties = is_limit_with_ties;
    select_query->group_by_with_totals = is_group_by_with_totals;
    select_query->group_by_with_rollup = is_group_by_with_rollup;
    select_query->group_by_with_cube = is_group_by_with_cube;
    select_query->group_by_with_grouping_sets = is_group_by_with_grouping_sets;
    select_query->group_by_all = is_group_by_all;
    select_query->order_by_all = is_order_by_all;
    select_query->limit_by_all = is_limit_by_all;

    if (hasWith())
    {
        const auto & with = getWith();
        auto expression_list_ast = make_intrusive<ASTExpressionList>();
        expression_list_ast->children.reserve(with.getNodes().size());

        for (const auto & with_node : with)
        {
            auto with_node_ast = with_node->toAST(options);
            expression_list_ast->children.push_back(with_node_ast);

            const auto * with_query_node = with_node->as<QueryNode>();
            const auto * with_union_node = with_node->as<UnionNode>();
            if (!with_query_node && !with_union_node)
                continue;

            bool is_with_node_cte = with_query_node ? with_query_node->isCTE() : with_union_node->isCTE();
            if (!is_with_node_cte)
                continue;

            const auto & with_node_cte_name = with_query_node ? with_query_node->cte_name : with_union_node->getCTEName();

            auto * with_node_ast_subquery = with_node_ast->as<ASTSubquery>();
            if (with_node_ast_subquery)
                with_node_ast_subquery->cte_name = "";

            auto with_element_ast = make_intrusive<ASTWithElement>();
            with_element_ast->name = with_node_cte_name;
            with_element_ast->subquery = std::move(with_node_ast);
            with_element_ast->children.push_back(with_element_ast->subquery);
            with_element_ast->is_materialized = with_query_node ? with_query_node->isMaterialized() : with_union_node->isMaterialized();

            expression_list_ast->children.back() = std::move(with_element_ast);
        }

        select_query->setExpression(ASTSelectQuery::Expression::WITH, std::move(expression_list_ast));
    }

    auto projection_ast = getProjection().toAST(options);
    auto & projection_expression_list_ast = projection_ast->as<ASTExpressionList &>();
    size_t projection_expression_list_ast_children_size = projection_expression_list_ast.children.size();
    if (projection_expression_list_ast_children_size != getProjection().getNodes().size())
        throw Exception(ErrorCodes::LOGICAL_ERROR, "Query node invalid projection conversion to AST");