Method checkTableEngine

src/Databases/DatabaseReplicated.cpp:1167–1234

}

void DatabaseReplicated::checkTableEngine(const ASTCreateQuery & query, ASTStorage & storage, ContextPtr query_context) const
{
    bool replicated_table = storage.engine &&
        (startsWith(storage.engine->name, "Replicated") || startsWith(storage.engine->name, "Shared"));
    if (!replicated_table || !storage.engine->arguments)
        return;

    ASTs & args_ref = storage.engine->arguments->children;
    ASTs args = args_ref;
    if (args.size() < 2)
        return;

    /// It can be a constant expression. Try to evaluate it, ignore exception if we cannot.
    bool has_expression_argument = args_ref[0]->as<ASTFunction>() || args_ref[1]->as<ASTFunction>();
    if (has_expression_argument)
    {
        try
        {
            args[0] = evaluateConstantExpressionAsLiteral(args_ref[0]->clone(), query_context);
            args[1] = evaluateConstantExpressionAsLiteral(args_ref[1]->clone(), query_context);
        }
        catch (const Exception &) // NOLINT(bugprone-empty-catch)
        {
        }
    }

    ASTLiteral * arg1 = args[0]->as<ASTLiteral>();
    ASTLiteral * arg2 = args[1]->as<ASTLiteral>();
    if (!arg1 || !arg2 || arg1->value.getType() != Field::Types::String || arg2->value.getType() != Field::Types::String)
        return;

    String maybe_path = arg1->value.safeGet<String>();
    String maybe_replica = arg2->value.safeGet<String>();

    /// Looks like it's ReplicatedMergeTree with explicit zookeeper_path and replica_name arguments.
    /// Let's ensure that some macros are used.
    /// NOTE: we cannot check here that substituted values will be actually different on shards and replicas.

    Macros::MacroExpansionInfo info;
    info.table_id = {getDatabaseName(), query.getTable(), query.uuid};
    info.shard = getShardName();
    info.replica = getReplicaName();
    query_context->getMacros()->expand(maybe_path, info);
    bool maybe_shard_macros = info.expanded_other;
    info.expanded_other = false;
    query_context->getMacros()->expand(maybe_replica, info);
    bool maybe_replica_macros = info.expanded_other;
    bool enable_functional_tests_helper = getContext()->getConfigRef().has("_functional_tests_helper_database_replicated_replace_args_macros");

    if (maybe_shard_macros && maybe_replica_macros)
        return;

    if (enable_functional_tests_helper && !has_expression_argument)
    {
        if (maybe_path.empty() || maybe_path.back() != '/')
            maybe_path += '/';
        args_ref[0]->as<ASTLiteral>()->value = maybe_path + "auto_{shard}";
        args_ref[1]->as<ASTLiteral>()->value = maybe_replica + "auto_{replica}";