mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-06 00:20:51 +00:00
57 lines
1.9 KiB
C++
57 lines
1.9 KiB
C++
#include <TableFunctions/TableFunctionCatBoostPool.h>
|
|
#include <Storages/StorageCatBoostPool.h>
|
|
#include <TableFunctions/TableFunctionFactory.h>
|
|
#include <Parsers/ASTFunction.h>
|
|
#include <Common/typeid_cast.h>
|
|
#include <Parsers/ASTLiteral.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
|
|
extern const int BAD_ARGUMENTS;
|
|
}
|
|
|
|
|
|
StoragePtr TableFunctionCatBoostPool::execute(const ASTPtr & ast_function, const Context & context) const
|
|
{
|
|
ASTs & args_func = typeid_cast<ASTFunction &>(*ast_function).children;
|
|
|
|
std::string err = "Table function '" + getName() + "' requires 2 parameters: "
|
|
+ "column descriptions file, dataset description file";
|
|
|
|
if (args_func.size() != 1)
|
|
throw Exception(err, ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH);
|
|
|
|
ASTs & args = typeid_cast<ASTExpressionList &>(*args_func.at(0)).children;
|
|
|
|
if (args.size() != 2)
|
|
throw Exception(err, ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH);
|
|
|
|
auto getStringLiteral = [](const IAST & node, const char * description)
|
|
{
|
|
auto lit = typeid_cast<const ASTLiteral *>(&node);
|
|
if (!lit)
|
|
throw Exception(description + String(" must be string literal (in single quotes)."), ErrorCodes::BAD_ARGUMENTS);
|
|
|
|
if (lit->value.getType() != Field::Types::String)
|
|
throw Exception(description + String(" must be string literal (in single quotes)."), ErrorCodes::BAD_ARGUMENTS);
|
|
|
|
return safeGet<const String &>(lit->value);
|
|
};
|
|
String column_descriptions_file = getStringLiteral(*args[0], "Column descriptions file");
|
|
String dataset_description_file = getStringLiteral(*args[1], "Dataset description file");
|
|
|
|
return StorageCatBoostPool::create(context, column_descriptions_file, dataset_description_file);
|
|
}
|
|
|
|
void registerTableFunctionCatBoostPool(TableFunctionFactory & factory)
|
|
{
|
|
factory.registerFunction<TableFunctionCatBoostPool>();
|
|
}
|
|
|
|
}
|