Method parseArguments

src/TableFunctions/TableFunctionMongoDB.cpp:74–119

}

void TableFunctionMongoDB::parseArguments(const ASTPtr & ast_function, ContextPtr context)
{
    const auto & func_args = ast_function->as<ASTFunction &>();
    if (!func_args.arguments)
        throw Exception(ErrorCodes::BAD_ARGUMENTS, "Table function 'mongodb' must have arguments.");

    ASTs & args = func_args.arguments->children;

    if (auto named_collection = tryGetNamedCollectionWithOverrides(args, context))
    {
        if (!named_collection->has("structure"))
            throw Exception(ErrorCodes::BAD_ARGUMENTS, "Required key (structure) is not specified.");

        structure = named_collection->get<String>("structure");
        named_collection->remove("structure");
        configuration = std::make_shared<MongoDBConfiguration>(StorageMongoDB::getConfigurationFromCollection(named_collection, context));
        return;
    }

    if ((args.size() < 3 || args.size() > 4) && (args.size() < 6 || args.size() > 8))
        throw Exception(ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH,
                        "Incorrect argument count for table function '{}'. Usage: "
                        "mongodb('host:port', database, collection, user, password, structure[, options[, oid_columns]]) or mongodb(uri, collection, structure[, oid_columns]).",
                        getName());

    ASTs main_arguments;
    main_arguments.reserve(args.size() - 1);
    const size_t structure_position = args.size() <= 4 ? 2 : 5;
    for (size_t i = 0; i < args.size(); ++i)
    {
        if (const auto * ast_func = typeid_cast<const ASTFunction *>(args[i].get()))
        {
            const auto & [arg_name, arg_value] = getKeyValueMongoDBArgument(ast_func);
            if (arg_name == "structure")
                structure = checkAndGetLiteralArgument<String>(arg_value, arg_name);
            else if (arg_name == "options" || arg_name == "oid_columns")
                main_arguments.push_back(arg_value);
        }
        else if (i == structure_position)
            structure = checkAndGetLiteralArgument<String>(args[i], "structure");
        else
            main_arguments.push_back(args[i]);
    }

    configuration = std::make_shared<MongoDBConfiguration>(StorageMongoDB::getConfiguration(main_arguments, context));
}

}