ClickHouse/src/Functions/FunctionStringToString.h

78 lines
2.3 KiB
C++
Raw Normal View History

2020-10-10 18:37:02 +00:00
#pragma once
2018-09-09 23:36:06 +00:00
#include <DataTypes/DataTypeString.h>
2018-09-09 23:47:56 +00:00
#include <Columns/ColumnString.h>
#include <Columns/ColumnFixedString.h>
#include <Functions/FunctionHelpers.h>
#include <Functions/IFunctionImpl.h>
2018-09-09 23:36:06 +00:00
namespace DB
{
namespace ErrorCodes
{
extern const int ILLEGAL_COLUMN;
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
}
template <typename Impl, typename Name, bool is_injective = false>
class FunctionStringToString : public IFunction
{
public:
static constexpr auto name = Name::name;
static FunctionPtr create(const Context &)
{
return std::make_shared<FunctionStringToString>();
}
String getName() const override
{
return name;
}
size_t getNumberOfArguments() const override
{
return 1;
}
bool isInjective(const ColumnsWithTypeAndName &) const override
2018-09-09 23:36:06 +00:00
{
return is_injective;
}
DataTypePtr getReturnTypeImpl(const DataTypes & arguments) const override
{
if (!isStringOrFixedString(arguments[0]))
throw Exception(
"Illegal type " + arguments[0]->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
return arguments[0];
}
bool useDefaultImplementationForConstants() const override { return true; }
void executeImpl(ColumnsWithTypeAndName & block, const ColumnNumbers & arguments, size_t result, size_t /*input_rows_count*/) const override
2018-09-09 23:36:06 +00:00
{
2020-10-10 18:24:57 +00:00
const ColumnPtr column = block[arguments[0]].column;
2018-09-09 23:36:06 +00:00
if (const ColumnString * col = checkAndGetColumn<ColumnString>(column.get()))
{
auto col_res = ColumnString::create();
Impl::vector(col->getChars(), col->getOffsets(), col_res->getChars(), col_res->getOffsets());
2020-10-10 18:24:57 +00:00
block[result].column = std::move(col_res);
2018-09-09 23:36:06 +00:00
}
else if (const ColumnFixedString * col_fixed = checkAndGetColumn<ColumnFixedString>(column.get()))
2018-09-09 23:36:06 +00:00
{
auto col_res = ColumnFixedString::create(col_fixed->getN());
Impl::vectorFixed(col_fixed->getChars(), col_fixed->getN(), col_res->getChars());
2020-10-10 18:24:57 +00:00
block[result].column = std::move(col_res);
2018-09-09 23:36:06 +00:00
}
else
throw Exception(
2020-10-10 18:24:57 +00:00
"Illegal column " + block[arguments[0]].column->getName() + " of argument of function " + getName(),
2018-09-09 23:36:06 +00:00
ErrorCodes::ILLEGAL_COLUMN);
}
};
}