Method getTable

src/Interpreters/InterpreterInsertQuery.cpp:165–232

StoragePtr InterpreterInsertQuery::getTable(ASTInsertQuery & query)
{
    if (query.table_function)
    {
        const auto & factory = TableFunctionFactory::instance();
        TableFunctionPtr table_function_ptr = factory.get(query.table_function, getContext());
        return table_function_ptr->execute(query.table_function, getContext(), table_function_ptr->getName());
    }

    if (getContext()->getServerType() == ServerType::cnch_worker && !query.select)
    {
        query.table_id = getContext()->resolveStorageID(query.table_id);
        auto storage = DatabaseCatalog::instance().tryGetTable(query.table_id, getContext());
        if (auto * cnch_table = dynamic_cast<StorageCnchMergeTree *>(storage.get()))
        {
            auto create_query = cnch_table->getCreateQueryForCloudTable(
                cnch_table->getCreateTableSql(),
                query.table_id.table_name,
                nullptr,
                false,
                std::nullopt,
                Strings{},
                query.table_id.database_name);
            LOG_TRACE(&Poco::Logger::get(__PRETTY_FUNCTION__), "Worker side create query: {}", create_query);

            NameSet view_create_sqls = genViewDependencyCreateQueries(storage, getContext());
            if (!view_create_sqls.empty())
            {
                ContextMutablePtr mutable_context = const_pointer_cast<Context>(getContext());
                if (!mutable_context->tryGetCnchWorkerResource())
                    mutable_context->initCnchWorkerResource();
                view_create_sqls.insert(create_query);
                for (const auto & create_sql : view_create_sqls)
                    mutable_context->getCnchWorkerResource()->executeCreateQuery(mutable_context, create_sql);
                if (auto worker_txn = dynamic_pointer_cast<CnchWorkerTransaction>(mutable_context->getCurrentTransaction()); worker_txn)
                {
                    if (view_create_sqls.size() > 1)
                    {
                        worker_txn->enableExplicitCommit();
                        worker_txn->setExplicitCommitStorageID(storage->getStorageID());
                    }

                    mutable_context->setCurrentTransaction(worker_txn);
                }
                return mutable_context->getCnchWorkerResource()->getTable(query.table_id);
            }
            else
            {
                query.table_id = getContext()->resolveStorageID(query.table_id);
                return DatabaseCatalog::instance().getTable(query.table_id, getContext());
            }
        }
        else
            return storage;
    }
    else
    {
        query.table_id = getContext()->resolveStorageID(query.table_id);