mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-14 11:33:46 +00:00
107 lines
2.8 KiB
C++
107 lines
2.8 KiB
C++
#pragma once
|
|
|
|
#include <DB/Functions/IFunction.h>
|
|
#include <DB/Functions/FunctionsLogical.h>
|
|
#include <DB/DataTypes/DataTypesNumberFixed.h>
|
|
#include <DB/Columns/ColumnNullable.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class FunctionIsNull : public IFunction
|
|
{
|
|
public:
|
|
static constexpr auto name = "isNull";
|
|
static FunctionPtr create(const Context & context) { return std::make_shared<FunctionIsNull>(); }
|
|
|
|
std::string getName() const override
|
|
{
|
|
return name;
|
|
}
|
|
|
|
bool hasSpecialSupportForNulls() const override
|
|
{
|
|
return true;
|
|
}
|
|
|
|
DataTypePtr getReturnTypeImpl(const DataTypes & arguments) const override
|
|
{
|
|
if (arguments.size() != 1)
|
|
throw Exception("Number of arguments for function " + getName() + " doesn't match: passed "
|
|
+ toString(arguments.size()) + ", should be 1.",
|
|
ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH);
|
|
|
|
return std::make_shared<DataTypeUInt8>();
|
|
}
|
|
|
|
void executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) override
|
|
{
|
|
ColumnWithTypeAndName & elem = block.getByPosition(arguments[0]);
|
|
if (elem.column->isNullable())
|
|
{
|
|
// Merely return the embedded null map.
|
|
ColumnNullable & nullable_col = static_cast<ColumnNullable &>(*elem.column);
|
|
block.getByPosition(result).column = nullable_col.getNullValuesByteMap();
|
|
}
|
|
else
|
|
{
|
|
/// Since no element is nullable, create a zero-filled null map.
|
|
ColumnPtr holder = std::make_shared<ColumnUInt8>();
|
|
ColumnUInt8 & content = static_cast<ColumnUInt8 &>(*(holder.get()));
|
|
NullValuesByteMap & null_map = content.getData();
|
|
|
|
null_map.resize_fill(elem.column.get()->size());
|
|
block.getByPosition(result).column = holder;
|
|
}
|
|
}
|
|
};
|
|
|
|
class FunctionIsNotNull : public IFunction
|
|
{
|
|
public:
|
|
static constexpr auto name = "isNotNull";
|
|
static FunctionPtr create(const Context & context) { return std::make_shared<FunctionIsNotNull>(); }
|
|
|
|
std::string getName() const override
|
|
{
|
|
return name;
|
|
}
|
|
|
|
bool hasSpecialSupportForNulls() const override
|
|
{
|
|
return true;
|
|
}
|
|
|
|
DataTypePtr getReturnTypeImpl(const DataTypes & arguments) const override
|
|
{
|
|
if (arguments.size() != 1)
|
|
throw Exception("Number of arguments for function " + getName() + " doesn't match: passed "
|
|
+ toString(arguments.size()) + ", should be 1.",
|
|
ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH);
|
|
|
|
return std::make_shared<DataTypeUInt8>();
|
|
}
|
|
|
|
void executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) override
|
|
{
|
|
FunctionIsNull is_null_func;
|
|
|
|
ColumnWithTypeAndName temp_res;
|
|
temp_res.type = std::make_shared<DataTypeUInt8>();
|
|
|
|
block.insert(temp_res);
|
|
|
|
size_t temp_res_num = block.columns() - 1;
|
|
|
|
is_null_func.executeImpl(block, arguments, temp_res_num);
|
|
|
|
FunctionNot not_func;
|
|
|
|
ColumnNumbers new_args;
|
|
new_args.push_back(temp_res_num);
|
|
not_func.executeImpl(block, new_args, result);
|
|
}
|
|
};
|
|
|
|
}
|