mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-04 05:22:17 +00:00
43 lines
1.6 KiB
C++
43 lines
1.6 KiB
C++
#include <Functions/FunctionFactory.h>
|
|
#include <Functions/FunctionsStringArray.h>
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
|
|
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
|
|
}
|
|
|
|
DataTypePtr FunctionArrayStringConcat::getReturnTypeImpl(const DataTypes & arguments) const
|
|
{
|
|
if (arguments.size() != 1 && arguments.size() != 2)
|
|
throw Exception(
|
|
"Number of arguments for function " + getName() + " doesn't match: passed " + toString(arguments.size())
|
|
+ ", should be 1 or 2.",
|
|
ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH);
|
|
|
|
const DataTypeArray * array_type = checkAndGetDataType<DataTypeArray>(arguments[0].get());
|
|
if (!array_type)
|
|
throw Exception("First argument for function " + getName() + " must be an array.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
|
|
|
|
if (arguments.size() == 2 && !isString(arguments[1]))
|
|
throw Exception("Second argument for function " + getName() + " must be constant string.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
|
|
|
|
return std::make_shared<DataTypeString>();
|
|
}
|
|
|
|
void registerFunctionsStringArray(FunctionFactory & factory)
|
|
{
|
|
factory.registerFunction<FunctionExtractAll>();
|
|
factory.registerFunction<FunctionAlphaTokens>();
|
|
factory.registerFunction<FunctionSplitByNonAlpha>();
|
|
factory.registerFunction<FunctionSplitByWhitespace>();
|
|
factory.registerFunction<FunctionSplitByChar>();
|
|
factory.registerFunction<FunctionSplitByString>();
|
|
factory.registerFunction<FunctionSplitByRegexp>();
|
|
factory.registerFunction<FunctionArrayStringConcat>();
|
|
}
|
|
|
|
}
|