thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-12-15 02:41:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

reorder branches

Browse Source
This commit is contained in:
taiyang-li 2023-01-16 21:02:33 +08:00
parent 5b3f9eb62e
commit 87f869a129
1 changed files with 16 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/Functions/multiIf.cpp
View File

@ -18,7 +18,7 @@ namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
extern const int LOGICAL_ERROR;
extern const int NOT_IMPLEMENTED;
}
namespace
@ -232,8 +232,14 @@ public:
size_t rows = input_rows_count;
const auto & settings = context->getSettingsRef();
const WhichDataType which(result_type);
if (settings.allow_execute_multiif_columnar && (which.isInt() || which.isUInt() || which.isFloat()))
bool execute_multiif_columnar = settings.allow_execute_multiif_columnar && (which.isInt() || which.isUInt() || which.isFloat());
if (!execute_multiif_columnar)
{
MutableColumnPtr res = return_type->createColumn();
executeInstructions(instructions, rows, res);
return std::move(res);
}
#define EXECUTE_INSTRUCTIONS_COLUMNAR(TYPE) \
if (which.is##TYPE()) \
{ \
@ -258,20 +264,12 @@ public:
M(Float32) \
M(Float64)
ENUMERATE_NUMERIC_TYPES(EXECUTE_INSTRUCTIONS_COLUMNAR)
}
else
{
MutableColumnPtr res = return_type->createColumn();
executeInstructions(instructions, rows, res);
return std::move(res);
}
return {};
}
ENUMERATE_NUMERIC_TYPES(EXECUTE_INSTRUCTIONS_COLUMNAR)
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "Columnar multiIf not implemented for type {}", result_type->getName());
}
private:
static NO_INLINE void executeInstructions(std::vector<Instruction> & instructions, size_t rows, const MutableColumnPtr & res)
static void executeInstructions(std::vector<Instruction> & instructions, size_t rows, const MutableColumnPtr & res)
{
for (size_t i = 0; i < rows; ++i)
{
@ -308,7 +306,7 @@ private:
}
/// We should read source from which instruction on each row?
static NO_INLINE void calculateInserts(std::vector<Instruction> & instructions, size_t rows, PaddedPODArray<Int64> & inserts)
static void calculateInserts(std::vector<Instruction> & instructions, size_t rows, PaddedPODArray<Int64> & inserts)
{
for (Int64 i = instructions.size() - 1; i >= 0; --i)
{
@ -352,7 +350,7 @@ private:
}
template <typename T>
static NO_INLINE void executeInstructionsColumnar(std::vector<Instruction> & instructions, size_t rows, const MutableColumnPtr & res)
static void executeInstructionsColumnar(std::vector<Instruction> & instructions, size_t rows, const MutableColumnPtr & res)
{
PaddedPODArray<Int64> inserts(rows, -1);
calculateInserts(instructions, rows, inserts);
@ -361,8 +359,8 @@ private:
for (size_t row_i = 0; row_i < rows; ++row_i)
{
auto & instruction = instructions[inserts[row_i]];
auto str_ref = instruction.source->getDataAt(row_i);
memcpy(&res_data[row_i], str_ref.data, str_ref.size);
auto ref = instruction.source->getDataAt(row_i);
memcpy(&res_data[row_i], ref.data, ref.size);
}
}