ClickHouse/src/Functions/FunctionStringOrArrayToT.h

103 lines
3.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 <DataTypes/DataTypesNumber.h>
#include <Functions/IFunctionImpl.h>
2018-09-09 23:36:06 +00:00
#include <Functions/FunctionHelpers.h>
#include <Columns/ColumnVector.h>
#include <Columns/ColumnString.h>
#include <Columns/ColumnFixedString.h>
#include <Columns/ColumnArray.h>
namespace DB
{
namespace ErrorCodes
{
extern const int ILLEGAL_COLUMN;
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
}
template <typename Impl, typename Name, typename ResultType>
class FunctionStringOrArrayToT : public IFunction
{
public:
static constexpr auto name = Name::name;
static FunctionPtr create(const Context &)
{
return std::make_shared<FunctionStringOrArrayToT>();
}
String getName() const override
{
return name;
}
size_t getNumberOfArguments() const override
{
return 1;
}
DataTypePtr getReturnTypeImpl(const DataTypes & arguments) const override
{
if (!isStringOrFixedString(arguments[0])
&& !isArray(arguments[0]))
throw Exception("Illegal type " + arguments[0]->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
return std::make_shared<DataTypeNumber<ResultType>>();
}
bool useDefaultImplementationForConstants() const override { return true; }
2020-10-17 16:48:53 +00:00
ColumnPtr executeImpl(ColumnsWithTypeAndName & arguments, const DataTypePtr & result_type, size_t /*input_rows_count*/) const override
2018-09-09 23:36:06 +00:00
{
2020-10-17 16:48:53 +00:00
const ColumnPtr column = arguments[0].column;
2018-09-09 23:36:06 +00:00
if (const ColumnString * col = checkAndGetColumn<ColumnString>(column.get()))
{
auto col_res = ColumnVector<ResultType>::create();
typename ColumnVector<ResultType>::Container & vec_res = col_res->getData();
vec_res.resize(col->size());
Impl::vector(col->getChars(), col->getOffsets(), vec_res);
2020-10-17 16:48:53 +00:00
return 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
{
if (Impl::is_fixed_to_constant)
{
ResultType res = 0;
Impl::vectorFixedToConstant(col_fixed->getChars(), col_fixed->getN(), res);
2018-09-09 23:36:06 +00:00
2020-10-17 16:48:53 +00:00
return result_type->createColumnConst(col_fixed->size(), toField(res));
2018-09-09 23:36:06 +00:00
}
else
{
auto col_res = ColumnVector<ResultType>::create();
typename ColumnVector<ResultType>::Container & vec_res = col_res->getData();
vec_res.resize(col_fixed->size());
Impl::vectorFixedToVector(col_fixed->getChars(), col_fixed->getN(), vec_res);
2018-09-09 23:36:06 +00:00
2020-10-17 16:48:53 +00:00
return col_res;
2018-09-09 23:36:06 +00:00
}
}
else if (const ColumnArray * col_arr = checkAndGetColumn<ColumnArray>(column.get()))
2018-09-09 23:36:06 +00:00
{
auto col_res = ColumnVector<ResultType>::create();
typename ColumnVector<ResultType>::Container & vec_res = col_res->getData();
vec_res.resize(col_arr->size());
Impl::array(col_arr->getOffsets(), vec_res);
2018-09-09 23:36:06 +00:00
2020-10-17 16:48:53 +00:00
return col_res;
2018-09-09 23:36:06 +00:00
}
else
2020-10-17 16:48:53 +00:00
throw Exception("Illegal column " + arguments[0].column->getName() + " of argument of function " + getName(),
2018-09-09 23:36:06 +00:00
ErrorCodes::ILLEGAL_COLUMN);
}
};
}