Function buildQueryTreeDistributed

src/Storages/StorageDistributed.cpp:865–954

}

QueryTreeNodePtr buildQueryTreeDistributed(SelectQueryInfo & query_info,
    const StorageSnapshotPtr & distributed_storage_snapshot,
    const StorageID & remote_storage_id,
    const ASTPtr & remote_table_function)
{
    auto & planner_context = query_info.planner_context;
    const auto & query_context = planner_context->getQueryContext();

    std::optional<TableExpressionModifiers> table_expression_modifiers;

    if (auto * query_info_table_node = query_info.table_expression->as<TableNode>())
        table_expression_modifiers = query_info_table_node->getTableExpressionModifiers();
    else if (auto * query_info_table_function_node = query_info.table_expression->as<TableFunctionNode>())
        table_expression_modifiers = query_info_table_function_node->getTableExpressionModifiers();

    QueryTreeNodePtr replacement_table_expression;

    if (remote_table_function)
    {
        auto remote_table_function_query_tree = buildQueryTree(remote_table_function, query_context);
        auto & remote_table_function_node = remote_table_function_query_tree->as<FunctionNode &>();

        auto table_function_node = std::make_shared<TableFunctionNode>(remote_table_function_node.getFunctionName());
        table_function_node->getArgumentsNode() = remote_table_function_node.getArgumentsNode();

        if (table_expression_modifiers)
            table_function_node->setTableExpressionModifiers(*table_expression_modifiers);

        /// Subquery in table function `view` may reference tables that don't exist on the initiator.
        if (table_function_node->getTableFunctionName() == "view")
        {
            auto get_column_options = GetColumnsOptions(GetColumnsOptions::All).withVirtuals(VirtualsKind::All, VirtualsMaterializationPlace::All);
            auto column_names_and_types = distributed_storage_snapshot->getColumns(get_column_options);

            StorageID fake_storage_id = StorageID::createEmpty();
            if (auto * table_node = query_info.table_expression->as<TableNode>())
                fake_storage_id = table_node->getStorage()->getStorageID();
            else if (auto * original_table_function_node = query_info.table_expression->as<TableFunctionNode>())
                fake_storage_id = original_table_function_node->getStorage()->getStorageID();

            auto storage = std::make_shared<StorageDummy>(fake_storage_id, ColumnsDescription{column_names_and_types});

            table_function_node->resolve({}, std::move(storage), query_context, /*unresolved_arguments_indexes_=*/{ 0 });
        }
        else
        {
            QueryAnalysisPass query_analysis_pass;
            QueryTreeNodePtr node = table_function_node;
            query_analysis_pass.run(node, query_context);
        }

        replacement_table_expression = std::move(table_function_node);
    }
    else
    {
        auto get_column_options = GetColumnsOptions(GetColumnsOptions::All).withVirtuals(VirtualsKind::All, VirtualsMaterializationPlace::All);

        auto column_names_and_types = distributed_storage_snapshot->getColumns(get_column_options);