#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace DB { /// Implementation of FunctionArray. String FunctionArray::getName() const { return name; } FunctionPtr FunctionArray::create(const Context & context) { return std::make_shared(context); } FunctionArray::FunctionArray(const Context & context) : context(context) { } DataTypePtr FunctionArray::getReturnTypeImpl(const DataTypes & arguments) const { return std::make_shared(getLeastCommonType(arguments)); } void FunctionArray::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { size_t num_elements = arguments.size(); const DataTypePtr & return_type = block.getByPosition(result).type; const DataTypePtr & elem_type = static_cast(*return_type).getNestedType(); size_t block_size = block.rows(); /** If part of columns have not same type as common type of all elements of array, * then convert them to common type. * If part of columns are constants, * then convert them to full columns. */ Columns columns_holder(num_elements); const IColumn * columns[num_elements]; for (size_t i = 0; i < num_elements; ++i) { const auto & arg = block.getByPosition(arguments[i]); ColumnPtr preprocessed_column = arg.column; if (!arg.type->equals(*elem_type)) preprocessed_column = castColumn(arg, elem_type, context); if (ColumnPtr materialized_column = preprocessed_column->convertToFullColumnIfConst()) preprocessed_column = materialized_column; columns_holder[i] = std::move(preprocessed_column); columns[i] = columns_holder[i].get(); } /** Create and fill the result array. */ auto out = ColumnArray::create(elem_type->createColumn()); IColumn & out_data = out->getData(); IColumn::Offsets & out_offsets = out->getOffsets(); out_data.reserve(block_size * num_elements); out_offsets.resize(block_size); IColumn::Offset current_offset = 0; for (size_t i = 0; i < block_size; ++i) { for (size_t j = 0; j < num_elements; ++j) out_data.insertFrom(*columns[j], i); current_offset += num_elements; out_offsets[i] = current_offset; } block.getByPosition(result).column = std::move(out); } /// Implementation of FunctionArrayElement. namespace ArrayImpl { class NullMapBuilder { public: operator bool() const { return src_null_map; } bool operator!() const { return !src_null_map; } void initSource(const UInt8 * src_null_map_) { src_null_map = src_null_map_; } void initSink(size_t size) { auto sink = ColumnUInt8::create(size); sink_null_map = sink->getData().data(); sink_null_map_holder = std::move(sink); } void update(size_t from) { sink_null_map[index] = bool(src_null_map && src_null_map[from]); ++index; } void update() { sink_null_map[index] = bool(src_null_map); ++index; } ColumnPtr getNullMapData() && { return std::move(sink_null_map_holder); } private: const UInt8 * src_null_map = nullptr; UInt8 * sink_null_map = nullptr; MutableColumnPtr sink_null_map_holder; size_t index = 0; }; } namespace { template struct ArrayElementNumImpl { /** Implementation for constant index. * If negative = false - index is from beginning of array, started from 0. * If negative = true - index is from end of array, started from 0. */ template static void vectorConst( const PaddedPODArray & data, const ColumnArray::Offsets & offsets, const ColumnArray::Offset index, PaddedPODArray & result, ArrayImpl::NullMapBuilder & builder) { size_t size = offsets.size(); result.resize(size); ColumnArray::Offset current_offset = 0; for (size_t i = 0; i < size; ++i) { size_t array_size = offsets[i] - current_offset; if (index < array_size) { size_t j = !negative ? (current_offset + index) : (offsets[i] - index - 1); result[i] = data[j]; if (builder) builder.update(j); } else { result[i] = T(); if (builder) builder.update(); } current_offset = offsets[i]; } } /** Implementation for non-constant index. */ template static void vector( const PaddedPODArray & data, const ColumnArray::Offsets & offsets, const PaddedPODArray & indices, PaddedPODArray & result, ArrayImpl::NullMapBuilder & builder) { size_t size = offsets.size(); result.resize(size); ColumnArray::Offset current_offset = 0; for (size_t i = 0; i < size; ++i) { size_t array_size = offsets[i] - current_offset; TIndex index = indices[i]; if (index > 0 && static_cast(index) <= array_size) { size_t j = current_offset + index - 1; result[i] = data[j]; if (builder) builder.update(j); } else if (index < 0 && static_cast(-index) <= array_size) { size_t j = offsets[i] + index; result[i] = data[j]; if (builder) builder.update(j); } else { result[i] = T(); if (builder) builder.update(); } current_offset = offsets[i]; } } }; struct ArrayElementStringImpl { template static void vectorConst( const ColumnString::Chars_t & data, const ColumnArray::Offsets & offsets, const ColumnString::Offsets & string_offsets, const ColumnArray::Offset index, ColumnString::Chars_t & result_data, ColumnArray::Offsets & result_offsets, ArrayImpl::NullMapBuilder & builder) { size_t size = offsets.size(); result_offsets.resize(size); result_data.reserve(data.size()); ColumnArray::Offset current_offset = 0; ColumnArray::Offset current_result_offset = 0; for (size_t i = 0; i < size; ++i) { size_t array_size = offsets[i] - current_offset; if (index < array_size) { size_t adjusted_index = !negative ? index : (array_size - index - 1); size_t j = current_offset + adjusted_index; if (builder) builder.update(j); ColumnArray::Offset string_pos = current_offset == 0 && adjusted_index == 0 ? 0 : string_offsets[current_offset + adjusted_index - 1]; ColumnArray::Offset string_size = string_offsets[current_offset + adjusted_index] - string_pos; result_data.resize(current_result_offset + string_size); memcpySmallAllowReadWriteOverflow15(&result_data[current_result_offset], &data[string_pos], string_size); current_result_offset += string_size; result_offsets[i] = current_result_offset; } else { /// Insert an empty row. result_data.resize(current_result_offset + 1); result_data[current_result_offset] = 0; current_result_offset += 1; result_offsets[i] = current_result_offset; if (builder) builder.update(); } current_offset = offsets[i]; } } /** Implementation for non-constant index. */ template static void vector( const ColumnString::Chars_t & data, const ColumnArray::Offsets & offsets, const ColumnString::Offsets & string_offsets, const PaddedPODArray & indices, ColumnString::Chars_t & result_data, ColumnArray::Offsets & result_offsets, ArrayImpl::NullMapBuilder & builder) { size_t size = offsets.size(); result_offsets.resize(size); result_data.reserve(data.size()); ColumnArray::Offset current_offset = 0; ColumnArray::Offset current_result_offset = 0; for (size_t i = 0; i < size; ++i) { size_t array_size = offsets[i] - current_offset; size_t adjusted_index; /// index in array from zero TIndex index = indices[i]; if (index > 0 && static_cast(index) <= array_size) adjusted_index = index - 1; else if (index < 0 && static_cast(-index) <= array_size) adjusted_index = array_size + index; else adjusted_index = array_size; /// means no element should be taken if (adjusted_index < array_size) { size_t j = current_offset + adjusted_index; if (builder) builder.update(j); ColumnArray::Offset string_pos = current_offset == 0 && adjusted_index == 0 ? 0 : string_offsets[current_offset + adjusted_index - 1]; ColumnArray::Offset string_size = string_offsets[current_offset + adjusted_index] - string_pos; result_data.resize(current_result_offset + string_size); memcpySmallAllowReadWriteOverflow15(&result_data[current_result_offset], &data[string_pos], string_size); current_result_offset += string_size; result_offsets[i] = current_result_offset; } else { /// Insert empty string result_data.resize(current_result_offset + 1); result_data[current_result_offset] = 0; current_result_offset += 1; result_offsets[i] = current_result_offset; if (builder) builder.update(); } current_offset = offsets[i]; } } }; /// Generic implementation for other nested types. struct ArrayElementGenericImpl { template static void vectorConst( const IColumn & data, const ColumnArray::Offsets & offsets, const ColumnArray::Offset index, IColumn & result, ArrayImpl::NullMapBuilder & builder) { size_t size = offsets.size(); result.reserve(size); ColumnArray::Offset current_offset = 0; for (size_t i = 0; i < size; ++i) { size_t array_size = offsets[i] - current_offset; if (index < array_size) { size_t j = !negative ? current_offset + index : offsets[i] - index - 1; result.insertFrom(data, j); if (builder) builder.update(j); } else { result.insertDefault(); if (builder) builder.update(); } current_offset = offsets[i]; } } /** Implementation for non-constant index. */ template static void vector( const IColumn & data, const ColumnArray::Offsets & offsets, const PaddedPODArray & indices, IColumn & result, ArrayImpl::NullMapBuilder & builder) { size_t size = offsets.size(); result.reserve(size); ColumnArray::Offset current_offset = 0; for (size_t i = 0; i < size; ++i) { size_t array_size = offsets[i] - current_offset; TIndex index = indices[i]; if (index > 0 && static_cast(index) <= array_size) { size_t j = current_offset + index - 1; result.insertFrom(data, j); if (builder) builder.update(j); } else if (index < 0 && static_cast(-index) <= array_size) { size_t j = offsets[i] + index; result.insertFrom(data, j); if (builder) builder.update(j); } else { result.insertDefault(); if (builder) builder.update(); } current_offset = offsets[i]; } } }; } FunctionPtr FunctionArrayElement::create(const Context &) { return std::make_shared(); } template bool FunctionArrayElement::executeNumberConst(Block & block, const ColumnNumbers & arguments, size_t result, const Field & index, ArrayImpl::NullMapBuilder & builder) { const ColumnArray * col_array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get()); if (!col_array) return false; const ColumnVector * col_nested = checkAndGetColumn>(&col_array->getData()); if (!col_nested) return false; auto col_res = ColumnVector::create(); block.getByPosition(result).column = col_res; if (index.getType() == Field::Types::UInt64) ArrayElementNumImpl::template vectorConst( col_nested->getData(), col_array->getOffsets(), safeGet(index) - 1, col_res->getData(), builder); else if (index.getType() == Field::Types::Int64) ArrayElementNumImpl::template vectorConst( col_nested->getData(), col_array->getOffsets(), -safeGet(index) - 1, col_res->getData(), builder); else throw Exception("Illegal type of array index", ErrorCodes::LOGICAL_ERROR); return true; } template bool FunctionArrayElement::executeNumber(Block & block, const ColumnNumbers & arguments, size_t result, const PaddedPODArray & indices, ArrayImpl::NullMapBuilder & builder) { const ColumnArray * col_array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get()); if (!col_array) return false; const ColumnVector * col_nested = checkAndGetColumn>(&col_array->getData()); if (!col_nested) return false; auto col_res = ColumnVector::create(); ArrayElementNumImpl::template vector( col_nested->getData(), col_array->getOffsets(), indices, col_res->getData(), builder); block.getByPosition(result).column = std::move(col_res); return true; } bool FunctionArrayElement::executeStringConst(Block & block, const ColumnNumbers & arguments, size_t result, const Field & index, ArrayImpl::NullMapBuilder & builder) { const ColumnArray * col_array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get()); if (!col_array) return false; const ColumnString * col_nested = checkAndGetColumn(&col_array->getData()); if (!col_nested) return false; auto col_res = ColumnString::create(); if (index.getType() == Field::Types::UInt64) ArrayElementStringImpl::vectorConst( col_nested->getChars(), col_array->getOffsets(), col_nested->getOffsets(), safeGet(index) - 1, col_res->getChars(), col_res->getOffsets(), builder); else if (index.getType() == Field::Types::Int64) ArrayElementStringImpl::vectorConst( col_nested->getChars(), col_array->getOffsets(), col_nested->getOffsets(), -safeGet(index) - 1, col_res->getChars(), col_res->getOffsets(), builder); else throw Exception("Illegal type of array index", ErrorCodes::LOGICAL_ERROR); block.getByPosition(result).column = std::move(col_res); return true; } template bool FunctionArrayElement::executeString(Block & block, const ColumnNumbers & arguments, size_t result, const PaddedPODArray & indices, ArrayImpl::NullMapBuilder & builder) { const ColumnArray * col_array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get()); if (!col_array) return false; const ColumnString * col_nested = checkAndGetColumn(&col_array->getData()); if (!col_nested) return false; auto col_res = ColumnString::create(); ArrayElementStringImpl::vector( col_nested->getChars(), col_array->getOffsets(), col_nested->getOffsets(), indices, col_res->getChars(), col_res->getOffsets(), builder); block.getByPosition(result).column = std::move(col_res); return true; } bool FunctionArrayElement::executeGenericConst(Block & block, const ColumnNumbers & arguments, size_t result, const Field & index, ArrayImpl::NullMapBuilder & builder) { const ColumnArray * col_array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get()); if (!col_array) return false; const auto & col_nested = col_array->getData(); auto col_res = col_nested.cloneEmpty(); if (index.getType() == Field::Types::UInt64) ArrayElementGenericImpl::vectorConst( col_nested, col_array->getOffsets(), safeGet(index) - 1, *col_res, builder); else if (index.getType() == Field::Types::Int64) ArrayElementGenericImpl::vectorConst( col_nested, col_array->getOffsets(), -safeGet(index) - 1, *col_res, builder); else throw Exception("Illegal type of array index", ErrorCodes::LOGICAL_ERROR); block.getByPosition(result).column = std::move(col_res); return true; } template bool FunctionArrayElement::executeGeneric(Block & block, const ColumnNumbers & arguments, size_t result, const PaddedPODArray & indices, ArrayImpl::NullMapBuilder & builder) { const ColumnArray * col_array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get()); if (!col_array) return false; const auto & col_nested = col_array->getData(); auto col_res = col_nested.cloneEmpty(); ArrayElementGenericImpl::vector( col_nested, col_array->getOffsets(), indices, *col_res, builder); block.getByPosition(result).column = std::move(col_res); return true; } template bool FunctionArrayElement::executeConst(Block & block, const ColumnNumbers & arguments, size_t result, const PaddedPODArray & indices, ArrayImpl::NullMapBuilder & builder) { const ColumnArray * col_array = checkAndGetColumnConstData(block.getByPosition(arguments[0]).column.get()); if (!col_array) return false; auto res = block.getByPosition(result).type->createColumn(); size_t rows = block.rows(); const IColumn & array_elements = col_array->getData(); size_t array_size = array_elements.size(); for (size_t i = 0; i < rows; ++i) { IndexType index = indices[i]; if (index > 0 && static_cast(index) <= array_size) { size_t j = index - 1; res->insertFrom(array_elements, j); if (builder) builder.update(j); } else if (index < 0 && static_cast(-index) <= array_size) { size_t j = array_size + index; res->insertFrom(array_elements, j); if (builder) builder.update(j); } else { res->insertDefault(); if (builder) builder.update(); } } block.getByPosition(result).column = std::move(res); return true; } template bool FunctionArrayElement::executeArgument(Block & block, const ColumnNumbers & arguments, size_t result, ArrayImpl::NullMapBuilder & builder) { auto index = checkAndGetColumn>(block.getByPosition(arguments[1]).column.get()); if (!index) return false; const auto & index_data = index->getData(); if (builder) builder.initSink(index_data.size()); if (!( executeNumber(block, arguments, result, index_data, builder) || executeNumber(block, arguments, result, index_data, builder) || executeNumber(block, arguments, result, index_data, builder) || executeNumber(block, arguments, result, index_data, builder) || executeNumber(block, arguments, result, index_data, builder) || executeNumber(block, arguments, result, index_data, builder) || executeNumber(block, arguments, result, index_data, builder) || executeNumber(block, arguments, result, index_data, builder) || executeNumber(block, arguments, result, index_data, builder) || executeNumber(block, arguments, result, index_data, builder) || executeConst (block, arguments, result, index_data, builder) || executeString(block, arguments, result, index_data, builder) || executeGeneric(block, arguments, result, index_data, builder))) throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); return true; } bool FunctionArrayElement::executeTuple(Block & block, const ColumnNumbers & arguments, size_t result) { const ColumnArray * col_array = typeid_cast(block.getByPosition(arguments[0]).column.get()); if (!col_array) return false; const ColumnTuple * col_nested = typeid_cast(&col_array->getData()); if (!col_nested) return false; const Columns & tuple_columns = col_nested->getColumns(); size_t tuple_size = tuple_columns.size(); const DataTypes & tuple_types = typeid_cast( *typeid_cast(*block.getByPosition(arguments[0]).type).getNestedType()).getElements(); /** We will calculate the function for the tuple of the internals of the array. * To do this, create a temporary block. * It will consist of the following columns * - the index of the array to be taken; * - an array of the first elements of the tuples; * - the result of taking the elements by the index for an array of the first elements of the tuples; * - array of the second elements of the tuples; * - result of taking elements by index for an array of second elements of tuples; * ... */ Block block_of_temporary_results; block_of_temporary_results.insert(block.getByPosition(arguments[1])); /// results of taking elements by index for arrays from each element of the tuples; Columns result_tuple_columns; for (size_t i = 0; i < tuple_size; ++i) { ColumnWithTypeAndName array_of_tuple_section; array_of_tuple_section.column = ColumnArray::create(tuple_columns[i], col_array->getOffsetsPtr()); array_of_tuple_section.type = std::make_shared(tuple_types[i]); block_of_temporary_results.insert(array_of_tuple_section); ColumnWithTypeAndName array_elements_of_tuple_section; array_elements_of_tuple_section.type = getReturnTypeImpl( {block_of_temporary_results.getByPosition(i * 2 + 1).type, block_of_temporary_results.getByPosition(0).type}); block_of_temporary_results.insert(array_elements_of_tuple_section); executeImpl(block_of_temporary_results, ColumnNumbers{i * 2 + 1, 0}, i * 2 + 2); result_tuple_columns.emplace_back(std::move(block_of_temporary_results.getByPosition(i * 2 + 2).column)); } block.getByPosition(result).column = ColumnTuple::create(result_tuple_columns); return true; } String FunctionArrayElement::getName() const { return name; } DataTypePtr FunctionArrayElement::getReturnTypeImpl(const DataTypes & arguments) const { const DataTypeArray * array_type = checkAndGetDataType(arguments[0].get()); if (!array_type) throw Exception("First argument for function " + getName() + " must be array.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); if (!arguments[1]->isInteger()) throw Exception("Second argument for function " + getName() + " must be integer.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return array_type->getNestedType(); } void FunctionArrayElement::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { /// Check nullability. bool is_nullable_array = false; const ColumnArray * col_array = nullptr; const ColumnArray * col_const_array = nullptr; col_array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get()); if (col_array) is_nullable_array = col_array->getData().isColumnNullable(); else { col_const_array = checkAndGetColumnConstData(block.getByPosition(arguments[0]).column.get()); if (col_const_array) is_nullable_array = col_const_array->getData().isColumnNullable(); else throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } if (!is_nullable_array) { ArrayImpl::NullMapBuilder builder; perform(block, arguments, result, builder); } else { /// Perform initializations. ArrayImpl::NullMapBuilder builder; Block source_block; const auto & input_type = typeid_cast(*typeid_cast(*block.getByPosition(arguments[0]).type).getNestedType()).getNestedType(); const auto & tmp_ret_type = typeid_cast(*block.getByPosition(result).type).getNestedType(); if (col_array) { const auto & nullable_col = typeid_cast(col_array->getData()); const auto & nested_col = nullable_col.getNestedColumnPtr(); /// Put nested_col inside a ColumnArray. source_block = { { ColumnArray::create(nested_col, col_array->getOffsetsPtr()), std::make_shared(input_type), "" }, block.getByPosition(arguments[1]), { nullptr, tmp_ret_type, "" } }; builder.initSource(nullable_col.getNullMapData().data()); } else { /// ColumnConst(ColumnArray(ColumnNullable(...))) const auto & nullable_col = static_cast(col_const_array->getData()); const auto & nested_col = nullable_col.getNestedColumnPtr(); source_block = { { ColumnConst::create(ColumnArray::create(nested_col, col_const_array->getOffsetsPtr()), block.rows()), std::make_shared(input_type), "" }, block.getByPosition(arguments[1]), { nullptr, tmp_ret_type, "" } }; builder.initSource(nullable_col.getNullMapData().data()); } perform(source_block, {0, 1}, 2, builder); /// Store the result. const ColumnWithTypeAndName & source_col = source_block.getByPosition(2); ColumnWithTypeAndName & dest_col = block.getByPosition(result); dest_col.column = ColumnNullable::create(source_col.column, std::move(builder).getNullMapData()); } } void FunctionArrayElement::perform(Block & block, const ColumnNumbers & arguments, size_t result, ArrayImpl::NullMapBuilder & builder) { if (executeTuple(block, arguments, result)) { } else if (!block.getByPosition(arguments[1]).column->isColumnConst()) { if (!( executeArgument(block, arguments, result, builder) || executeArgument(block, arguments, result, builder) || executeArgument(block, arguments, result, builder) || executeArgument(block, arguments, result, builder) || executeArgument(block, arguments, result, builder) || executeArgument(block, arguments, result, builder) || executeArgument(block, arguments, result, builder) || executeArgument(block, arguments, result, builder))) throw Exception("Second argument for function " + getName() + " must must have UInt or Int type.", ErrorCodes::ILLEGAL_COLUMN); } else { Field index = (*block.getByPosition(arguments[1]).column)[0]; if (builder) builder.initSink(block.rows()); if (index == UInt64(0)) throw Exception("Array indices is 1-based", ErrorCodes::ZERO_ARRAY_OR_TUPLE_INDEX); if (!( executeNumberConst(block, arguments, result, index, builder) || executeNumberConst(block, arguments, result, index, builder) || executeNumberConst(block, arguments, result, index, builder) || executeNumberConst(block, arguments, result, index, builder) || executeNumberConst(block, arguments, result, index, builder) || executeNumberConst(block, arguments, result, index, builder) || executeNumberConst(block, arguments, result, index, builder) || executeNumberConst(block, arguments, result, index, builder) || executeNumberConst(block, arguments, result, index, builder) || executeNumberConst(block, arguments, result, index, builder) || executeStringConst (block, arguments, result, index, builder) || executeGenericConst (block, arguments, result, index, builder))) throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } } /// Implementation of FunctionArrayEnumerate. FunctionPtr FunctionArrayEnumerate::create(const Context &) { return std::make_shared(); } String FunctionArrayEnumerate::getName() const { return name; } DataTypePtr FunctionArrayEnumerate::getReturnTypeImpl(const DataTypes & arguments) const { const DataTypeArray * array_type = checkAndGetDataType(arguments[0].get()); if (!array_type) throw Exception("First argument for function " + getName() + " must be an array but it has type " + arguments[0]->getName() + ".", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return std::make_shared(std::make_shared()); } void FunctionArrayEnumerate::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { if (const ColumnArray * array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get())) { const ColumnArray::Offsets & offsets = array->getOffsets(); auto res_nested = ColumnUInt32::create(); auto res_array = ColumnArray::create(res_nested, array->getOffsetsPtr()); ColumnUInt32::Container & res_values = res_nested->getData(); res_values.resize(array->getData().size()); size_t prev_off = 0; for (size_t i = 0; i < offsets.size(); ++i) { size_t off = offsets[i]; for (size_t j = prev_off; j < off; ++j) { res_values[j] = j - prev_off + 1; } prev_off = off; } block.getByPosition(result).column = std::move(res_array); } else { throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } } /// Implementation of FunctionArrayUniq. FunctionPtr FunctionArrayUniq::create(const Context &) { return std::make_shared(); } String FunctionArrayUniq::getName() const { return name; } DataTypePtr FunctionArrayUniq::getReturnTypeImpl(const DataTypes & arguments) const { if (arguments.size() == 0) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + toString(arguments.size()) + ", should be at least 1.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); for (size_t i = 0; i < arguments.size(); ++i) { const DataTypeArray * array_type = checkAndGetDataType(arguments[i].get()); if (!array_type) throw Exception("All arguments for function " + getName() + " must be arrays but argument " + toString(i + 1) + " has type " + arguments[i]->getName() + ".", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } return std::make_shared(); } void FunctionArrayUniq::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { Columns array_columns(arguments.size()); const ColumnArray::Offsets * offsets = nullptr; ColumnRawPtrs data_columns(arguments.size()); ColumnRawPtrs original_data_columns(arguments.size()); ColumnRawPtrs null_maps(arguments.size()); bool has_nullable_columns = false; for (size_t i = 0; i < arguments.size(); ++i) { ColumnPtr array_ptr = block.getByPosition(arguments[i]).column; const ColumnArray * array = checkAndGetColumn(array_ptr.get()); if (!array) { const ColumnConst * const_array = checkAndGetColumnConst( block.getByPosition(arguments[i]).column.get()); if (!const_array) throw Exception("Illegal column " + block.getByPosition(arguments[i]).column->getName() + " of " + toString(i + 1) + getOrdinalSuffix(i + 1) + " argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); array_ptr = const_array->convertToFullColumn(); array = static_cast(array_ptr.get()); } array_columns[i] = array_ptr; const ColumnArray::Offsets & offsets_i = array->getOffsets(); if (i == 0) offsets = &offsets_i; else if (offsets_i != *offsets) throw Exception("Lengths of all arrays passsed to " + getName() + " must be equal.", ErrorCodes::SIZES_OF_ARRAYS_DOESNT_MATCH); data_columns[i] = &array->getData(); original_data_columns[i] = data_columns[i]; if (data_columns[i]->isColumnNullable()) { has_nullable_columns = true; const auto & nullable_col = static_cast(*data_columns[i]); data_columns[i] = &nullable_col.getNestedColumn(); null_maps[i] = &nullable_col.getNullMapColumn(); } else null_maps[i] = nullptr; } const ColumnArray * first_array = static_cast(array_columns[0].get()); const IColumn * first_null_map = null_maps[0]; auto res = ColumnUInt32::create(); ColumnUInt32::Container & res_values = res->getData(); res_values.resize(offsets->size()); if (arguments.size() == 1) { if (!( executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeString(first_array, first_null_map, res_values))) throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } else { if (!execute128bit(*offsets, data_columns, null_maps, res_values, has_nullable_columns)) executeHashed(*offsets, original_data_columns, res_values); } block.getByPosition(result).column = std::move(res); } template bool FunctionArrayUniq::executeNumber(const ColumnArray * array, const IColumn * null_map, ColumnUInt32::Container & res_values) { const IColumn * inner_col; const auto & array_data = array->getData(); if (array_data.isColumnNullable()) { const auto & nullable_col = static_cast(array_data); inner_col = &nullable_col.getNestedColumn(); } else inner_col = &array_data; const ColumnVector * nested = checkAndGetColumn>(inner_col); if (!nested) return false; const ColumnArray::Offsets & offsets = array->getOffsets(); const typename ColumnVector::Container & values = nested->getData(); using Set = ClearableHashSet, HashTableGrower, HashTableAllocatorWithStackMemory<(1ULL << INITIAL_SIZE_DEGREE) * sizeof(T)>>; const PaddedPODArray * null_map_data = nullptr; if (null_map) null_map_data = &static_cast(null_map)->getData(); Set set; size_t prev_off = 0; for (size_t i = 0; i < offsets.size(); ++i) { set.clear(); bool found_null = false; size_t off = offsets[i]; for (size_t j = prev_off; j < off; ++j) { if (null_map_data && ((*null_map_data)[j] == 1)) found_null = true; else set.insert(values[j]); } res_values[i] = set.size() + found_null; prev_off = off; } return true; } bool FunctionArrayUniq::executeString(const ColumnArray * array, const IColumn * null_map, ColumnUInt32::Container & res_values) { const IColumn * inner_col; const auto & array_data = array->getData(); if (array_data.isColumnNullable()) { const auto & nullable_col = static_cast(array_data); inner_col = &nullable_col.getNestedColumn(); } else inner_col = &array_data; const ColumnString * nested = checkAndGetColumn(inner_col); if (!nested) return false; const ColumnArray::Offsets & offsets = array->getOffsets(); using Set = ClearableHashSet, HashTableAllocatorWithStackMemory<(1ULL << INITIAL_SIZE_DEGREE) * sizeof(StringRef)>>; const PaddedPODArray * null_map_data = nullptr; if (null_map) null_map_data = &static_cast(null_map)->getData(); Set set; size_t prev_off = 0; for (size_t i = 0; i < offsets.size(); ++i) { set.clear(); bool found_null = false; size_t off = offsets[i]; for (size_t j = prev_off; j < off; ++j) { if (null_map_data && ((*null_map_data)[j] == 1)) found_null = true; else set.insert(nested->getDataAt(j)); } res_values[i] = set.size() + found_null; prev_off = off; } return true; } bool FunctionArrayUniq::execute128bit( const ColumnArray::Offsets & offsets, const ColumnRawPtrs & columns, const ColumnRawPtrs & null_maps, ColumnUInt32::Container & res_values, bool has_nullable_columns) { size_t count = columns.size(); size_t keys_bytes = 0; Sizes key_sizes(count); for (size_t j = 0; j < count; ++j) { if (!columns[j]->isFixedAndContiguous()) return false; key_sizes[j] = columns[j]->sizeOfValueIfFixed(); keys_bytes += key_sizes[j]; } if (has_nullable_columns) keys_bytes += std::tuple_size>::value; if (keys_bytes > 16) return false; using Set = ClearableHashSet, HashTableAllocatorWithStackMemory<(1ULL << INITIAL_SIZE_DEGREE) * sizeof(UInt128)>>; /// Suppose that, for a given row, each of the N columns has an array whose length is M. /// Denote arr_i each of these arrays (1 <= i <= N). Then the following is performed: /// /// col1 ... colN /// /// arr_1[1], ..., arr_N[1] -> pack into a binary blob b1 /// . /// . /// . /// arr_1[M], ..., arr_N[M] -> pack into a binary blob bM /// /// Each binary blob is inserted into a hash table. /// Set set; size_t prev_off = 0; for (size_t i = 0; i < offsets.size(); ++i) { set.clear(); size_t off = offsets[i]; for (size_t j = prev_off; j < off; ++j) { if (has_nullable_columns) { KeysNullMap bitmap{}; for (size_t i = 0; i < columns.size(); ++i) { if (null_maps[i]) { const auto & null_map = static_cast(*null_maps[i]).getData(); if (null_map[j] == 1) { size_t bucket = i / 8; size_t offset = i % 8; bitmap[bucket] |= UInt8(1) << offset; } } } set.insert(packFixed(j, count, columns, key_sizes, bitmap)); } else set.insert(packFixed(j, count, columns, key_sizes)); } res_values[i] = set.size(); prev_off = off; } return true; } void FunctionArrayUniq::executeHashed( const ColumnArray::Offsets & offsets, const ColumnRawPtrs & columns, ColumnUInt32::Container & res_values) { size_t count = columns.size(); using Set = ClearableHashSet, HashTableAllocatorWithStackMemory<(1ULL << INITIAL_SIZE_DEGREE) * sizeof(UInt128)>>; Set set; size_t prev_off = 0; for (size_t i = 0; i < offsets.size(); ++i) { set.clear(); size_t off = offsets[i]; for (size_t j = prev_off; j < off; ++j) set.insert(hash128(j, count, columns)); res_values[i] = set.size(); prev_off = off; } } /// Implementation of FunctionArrayEnumerateUniq. FunctionPtr FunctionArrayEnumerateUniq::create(const Context &) { return std::make_shared(); } String FunctionArrayEnumerateUniq::getName() const { return name; } DataTypePtr FunctionArrayEnumerateUniq::getReturnTypeImpl(const DataTypes & arguments) const { if (arguments.size() == 0) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + toString(arguments.size()) + ", should be at least 1.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); for (size_t i = 0; i < arguments.size(); ++i) { const DataTypeArray * array_type = checkAndGetDataType(arguments[i].get()); if (!array_type) throw Exception("All arguments for function " + getName() + " must be arrays but argument " + toString(i + 1) + " has type " + arguments[i]->getName() + ".", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } return std::make_shared(std::make_shared()); } void FunctionArrayEnumerateUniq::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { Columns array_columns(arguments.size()); const ColumnArray::Offsets * offsets = nullptr; ColumnRawPtrs data_columns(arguments.size()); ColumnRawPtrs original_data_columns(arguments.size()); ColumnRawPtrs null_maps(arguments.size()); bool has_nullable_columns = false; for (size_t i = 0; i < arguments.size(); ++i) { ColumnPtr array_ptr = block.getByPosition(arguments[i]).column; const ColumnArray * array = checkAndGetColumn(array_ptr.get()); if (!array) { const ColumnConst * const_array = checkAndGetColumnConst( block.getByPosition(arguments[i]).column.get()); if (!const_array) throw Exception("Illegal column " + block.getByPosition(arguments[i]).column->getName() + " of " + toString(i + 1) + "-th argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); array_ptr = const_array->convertToFullColumn(); array = checkAndGetColumn(array_ptr.get()); } array_columns[i] = array_ptr; const ColumnArray::Offsets & offsets_i = array->getOffsets(); if (i == 0) offsets = &offsets_i; else if (offsets_i != *offsets) throw Exception("Lengths of all arrays passsed to " + getName() + " must be equal.", ErrorCodes::SIZES_OF_ARRAYS_DOESNT_MATCH); data_columns[i] = &array->getData(); original_data_columns[i] = data_columns[i]; if (data_columns[i]->isColumnNullable()) { has_nullable_columns = true; const auto & nullable_col = static_cast(*data_columns[i]); data_columns[i] = &nullable_col.getNestedColumn(); null_maps[i] = &nullable_col.getNullMapColumn(); } else null_maps[i] = nullptr; } const ColumnArray * first_array = checkAndGetColumn(array_columns[0].get()); const IColumn * first_null_map = null_maps[0]; auto res_nested = ColumnUInt32::create(); ColumnUInt32::Container & res_values = res_nested->getData(); if (!offsets->empty()) res_values.resize(offsets->back()); if (arguments.size() == 1) { if (!( executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeNumber(first_array, first_null_map, res_values) || executeString (first_array, first_null_map, res_values))) throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } else { if (!execute128bit(*offsets, data_columns, null_maps, res_values, has_nullable_columns)) executeHashed(*offsets, original_data_columns, res_values); } block.getByPosition(result).column = ColumnArray::create(std::move(res_nested), first_array->getOffsetsPtr()); } template bool FunctionArrayEnumerateUniq::executeNumber(const ColumnArray * array, const IColumn * null_map, ColumnUInt32::Container & res_values) { const IColumn * inner_col; const auto & array_data = array->getData(); if (array_data.isColumnNullable()) { const auto & nullable_col = static_cast(array_data); inner_col = &nullable_col.getNestedColumn(); } else inner_col = &array_data; const ColumnVector * nested = checkAndGetColumn>(inner_col); if (!nested) return false; const ColumnArray::Offsets & offsets = array->getOffsets(); const typename ColumnVector::Container & values = nested->getData(); using ValuesToIndices = ClearableHashMap, HashTableGrower, HashTableAllocatorWithStackMemory<(1ULL << INITIAL_SIZE_DEGREE) * sizeof(T)>>; const PaddedPODArray * null_map_data = nullptr; if (null_map) null_map_data = &static_cast(null_map)->getData(); ValuesToIndices indices; size_t prev_off = 0; for (size_t i = 0; i < offsets.size(); ++i) { indices.clear(); UInt32 null_count = 0; size_t off = offsets[i]; for (size_t j = prev_off; j < off; ++j) { if (null_map_data && ((*null_map_data)[j] == 1)) res_values[j] = ++null_count; else res_values[j] = ++indices[values[j]]; } prev_off = off; } return true; } bool FunctionArrayEnumerateUniq::executeString(const ColumnArray * array, const IColumn * null_map, ColumnUInt32::Container & res_values) { const IColumn * inner_col; const auto & array_data = array->getData(); if (array_data.isColumnNullable()) { const auto & nullable_col = static_cast(array_data); inner_col = &nullable_col.getNestedColumn(); } else inner_col = &array_data; const ColumnString * nested = checkAndGetColumn(inner_col); if (!nested) return false; const ColumnArray::Offsets & offsets = array->getOffsets(); size_t prev_off = 0; using ValuesToIndices = ClearableHashMap, HashTableAllocatorWithStackMemory<(1ULL << INITIAL_SIZE_DEGREE) * sizeof(StringRef)>>; const PaddedPODArray * null_map_data = nullptr; if (null_map) null_map_data = &static_cast(null_map)->getData(); ValuesToIndices indices; for (size_t i = 0; i < offsets.size(); ++i) { indices.clear(); UInt32 null_count = 0; size_t off = offsets[i]; for (size_t j = prev_off; j < off; ++j) { if (null_map_data && ((*null_map_data)[j] == 1)) res_values[j] = ++null_count; else res_values[j] = ++indices[nested->getDataAt(j)]; } prev_off = off; } return true; } bool FunctionArrayEnumerateUniq::execute128bit( const ColumnArray::Offsets & offsets, const ColumnRawPtrs & columns, const ColumnRawPtrs & null_maps, ColumnUInt32::Container & res_values, bool has_nullable_columns) { size_t count = columns.size(); size_t keys_bytes = 0; Sizes key_sizes(count); for (size_t j = 0; j < count; ++j) { if (!columns[j]->isFixedAndContiguous()) return false; key_sizes[j] = columns[j]->sizeOfValueIfFixed(); keys_bytes += key_sizes[j]; } if (has_nullable_columns) keys_bytes += std::tuple_size>::value; if (keys_bytes > 16) return false; using ValuesToIndices = ClearableHashMap, HashTableAllocatorWithStackMemory<(1ULL << INITIAL_SIZE_DEGREE) * sizeof(UInt128)>>; ValuesToIndices indices; size_t prev_off = 0; for (size_t i = 0; i < offsets.size(); ++i) { indices.clear(); size_t off = offsets[i]; for (size_t j = prev_off; j < off; ++j) { if (has_nullable_columns) { KeysNullMap bitmap{}; for (size_t i = 0; i < columns.size(); ++i) { if (null_maps[i]) { const auto & null_map = static_cast(*null_maps[i]).getData(); if (null_map[j] == 1) { size_t bucket = i / 8; size_t offset = i % 8; bitmap[bucket] |= UInt8(1) << offset; } } } res_values[j] = ++indices[packFixed(j, count, columns, key_sizes, bitmap)]; } else res_values[j] = ++indices[packFixed(j, count, columns, key_sizes)]; } prev_off = off; } return true; } void FunctionArrayEnumerateUniq::executeHashed( const ColumnArray::Offsets & offsets, const ColumnRawPtrs & columns, ColumnUInt32::Container & res_values) { size_t count = columns.size(); using ValuesToIndices = ClearableHashMap, HashTableAllocatorWithStackMemory<(1ULL << INITIAL_SIZE_DEGREE) * sizeof(UInt128)>>; ValuesToIndices indices; size_t prev_off = 0; for (size_t i = 0; i < offsets.size(); ++i) { indices.clear(); size_t off = offsets[i]; for (size_t j = prev_off; j < off; ++j) { res_values[j] = ++indices[hash128(j, count, columns)]; } prev_off = off; } } /// Implementation of FunctionEmptyArrayToSingle. FunctionPtr FunctionEmptyArrayToSingle::create(const Context &) { return std::make_shared(); } String FunctionEmptyArrayToSingle::getName() const { return name; } DataTypePtr FunctionEmptyArrayToSingle::getReturnTypeImpl(const DataTypes & arguments) const { const DataTypeArray * array_type = checkAndGetDataType(arguments[0].get()); if (!array_type) throw Exception("Argument for function " + getName() + " must be array.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return arguments[0]->clone(); } namespace { namespace FunctionEmptyArrayToSingleImpl { bool executeConst(Block & block, const ColumnNumbers & arguments, size_t result) { if (const ColumnConst * const_array = checkAndGetColumnConst(block.getByPosition(arguments[0]).column.get())) { if (const_array->getValue().empty()) { auto nested_type = typeid_cast(*block.getByPosition(arguments[0]).type).getNestedType(); block.getByPosition(result).column = block.getByPosition(result).type->createColumnConst( block.rows(), Array{nested_type->getDefault()}); } else block.getByPosition(result).column = block.getByPosition(arguments[0]).column; return true; } else return false; } template bool executeNumber( const IColumn & src_data, const ColumnArray::Offsets & src_offsets, IColumn & res_data_col, ColumnArray::Offsets & res_offsets, const NullMap * src_null_map, NullMap * res_null_map) { if (const ColumnVector * src_data_concrete = checkAndGetColumn>(&src_data)) { const PaddedPODArray & src_data = src_data_concrete->getData(); PaddedPODArray & res_data = static_cast &>(res_data_col).getData(); size_t size = src_offsets.size(); res_offsets.resize(size); res_data.reserve(src_data.size()); if (nullable) res_null_map->reserve(src_null_map->size()); ColumnArray::Offset src_prev_offset = 0; ColumnArray::Offset res_prev_offset = 0; for (size_t i = 0; i < size; ++i) { if (src_offsets[i] != src_prev_offset) { size_t size_to_write = src_offsets[i] - src_prev_offset; res_data.resize(res_prev_offset + size_to_write); memcpy(&res_data[res_prev_offset], &src_data[src_prev_offset], size_to_write * sizeof(T)); if (nullable) { res_null_map->resize(res_prev_offset + size_to_write); memcpy(&(*res_null_map)[res_prev_offset], &(*src_null_map)[src_prev_offset], size_to_write); } res_prev_offset += size_to_write; res_offsets[i] = res_prev_offset; } else { res_data.push_back(T()); ++res_prev_offset; res_offsets[i] = res_prev_offset; if (nullable) res_null_map->push_back(1); /// Push NULL. } src_prev_offset = src_offsets[i]; } return true; } else return false; } template bool executeFixedString( const IColumn & src_data, const ColumnArray::Offsets & src_offsets, IColumn & res_data_col, ColumnArray::Offsets & res_offsets, const NullMap * src_null_map, NullMap * res_null_map) { if (const ColumnFixedString * src_data_concrete = checkAndGetColumn(&src_data)) { const size_t n = src_data_concrete->getN(); const ColumnFixedString::Chars_t & src_data = src_data_concrete->getChars(); auto concrete_res_data = typeid_cast(&res_data_col); if (!concrete_res_data) throw Exception{"Internal error", ErrorCodes::LOGICAL_ERROR}; ColumnFixedString::Chars_t & res_data = concrete_res_data->getChars(); size_t size = src_offsets.size(); res_offsets.resize(size); res_data.reserve(src_data.size()); if (nullable) res_null_map->reserve(src_null_map->size()); ColumnArray::Offset src_prev_offset = 0; ColumnArray::Offset res_prev_offset = 0; for (size_t i = 0; i < size; ++i) { if (src_offsets[i] != src_prev_offset) { size_t size_to_write = src_offsets[i] - src_prev_offset; size_t prev_res_data_size = res_data.size(); res_data.resize(prev_res_data_size + size_to_write * n); memcpy(&res_data[prev_res_data_size], &src_data[src_prev_offset * n], size_to_write * n); if (nullable) { res_null_map->resize(res_prev_offset + size_to_write); memcpy(&(*res_null_map)[res_prev_offset], &(*src_null_map)[src_prev_offset], size_to_write); } res_prev_offset += size_to_write; res_offsets[i] = res_prev_offset; } else { size_t prev_res_data_size = res_data.size(); res_data.resize(prev_res_data_size + n); memset(&res_data[prev_res_data_size], 0, n); ++res_prev_offset; res_offsets[i] = res_prev_offset; if (nullable) res_null_map->push_back(1); } src_prev_offset = src_offsets[i]; } return true; } else return false; } template bool executeString( const IColumn & src_data, const ColumnArray::Offsets & src_array_offsets, IColumn & res_data_col, ColumnArray::Offsets & res_array_offsets, const NullMap * src_null_map, NullMap * res_null_map) { if (const ColumnString * src_data_concrete = checkAndGetColumn(&src_data)) { const ColumnString::Offsets & src_string_offsets = src_data_concrete->getOffsets(); auto concrete_res_string_offsets = typeid_cast(&res_data_col); if (!concrete_res_string_offsets) throw Exception{"Internal error", ErrorCodes::LOGICAL_ERROR}; ColumnString::Offsets & res_string_offsets = concrete_res_string_offsets->getOffsets(); const ColumnString::Chars_t & src_data = src_data_concrete->getChars(); auto concrete_res_data = typeid_cast(&res_data_col); if (!concrete_res_data) throw Exception{"Internal error", ErrorCodes::LOGICAL_ERROR}; ColumnString::Chars_t & res_data = concrete_res_data->getChars(); size_t size = src_array_offsets.size(); res_array_offsets.resize(size); res_string_offsets.reserve(src_string_offsets.size()); res_data.reserve(src_data.size()); if (nullable) res_null_map->reserve(src_null_map->size()); ColumnArray::Offset src_array_prev_offset = 0; ColumnArray::Offset res_array_prev_offset = 0; ColumnString::Offset src_string_prev_offset = 0; ColumnString::Offset res_string_prev_offset = 0; for (size_t i = 0; i < size; ++i) { if (src_array_offsets[i] != src_array_prev_offset) { size_t array_size = src_array_offsets[i] - src_array_prev_offset; size_t bytes_to_copy = 0; size_t from_string_prev_offset_local = src_string_prev_offset; for (size_t j = 0; j < array_size; ++j) { size_t string_size = src_string_offsets[src_array_prev_offset + j] - from_string_prev_offset_local; res_string_prev_offset += string_size; res_string_offsets.push_back(res_string_prev_offset); from_string_prev_offset_local += string_size; bytes_to_copy += string_size; } size_t res_data_old_size = res_data.size(); res_data.resize(res_data_old_size + bytes_to_copy); memcpy(&res_data[res_data_old_size], &src_data[src_string_prev_offset], bytes_to_copy); if (nullable) { res_null_map->resize(res_array_prev_offset + array_size); memcpy(&(*res_null_map)[res_array_prev_offset], &(*src_null_map)[src_array_prev_offset], array_size); } res_array_prev_offset += array_size; res_array_offsets[i] = res_array_prev_offset; } else { res_data.push_back(0); /// An empty string, including zero at the end. if (nullable) res_null_map->push_back(1); ++res_string_prev_offset; res_string_offsets.push_back(res_string_prev_offset); ++res_array_prev_offset; res_array_offsets[i] = res_array_prev_offset; } src_array_prev_offset = src_array_offsets[i]; if (src_array_prev_offset) src_string_prev_offset = src_string_offsets[src_array_prev_offset - 1]; } return true; } else return false; } template void executeGeneric( const IColumn & src_data, const ColumnArray::Offsets & src_offsets, IColumn & res_data, ColumnArray::Offsets & res_offsets, const NullMap * src_null_map, NullMap * res_null_map) { size_t size = src_offsets.size(); res_offsets.resize(size); res_data.reserve(src_data.size()); if (nullable) res_null_map->reserve(src_null_map->size()); ColumnArray::Offset src_prev_offset = 0; ColumnArray::Offset res_prev_offset = 0; for (size_t i = 0; i < size; ++i) { if (src_offsets[i] != src_prev_offset) { size_t size_to_write = src_offsets[i] - src_prev_offset; res_data.insertRangeFrom(src_data, src_prev_offset, size_to_write); if (nullable) { res_null_map->resize(res_prev_offset + size_to_write); memcpy(&(*res_null_map)[res_prev_offset], &(*src_null_map)[src_prev_offset], size_to_write); } res_prev_offset += size_to_write; res_offsets[i] = res_prev_offset; } else { res_data.insertDefault(); ++res_prev_offset; res_offsets[i] = res_prev_offset; if (nullable) res_null_map->push_back(1); } src_prev_offset = src_offsets[i]; } } template void executeDispatch( const IColumn & src_data, const ColumnArray::Offsets & src_array_offsets, IColumn & res_data_col, ColumnArray::Offsets & res_array_offsets, const NullMap * src_null_map, NullMap * res_null_map) { if (!( executeNumber (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeNumber (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeNumber (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeNumber (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeNumber (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeNumber (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeNumber (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeNumber (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeNumber(src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeNumber(src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeString (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map) || executeFixedString (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map))) executeGeneric (src_data, src_array_offsets, res_data_col, res_array_offsets, src_null_map, res_null_map); } } } void FunctionEmptyArrayToSingle::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { if (FunctionEmptyArrayToSingleImpl::executeConst(block, arguments, result)) return; const ColumnArray * array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get()); if (!array) throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); MutableColumnPtr res_ptr = array->cloneEmpty(); ColumnArray & res = static_cast(*res_ptr); const IColumn & src_data = array->getData(); const ColumnArray::Offsets & src_offsets = array->getOffsets(); IColumn & res_data = res.getData(); ColumnArray::Offsets & res_offsets = res.getOffsets(); const NullMap * src_null_map = nullptr; NullMap * res_null_map = nullptr; const IColumn * inner_col; IColumn * inner_res_col; bool nullable = src_data.isColumnNullable(); if (nullable) { auto nullable_col = static_cast(&src_data); inner_col = &nullable_col->getNestedColumn(); src_null_map = &nullable_col->getNullMapData(); auto nullable_res_col = static_cast(&res_data); inner_res_col = &nullable_res_col->getNestedColumn(); res_null_map = &nullable_res_col->getNullMapData(); } else { inner_col = &src_data; inner_res_col = &res_data; } if (nullable) FunctionEmptyArrayToSingleImpl::executeDispatch(*inner_col, src_offsets, *inner_res_col, res_offsets, src_null_map, res_null_map); else FunctionEmptyArrayToSingleImpl::executeDispatch(*inner_col, src_offsets, *inner_res_col, res_offsets, src_null_map, res_null_map); block.getByPosition(result).column = std::move(res_ptr); } /// Implementation of FunctionRange. String FunctionRange::getName() const { return name; } DataTypePtr FunctionRange::getReturnTypeImpl(const DataTypes & arguments) const { const auto arg = arguments.front().get(); if (!arg->canBeUsedAsNonNegativeArrayIndex()) throw Exception{ "Illegal type " + arg->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT}; return std::make_shared(arg->clone()); } template bool FunctionRange::executeInternal(Block & block, const IColumn * arg, const size_t result) { static constexpr size_t max_elements = 100'000'000; if (const auto in = checkAndGetColumn>(arg)) { const auto & in_data = in->getData(); const auto total_values = std::accumulate(std::begin(in_data), std::end(in_data), size_t{}, [this] (const size_t lhs, const size_t rhs) { const auto sum = lhs + rhs; if (sum < lhs) throw Exception{ "A call to function " + getName() + " overflows, investigate the values of arguments you are passing", ErrorCodes::ARGUMENT_OUT_OF_BOUND}; return sum; }); if (total_values > max_elements) throw Exception{ "A call to function " + getName() + " would produce " + std::to_string(total_values) + " array elements, which is greater than the allowed maximum of " + std::to_string(max_elements), ErrorCodes::ARGUMENT_OUT_OF_BOUND}; const auto data_col = ColumnVector::create(total_values); const auto out = ColumnArray::create( data_col, std::make_shared(in->size())); block.getByPosition(result).column = out; auto & out_data = data_col->getData(); auto & out_offsets = out->getOffsets(); IColumn::Offset offset{}; for (size_t row_idx = 0, rows = in->size(); row_idx < rows; ++row_idx) { for (size_t elem_idx = 0, elems = in_data[row_idx]; elem_idx < elems; ++elem_idx) out_data[offset + elem_idx] = elem_idx; offset += in_data[row_idx]; out_offsets[row_idx] = offset; } return true; } else if (const auto in = checkAndGetColumnConst>(arg)) { T in_data = in->template getValue(); if ((in_data != 0) && (in->size() > (std::numeric_limits::max() / in_data))) throw Exception{ "A call to function " + getName() + " overflows, investigate the values of arguments you are passing", ErrorCodes::ARGUMENT_OUT_OF_BOUND}; const size_t total_values = in->size() * in_data; if (total_values > max_elements) throw Exception{ "A call to function " + getName() + " would produce " + toString(total_values) + " array elements, which is greater than the allowed maximum of " + toString(max_elements), ErrorCodes::ARGUMENT_OUT_OF_BOUND}; const auto data_col = ColumnVector::create(total_values); const auto out = ColumnArray::create( data_col, std::make_shared(in->size())); block.getByPosition(result).column = out; auto & out_data = data_col->getData(); auto & out_offsets = out->getOffsets(); IColumn::Offset offset{}; for (size_t row_idx = 0, rows = in->size(); row_idx < rows; ++row_idx) { for (size_t elem_idx = 0, elems = in_data; elem_idx < elems; ++elem_idx) out_data[offset + elem_idx] = elem_idx; offset += in_data; out_offsets[row_idx] = offset; } return true; } else return false; } void FunctionRange::executeImpl(Block & block, const ColumnNumbers & arguments, const size_t result) { const auto col = block.getByPosition(arguments[0]).column.get(); if (!executeInternal(block, col, result) && !executeInternal(block, col, result) && !executeInternal(block, col, result) && !executeInternal(block, col, result)) { throw Exception{ "Illegal column " + col->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN}; } } /// Implementation of FunctionArrayReverse. FunctionPtr FunctionArrayReverse::create(const Context &) { return std::make_shared(); } String FunctionArrayReverse::getName() const { return name; } DataTypePtr FunctionArrayReverse::getReturnTypeImpl(const DataTypes & arguments) const { const DataTypeArray * array_type = checkAndGetDataType(arguments[0].get()); if (!array_type) throw Exception("Argument for function " + getName() + " must be array.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return arguments[0]->clone(); } void FunctionArrayReverse::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { if (executeConst(block, arguments, result)) return; const ColumnArray * array = checkAndGetColumn(block.getByPosition(arguments[0]).column.get()); if (!array) throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); MutableColumnPtr res_ptr = array->cloneEmpty(); ColumnArray & res = static_cast(*res_ptr); const IColumn & src_data = array->getData(); const ColumnArray::Offsets & offsets = array->getOffsets(); IColumn & res_data = res.getData(); const ColumnNullable * nullable_col = nullptr; ColumnNullable * nullable_res_col = nullptr; const IColumn * inner_col; IColumn * inner_res_col; if (src_data.isColumnNullable()) { nullable_col = static_cast(&src_data); inner_col = &nullable_col->getNestedColumn(); nullable_res_col = static_cast(&res_data); inner_res_col = &nullable_res_col->getNestedColumn(); } else { inner_col = &src_data; inner_res_col = &res_data; } if (!( executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeNumber(*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeString (*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col) || executeFixedString (*inner_col, offsets, *inner_res_col, nullable_col, nullable_res_col))) throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); res_ptr->getOffsetsPtr() = array->getOffsetsPtr(); block.getByPosition(result).column = std::move(res_ptr); } bool FunctionArrayReverse::executeConst(Block & block, const ColumnNumbers & arguments, size_t result) { if (const ColumnConst * const_array = checkAndGetColumnConst(block.getByPosition(arguments[0]).column.get())) { Array arr = const_array->getValue(); size_t size = arr.size(); Array res(size); for (size_t i = 0; i < size; ++i) res[i] = arr[size - i - 1]; block.getByPosition(result).column = block.getByPosition(result).type->createColumnConst(block.rows(), res); return true; } else return false; } template bool FunctionArrayReverse::executeNumber( const IColumn & src_data, const ColumnArray::Offsets & src_offsets, IColumn & res_data_col, const ColumnNullable * nullable_col, ColumnNullable * nullable_res_col) { auto do_reverse = [](const auto & src_data, const auto & src_offsets, auto & res_data) { size_t size = src_offsets.size(); ColumnArray::Offset src_prev_offset = 0; for (size_t i = 0; i < size; ++i) { const auto * src = &src_data[src_prev_offset]; const auto * src_end = &src_data[src_offsets[i]]; if (src == src_end) continue; auto dst = &res_data[src_offsets[i] - 1]; while (src < src_end) { *dst = *src; ++src; --dst; } src_prev_offset = src_offsets[i]; } }; if (const ColumnVector * src_data_concrete = checkAndGetColumn>(&src_data)) { const PaddedPODArray & src_data = src_data_concrete->getData(); PaddedPODArray & res_data = typeid_cast &>(res_data_col).getData(); res_data.resize(src_data.size()); do_reverse(src_data, src_offsets, res_data); if ((nullable_col) && (nullable_res_col)) { /// Make a reverted null map. const auto & src_null_map = static_cast(nullable_col->getNullMapColumn()).getData(); auto & res_null_map = static_cast(nullable_res_col->getNullMapColumn()).getData(); res_null_map.resize(src_data.size()); do_reverse(src_null_map, src_offsets, res_null_map); } return true; } else return false; } bool FunctionArrayReverse::executeFixedString( const IColumn & src_data, const ColumnArray::Offsets & src_offsets, IColumn & res_data_col, const ColumnNullable * nullable_col, ColumnNullable * nullable_res_col) { if (const ColumnFixedString * src_data_concrete = checkAndGetColumn(&src_data)) { const size_t n = src_data_concrete->getN(); const ColumnFixedString::Chars_t & src_data = src_data_concrete->getChars(); ColumnFixedString::Chars_t & res_data = typeid_cast(res_data_col).getChars(); size_t size = src_offsets.size(); res_data.resize(src_data.size()); ColumnArray::Offset src_prev_offset = 0; for (size_t i = 0; i < size; ++i) { const UInt8 * src = &src_data[src_prev_offset * n]; const UInt8 * src_end = &src_data[src_offsets[i] * n]; if (src == src_end) continue; UInt8 * dst = &res_data[src_offsets[i] * n - n]; while (src < src_end) { /// NOTE: memcpySmallAllowReadWriteOverflow15 doesn't work correctly here. memcpy(dst, src, n); src += n; dst -= n; } src_prev_offset = src_offsets[i]; } if ((nullable_col) && (nullable_res_col)) { /// Make a reverted null map. const auto & src_null_map = static_cast(nullable_col->getNullMapColumn()).getData(); auto & res_null_map = static_cast(nullable_res_col->getNullMapColumn()).getData(); res_null_map.resize(src_null_map.size()); ColumnArray::Offset src_prev_offset = 0; for (size_t i = 0; i < size; ++i) { const UInt8 * src = &src_null_map[src_prev_offset]; const UInt8 * src_end = &src_null_map[src_offsets[i]]; if (src == src_end) continue; UInt8 * dst = &res_null_map[src_offsets[i] - 1]; while (src < src_end) { *dst = *src; ++src; --dst; } src_prev_offset = src_offsets[i]; } } return true; } else return false; } bool FunctionArrayReverse::executeString( const IColumn & src_data, const ColumnArray::Offsets & src_array_offsets, IColumn & res_data_col, const ColumnNullable * nullable_col, ColumnNullable * nullable_res_col) { if (const ColumnString * src_data_concrete = checkAndGetColumn(&src_data)) { const ColumnString::Offsets & src_string_offsets = src_data_concrete->getOffsets(); ColumnString::Offsets & res_string_offsets = typeid_cast(res_data_col).getOffsets(); const ColumnString::Chars_t & src_data = src_data_concrete->getChars(); ColumnString::Chars_t & res_data = typeid_cast(res_data_col).getChars(); size_t size = src_array_offsets.size(); res_string_offsets.resize(src_string_offsets.size()); res_data.resize(src_data.size()); ColumnArray::Offset src_array_prev_offset = 0; ColumnString::Offset res_string_prev_offset = 0; for (size_t i = 0; i < size; ++i) { if (src_array_offsets[i] != src_array_prev_offset) { size_t array_size = src_array_offsets[i] - src_array_prev_offset; for (size_t j = 0; j < array_size; ++j) { size_t j_reversed = array_size - j - 1; auto src_pos = src_array_prev_offset + j_reversed == 0 ? 0 : src_string_offsets[src_array_prev_offset + j_reversed - 1]; size_t string_size = src_string_offsets[src_array_prev_offset + j_reversed] - src_pos; memcpySmallAllowReadWriteOverflow15(&res_data[res_string_prev_offset], &src_data[src_pos], string_size); res_string_prev_offset += string_size; res_string_offsets[src_array_prev_offset + j] = res_string_prev_offset; } } src_array_prev_offset = src_array_offsets[i]; } if ((nullable_col) && (nullable_res_col)) { /// Make a reverted null map. const auto & src_null_map = static_cast(nullable_col->getNullMapColumn()).getData(); auto & res_null_map = static_cast(nullable_res_col->getNullMapColumn()).getData(); res_null_map.resize(src_string_offsets.size()); size_t size = src_string_offsets.size(); ColumnArray::Offset src_prev_offset = 0; for (size_t i = 0; i < size; ++i) { const auto * src = &src_null_map[src_prev_offset]; const auto * src_end = &src_null_map[src_array_offsets[i]]; if (src == src_end) continue; auto dst = &res_null_map[src_array_offsets[i] - 1]; while (src < src_end) { *dst = *src; ++src; --dst; } src_prev_offset = src_array_offsets[i]; } } return true; } else return false; } /// Implementation of FunctionArrayReduce. FunctionPtr FunctionArrayReduce::create(const Context &) { return std::make_shared(); } String FunctionArrayReduce::getName() const { return name; } void FunctionArrayReduce::getReturnTypeAndPrerequisitesImpl( const ColumnsWithTypeAndName & arguments, DataTypePtr & out_return_type, std::vector & /*out_prerequisites*/) { /// The first argument is a constant string with the name of the aggregate function /// (possibly with parameters in parentheses, for example: "quantile(0.99)"). if (arguments.size() < 2) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + toString(arguments.size()) + ", should be at least 2.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); const ColumnConst * aggregate_function_name_column = checkAndGetColumnConst(arguments[0].column.get()); if (!aggregate_function_name_column) throw Exception("First argument for function " + getName() + " must be constant string: name of aggregate function.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); DataTypes argument_types(arguments.size() - 1); for (size_t i = 1, size = arguments.size(); i < size; ++i) { const DataTypeArray * arg = checkAndGetDataType(arguments[i].type.get()); if (!arg) throw Exception("Argument " + toString(i) + " for function " + getName() + " must be an array but it has type " + arguments[i].type->getName() + ".", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); argument_types[i - 1] = arg->getNestedType()->clone(); } if (!aggregate_function) { String aggregate_function_name_with_params = aggregate_function_name_column->getValue(); if (aggregate_function_name_with_params.empty()) throw Exception("First argument for function " + getName() + " (name of aggregate function) cannot be empty.", ErrorCodes::BAD_ARGUMENTS); String aggregate_function_name; Array params_row; getAggregateFunctionNameAndParametersArray(aggregate_function_name_with_params, aggregate_function_name, params_row, "function " + getName()); aggregate_function = AggregateFunctionFactory::instance().get(aggregate_function_name, argument_types, params_row); if (!params_row.empty()) aggregate_function->setParameters(params_row); aggregate_function->setArguments(argument_types); } out_return_type = aggregate_function->getReturnType(); } void FunctionArrayReduce::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { IAggregateFunction & agg_func = *aggregate_function.get(); std::unique_ptr place_holder { new char[agg_func.sizeOfData()] }; AggregateDataPtr place = place_holder.get(); std::unique_ptr arena = agg_func.allocatesMemoryInArena() ? std::make_unique() : nullptr; size_t rows = block.rows(); /// Aggregate functions do not support constant columns. Therefore, we materialize them. std::vector materialized_columns; std::vector aggregate_arguments_vec(arguments.size() - 1); bool is_const = true; for (size_t i = 0, size = arguments.size() - 1; i < size; ++i) { const IColumn * col = block.getByPosition(arguments[i + 1]).column.get(); if (const ColumnArray * arr = checkAndGetColumn(col)) { aggregate_arguments_vec[i] = &arr->getData(); is_const = false; } else if (const ColumnConst * arr = checkAndGetColumnConst(col)) { materialized_columns.emplace_back(arr->convertToFullColumn()); aggregate_arguments_vec[i] = &typeid_cast(*materialized_columns.back().get()).getData(); } else throw Exception("Illegal column " + col->getName() + " as argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } const IColumn ** aggregate_arguments = aggregate_arguments_vec.data(); const ColumnArray::Offsets & offsets = typeid_cast(!materialized_columns.empty() ? *materialized_columns.front().get() : *block.getByPosition(arguments[1]).column.get()).getOffsets(); MutableColumnPtr result_holder = block.getByPosition(result).type->createColumn(); IColumn & res_col = *result_holder; /// AggregateFunction's states should be inserted into column using specific way auto res_col_aggregate_function = typeid_cast(&res_col); if (!res_col_aggregate_function && agg_func.isState()) throw Exception("State function " + agg_func.getName() + " inserts results into non-state column " + block.getByPosition(result).type->getName(), ErrorCodes::ILLEGAL_COLUMN); ColumnArray::Offset current_offset = 0; for (size_t i = 0; i < rows; ++i) { agg_func.create(place); ColumnArray::Offset next_offset = offsets[i]; try { for (size_t j = current_offset; j < next_offset; ++j) agg_func.add(place, aggregate_arguments, j, arena.get()); if (!res_col_aggregate_function) agg_func.insertResultInto(place, res_col); else res_col_aggregate_function->insertFrom(place); } catch (...) { agg_func.destroy(place); throw; } agg_func.destroy(place); current_offset = next_offset; } if (!is_const) { block.getByPosition(result).column = std::move(result_holder); } else { block.getByPosition(result).column = block.getByPosition(result).type->createColumnConst(rows, res_col[0]); } } /// Implementation of FunctionArrayConcat. FunctionPtr FunctionArrayConcat::create(const Context & context) { return std::make_shared(context); } String FunctionArrayConcat::getName() const { return name; } DataTypePtr FunctionArrayConcat::getReturnTypeImpl(const DataTypes & arguments) const { if (arguments.empty()) throw Exception{"Function array requires at least one argument.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH}; auto array_type = typeid_cast(arguments[0].get()); if (!array_type) throw Exception("First argument for function " + getName() + " must be an array but it has type " + arguments[0]->getName() + ".", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return getLeastCommonType(arguments); } void FunctionArrayConcat::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { auto & result_column = block.getByPosition(result).column; const DataTypePtr & return_type = block.getByPosition(result).type; result_column = return_type->createColumn(); if (return_type->onlyNull()) { result_column = return_type->createColumnConst(block.rows(), Null()); return; } size_t rows = block.rows(); size_t num_args = arguments.size(); Columns preprocessed_columns(num_args); for (size_t i = 0; i < num_args; ++i) { const ColumnWithTypeAndName & arg = block.getByPosition(arguments[i]); ColumnPtr preprocessed_column = arg.column; if (!arg.type->equals(*return_type)) preprocessed_column = castColumn(arg, return_type, context); preprocessed_columns[i] = std::move(preprocessed_column); } std::vector> sources; for (auto & argument_column : preprocessed_columns) { bool is_const = false; if (auto argument_column_const = typeid_cast(argument_column.get())) { is_const = true; argument_column = argument_column_const->getDataColumnPtr(); } if (auto argument_column_array = typeid_cast(argument_column.get())) sources.emplace_back(createArraySource(*argument_column_array, is_const, rows)); else throw Exception{"Arguments for function " + getName() + " must be arrays.", ErrorCodes::LOGICAL_ERROR}; } auto sink = createArraySink(typeid_cast(*result_column.get()), rows); concat(sources, *sink); } /// Implementation of FunctionArraySlice. FunctionPtr FunctionArraySlice::create(const Context &) { return std::make_shared(); } String FunctionArraySlice::getName() const { return name; } DataTypePtr FunctionArraySlice::getReturnTypeImpl(const DataTypes & arguments) const { size_t number_of_arguments = arguments.size(); if (number_of_arguments < 2 || number_of_arguments > 3) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + toString(number_of_arguments) + ", should be 2 or 3", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); if (arguments[0]->onlyNull()) return arguments[0]; auto array_type = typeid_cast(arguments[0].get()); if (!array_type) throw Exception("First argument for function " + getName() + " must be an array but it has type " + arguments[0]->getName() + ".", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); for (size_t i = 1; i < number_of_arguments; ++i) { if (!removeNullable(arguments[i])->isInteger() && !arguments[i]->onlyNull()) throw Exception( "Argument " + toString(i) + " for function " + getName() + " must be integer but it has type " + arguments[i]->getName() + ".", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } return arguments[0]; } void FunctionArraySlice::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { auto & result_colum_with_name_and_type = block.getByPosition(result); auto & result_column = result_colum_with_name_and_type.column; auto & return_type = result_colum_with_name_and_type.type; result_column = return_type->createColumn(); auto array_column = block.getByPosition(arguments[0]).column; auto offset_column = block.getByPosition(arguments[1]).column; auto length_column = arguments.size() > 2 ? block.getByPosition(arguments[2]).column : nullptr; if (return_type->onlyNull()) { result_column = array_column; return; } std::unique_ptr source; size_t size = array_column->size(); bool is_const = false; if (auto const_array_column = typeid_cast(array_column.get())) { is_const = true; array_column = const_array_column->getDataColumnPtr(); } if (auto argument_column_array = typeid_cast(array_column.get())) source = createArraySource(*argument_column_array, is_const, size); else throw Exception{"First arguments for function " + getName() + " must be array.", ErrorCodes::LOGICAL_ERROR}; auto sink = createArraySink(typeid_cast(*result_column), size); if (offset_column->onlyNull()) { if (!length_column || length_column->onlyNull()) result_column = array_column; else if (length_column->isColumnConst()) sliceFromLeftConstantOffsetBounded(*source, *sink, 0, length_column->getInt(0)); else { auto const_offset_column = ColumnConst::create(ColumnInt8::create(1, 1), size); sliceDynamicOffsetBounded(*source, *sink, *const_offset_column, *length_column); } } else if (offset_column->isColumnConst()) { ssize_t offset = offset_column->getUInt(0); if (!length_column || length_column->onlyNull()) { if (offset > 0) sliceFromLeftConstantOffsetUnbounded(*source, *sink, static_cast(offset - 1)); else sliceFromRightConstantOffsetUnbounded(*source, *sink, static_cast(-offset)); } else if (length_column->isColumnConst()) { ssize_t length = length_column->getInt(0); if (offset > 0) sliceFromLeftConstantOffsetBounded(*source, *sink, static_cast(offset - 1), length); else sliceFromRightConstantOffsetBounded(*source, *sink, static_cast(-offset), length); } else sliceDynamicOffsetBounded(*source, *sink, *offset_column, *length_column); } else { if (!length_column || length_column->onlyNull()) sliceDynamicOffsetUnbounded(*source, *sink, *offset_column); else sliceDynamicOffsetBounded(*source, *sink, *offset_column, *length_column); } } /// Implementation of FunctionArrayPush. DataTypePtr FunctionArrayPush::getReturnTypeImpl(const DataTypes & arguments) const { if (arguments[0]->onlyNull()) return arguments[0]; auto array_type = typeid_cast(arguments[0].get()); if (!array_type) throw Exception("First argument for function " + getName() + " must be an array but it has type " + arguments[0]->getName() + ".", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); auto nested_type = array_type->getNestedType(); DataTypes types = {nested_type, arguments[1]}; return std::make_shared(getLeastCommonType(types)); } void FunctionArrayPush::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { auto & result_colum_with_name_and_type = block.getByPosition(result); auto & result_column = result_colum_with_name_and_type.column; auto & return_type = result_colum_with_name_and_type.type; auto array_column = block.getByPosition(arguments[0]).column; auto appended_column = block.getByPosition(arguments[1]).column; if (return_type->onlyNull()) { result_column = array_column; return; } result_column = return_type->createColumn(); if (!block.getByPosition(arguments[0]).type->equals(*return_type)) array_column = castColumn(block.getByPosition(arguments[0]), return_type, context); const DataTypePtr & return_nested_type = typeid_cast(*return_type).getNestedType(); if (!block.getByPosition(arguments[1]).type->equals(*return_nested_type)) appended_column = castColumn(block.getByPosition(arguments[1]), return_nested_type, context); std::vector> sources; size_t size = array_column->size(); bool is_const = false; if (auto const_array_column = typeid_cast(array_column.get())) { is_const = true; array_column = const_array_column->getDataColumnPtr(); } if (auto argument_column_array = typeid_cast(array_column.get())) sources.push_back(createArraySource(*argument_column_array, is_const, size)); else throw Exception{"First arguments for function " + getName() + " must be array.", ErrorCodes::LOGICAL_ERROR}; bool is_appended_const = false; if (auto const_appended_column = typeid_cast(appended_column.get())) { is_appended_const = true; appended_column = const_appended_column->getDataColumnPtr(); } auto offsets = std::make_shared(appended_column->size()); for (size_t i : ext::range(0, offsets->size())) offsets->getElement(i) = i + 1; ColumnArray::Ptr appended_array_column = ColumnArray::create(appended_column, offsets); sources.push_back(createArraySource(appended_array_column, is_appended_const, size)); auto sink = createArraySink(typeid_cast(*result_column), size); if (push_front) sources[0].swap(sources[1]); concat(sources, *sink); } /// Implementation of FunctionArrayPushFront. FunctionPtr FunctionArrayPushFront::create(const Context & context) { return std::make_shared(context); } /// Implementation of FunctionArrayPushBack. FunctionPtr FunctionArrayPushBack::create(const Context & context) { return std::make_shared(context); } /// Implementation of FunctionArrayPop. DataTypePtr FunctionArrayPop::getReturnTypeImpl(const DataTypes & arguments) const { if (arguments[0]->onlyNull()) return arguments[0]; auto array_type = typeid_cast(arguments[0].get()); if (!array_type) throw Exception("First argument for function " + getName() + " must be an array but it has type " + arguments[0]->getName() + ".", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return arguments[0]; } void FunctionArrayPop::executeImpl(Block & block, const ColumnNumbers & arguments, size_t result) { auto & result_colum_with_name_and_type = block.getByPosition(result); auto & result_column = result_colum_with_name_and_type.column; auto & return_type = result_colum_with_name_and_type.type; result_column = return_type->createColumn(); auto array_column = block.getByPosition(arguments[0]).column; if (return_type->onlyNull()) { result_column = array_column; return; } std::unique_ptr source; size_t size = array_column->size(); bool is_const = false; if (auto const_array_column = typeid_cast(array_column.get())) { is_const = true; array_column = const_array_column->getDataColumnPtr(); } if (auto argument_column_array = typeid_cast(array_column.get())) source = createArraySource(*argument_column_array, is_const, size); else throw Exception{"First arguments for function " + getName() + " must be array.", ErrorCodes::LOGICAL_ERROR}; auto sink = createArraySink(typeid_cast(*result_column), size); if (pop_front) sliceFromLeftConstantOffsetUnbounded(*source, *sink, 1); else sliceFromLeftConstantOffsetBounded(*source, *sink, 0, -1); } /// Implementation of FunctionArrayPopFront. FunctionPtr FunctionArrayPopFront::create(const Context &) { return std::make_shared(); } /// Implementation of FunctionArrayPopBack. FunctionPtr FunctionArrayPopBack::create(const Context &) { return std::make_shared(); } }