mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-24 18:50:49 +00:00
333 lines
11 KiB
C++
333 lines
11 KiB
C++
#pragma once
|
||
|
||
#include <string.h> // memcpy
|
||
|
||
#include <Poco/SharedPtr.h>
|
||
|
||
#include <DB/Common/Exception.h>
|
||
#include <DB/Common/Arena.h>
|
||
|
||
#include <DB/Columns/IColumn.h>
|
||
#include <DB/Columns/ColumnsNumber.h>
|
||
#include <DB/Columns/ColumnString.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
namespace ErrorCodes
|
||
{
|
||
extern const int ILLEGAL_COLUMN;
|
||
extern const int NOT_IMPLEMENTED;
|
||
extern const int BAD_ARGUMENTS;
|
||
}
|
||
|
||
using Poco::SharedPtr;
|
||
|
||
/** Cтолбeц значений типа массив.
|
||
* В памяти он представлен, как один столбец вложенного типа, размер которого равен сумме размеров всех массивов,
|
||
* и как массив смещений в нём, который позволяет достать каждый элемент.
|
||
*/
|
||
class ColumnArray final : public IColumn
|
||
{
|
||
public:
|
||
/** По индексу i находится смещение до начала i + 1 -го элемента. */
|
||
typedef ColumnVector<Offset_t> ColumnOffsets_t;
|
||
|
||
/** Создать пустой столбец массивов, с типом значений, как в столбце nested_column */
|
||
explicit ColumnArray(ColumnPtr nested_column, ColumnPtr offsets_column = nullptr)
|
||
: data(nested_column), offsets(offsets_column)
|
||
{
|
||
if (!offsets_column)
|
||
{
|
||
offsets = std::make_shared<ColumnOffsets_t>();
|
||
}
|
||
else
|
||
{
|
||
if (!typeid_cast<ColumnOffsets_t *>(&*offsets_column))
|
||
throw Exception("offsets_column must be a ColumnUInt64", ErrorCodes::ILLEGAL_COLUMN);
|
||
}
|
||
}
|
||
|
||
std::string getName() const override { return "ColumnArray(" + getData().getName() + ")"; }
|
||
|
||
ColumnPtr cloneEmpty() const override
|
||
{
|
||
return std::make_shared<ColumnArray>(getData().cloneEmpty());
|
||
}
|
||
|
||
size_t size() const override
|
||
{
|
||
return getOffsets().size();
|
||
}
|
||
|
||
Field operator[](size_t n) const override
|
||
{
|
||
size_t offset = offsetAt(n);
|
||
size_t size = sizeAt(n);
|
||
Array res(size);
|
||
|
||
for (size_t i = 0; i < size; ++i)
|
||
res[i] = getData()[offset + i];
|
||
|
||
return res;
|
||
}
|
||
|
||
void get(size_t n, Field & res) const override
|
||
{
|
||
size_t offset = offsetAt(n);
|
||
size_t size = sizeAt(n);
|
||
res = Array(size);
|
||
Array & res_arr = DB::get<Array &>(res);
|
||
|
||
for (size_t i = 0; i < size; ++i)
|
||
getData().get(offset + i, res_arr[i]);
|
||
}
|
||
|
||
StringRef getDataAt(size_t n) const override
|
||
{
|
||
/** Возвращает диапазон памяти, покрывающий все элементы массива.
|
||
* Работает для массивов значений фиксированной длины.
|
||
* Для массивов строк и массивов массивов полученный кусок памяти может не взаимно-однозначно соответствовать элементам,
|
||
* так как содержит лишь уложенные подряд данные, но не смещения.
|
||
*/
|
||
|
||
size_t array_size = sizeAt(n);
|
||
if (array_size == 0)
|
||
return StringRef();
|
||
|
||
size_t offset_of_first_elem = offsetAt(n);
|
||
StringRef first = getData().getDataAtWithTerminatingZero(offset_of_first_elem);
|
||
|
||
size_t offset_of_last_elem = getOffsets()[n] - 1;
|
||
StringRef last = getData().getDataAtWithTerminatingZero(offset_of_last_elem);
|
||
|
||
return StringRef(first.data, last.data + last.size - first.data);
|
||
}
|
||
|
||
void insertData(const char * pos, size_t length) override
|
||
{
|
||
/** Аналогично - только для массивов значений фиксированной длины.
|
||
*/
|
||
IColumn * data_ = data.get();
|
||
if (!data_->isFixed())
|
||
throw Exception("Method insertData is not supported for " + getName(), ErrorCodes::NOT_IMPLEMENTED);
|
||
|
||
size_t field_size = data_->sizeOfField();
|
||
|
||
const char * end = pos + length;
|
||
size_t elems = 0;
|
||
for (; pos + field_size <= end; pos += field_size, ++elems)
|
||
data_->insertData(pos, field_size);
|
||
|
||
if (pos != end)
|
||
throw Exception("Incorrect length argument for method ColumnArray::insertData", ErrorCodes::BAD_ARGUMENTS);
|
||
|
||
getOffsets().push_back((getOffsets().size() == 0 ? 0 : getOffsets().back()) + elems);
|
||
}
|
||
|
||
StringRef serializeValueIntoArena(size_t n, Arena & arena, char const *& begin) const override
|
||
{
|
||
size_t array_size = sizeAt(n);
|
||
size_t offset = offsetAt(n);
|
||
|
||
char * pos = arena.allocContinue(sizeof(array_size), begin);
|
||
memcpy(pos, &array_size, sizeof(array_size));
|
||
|
||
size_t values_size = 0;
|
||
for (size_t i = 0; i < array_size; ++i)
|
||
values_size += getData().serializeValueIntoArena(offset + i, arena, begin).size;
|
||
|
||
return StringRef(begin, sizeof(array_size) + values_size);
|
||
}
|
||
|
||
const char * deserializeAndInsertFromArena(const char * pos) override
|
||
{
|
||
size_t array_size = *reinterpret_cast<const size_t *>(pos);
|
||
pos += sizeof(array_size);
|
||
|
||
for (size_t i = 0; i < array_size; ++i)
|
||
pos = getData().deserializeAndInsertFromArena(pos);
|
||
|
||
getOffsets().push_back((getOffsets().size() == 0 ? 0 : getOffsets().back()) + array_size);
|
||
return pos;
|
||
}
|
||
|
||
void insertRangeFrom(const IColumn & src, size_t start, size_t length) override;
|
||
|
||
void insert(const Field & x) override
|
||
{
|
||
const Array & array = DB::get<const Array &>(x);
|
||
size_t size = array.size();
|
||
for (size_t i = 0; i < size; ++i)
|
||
getData().insert(array[i]);
|
||
getOffsets().push_back((getOffsets().size() == 0 ? 0 : getOffsets().back()) + size);
|
||
}
|
||
|
||
void insertFrom(const IColumn & src_, size_t n) override
|
||
{
|
||
const ColumnArray & src = static_cast<const ColumnArray &>(src_);
|
||
size_t size = src.sizeAt(n);
|
||
size_t offset = src.offsetAt(n);
|
||
|
||
getData().insertRangeFrom(src.getData(), offset, size);
|
||
getOffsets().push_back((getOffsets().size() == 0 ? 0 : getOffsets().back()) + size);
|
||
}
|
||
|
||
void insertDefault() override
|
||
{
|
||
getOffsets().push_back(getOffsets().size() == 0 ? 0 : getOffsets().back());
|
||
}
|
||
|
||
void popBack(size_t n) override
|
||
{
|
||
auto & offsets = getOffsets();
|
||
size_t nested_n = offsets.back() - offsetAt(offsets.size() - n);
|
||
getData().popBack(nested_n);
|
||
offsets.resize_assume_reserved(offsets.size() - n);
|
||
}
|
||
|
||
ColumnPtr filter(const Filter & filt, ssize_t result_size_hint) const override;
|
||
|
||
ColumnPtr permute(const Permutation & perm, size_t limit) const override;
|
||
|
||
int compareAt(size_t n, size_t m, const IColumn & rhs_, int nan_direction_hint) const override
|
||
{
|
||
const ColumnArray & rhs = static_cast<const ColumnArray &>(rhs_);
|
||
|
||
/// Не оптимально
|
||
size_t lhs_size = sizeAt(n);
|
||
size_t rhs_size = rhs.sizeAt(m);
|
||
size_t min_size = std::min(lhs_size, rhs_size);
|
||
for (size_t i = 0; i < min_size; ++i)
|
||
if (int res = getData().compareAt(offsetAt(n) + i, rhs.offsetAt(m) + i, *rhs.data.get(), nan_direction_hint))
|
||
return res;
|
||
|
||
return lhs_size < rhs_size
|
||
? -1
|
||
: (lhs_size == rhs_size
|
||
? 0
|
||
: 1);
|
||
}
|
||
|
||
template <bool positive>
|
||
struct less
|
||
{
|
||
const ColumnArray & parent;
|
||
|
||
less(const ColumnArray & parent_) : parent(parent_) {}
|
||
|
||
bool operator()(size_t lhs, size_t rhs) const
|
||
{
|
||
if (positive)
|
||
return parent.compareAt(lhs, rhs, parent, 1) < 0;
|
||
else
|
||
return parent.compareAt(lhs, rhs, parent, -1) > 0;
|
||
}
|
||
};
|
||
|
||
void getPermutation(bool reverse, size_t limit, Permutation & res) const override;
|
||
|
||
void reserve(size_t n) override
|
||
{
|
||
getOffsets().reserve(n);
|
||
getData().reserve(n); /// Средний размер массивов тут никак не учитывается. Или считается, что он не больше единицы.
|
||
}
|
||
|
||
size_t byteSize() const override
|
||
{
|
||
return getData().byteSize() + getOffsets().size() * sizeof(getOffsets()[0]);
|
||
}
|
||
|
||
void getExtremes(Field & min, Field & max) const override
|
||
{
|
||
min = Array();
|
||
max = Array();
|
||
}
|
||
|
||
bool hasEqualOffsets(const ColumnArray & other) const
|
||
{
|
||
if (offsets == other.offsets)
|
||
return true;
|
||
|
||
const Offsets_t & offsets1 = getOffsets();
|
||
const Offsets_t & offsets2 = other.getOffsets();
|
||
return offsets1.size() == offsets2.size() && 0 == memcmp(&offsets1[0], &offsets2[0], sizeof(offsets1[0]) * offsets1.size());
|
||
}
|
||
|
||
/** Более эффективные методы манипуляции */
|
||
IColumn & getData() { return *data.get(); }
|
||
const IColumn & getData() const { return *data.get(); }
|
||
|
||
ColumnPtr & getDataPtr() { return data; }
|
||
const ColumnPtr & getDataPtr() const { return data; }
|
||
|
||
Offsets_t & ALWAYS_INLINE getOffsets()
|
||
{
|
||
return static_cast<ColumnOffsets_t &>(*offsets.get()).getData();
|
||
}
|
||
|
||
const Offsets_t & ALWAYS_INLINE getOffsets() const
|
||
{
|
||
return static_cast<const ColumnOffsets_t &>(*offsets.get()).getData();
|
||
}
|
||
|
||
ColumnPtr & getOffsetsColumn() { return offsets; }
|
||
const ColumnPtr & getOffsetsColumn() const { return offsets; }
|
||
|
||
|
||
ColumnPtr replicate(const Offsets_t & replicate_offsets) const override;
|
||
|
||
|
||
ColumnPtr convertToFullColumnIfConst() const override
|
||
{
|
||
ColumnPtr new_data;
|
||
ColumnPtr new_offsets;
|
||
|
||
if (auto full_column = getData().convertToFullColumnIfConst())
|
||
new_data = full_column;
|
||
else
|
||
new_data = data;
|
||
|
||
if (auto full_column = offsets.get()->convertToFullColumnIfConst())
|
||
new_offsets = full_column;
|
||
else
|
||
new_offsets = offsets;
|
||
|
||
return std::make_shared<ColumnArray>(new_data, new_offsets);
|
||
}
|
||
|
||
private:
|
||
ColumnPtr data;
|
||
ColumnPtr offsets; /// Смещения могут быть разделяемыми для нескольких столбцов - для реализации вложенных структур данных.
|
||
|
||
size_t ALWAYS_INLINE offsetAt(size_t i) const { return i == 0 ? 0 : getOffsets()[i - 1]; }
|
||
size_t ALWAYS_INLINE sizeAt(size_t i) const { return i == 0 ? getOffsets()[0] : (getOffsets()[i] - getOffsets()[i - 1]); }
|
||
|
||
|
||
/// Размножить значения, если вложенный столбец - ColumnVector<T>.
|
||
template <typename T>
|
||
ColumnPtr replicateNumber(const Offsets_t & replicate_offsets) const;
|
||
|
||
/// Размножить значения, если вложенный столбец - ColumnString. Код слишком сложный.
|
||
ColumnPtr replicateString(const Offsets_t & replicate_offsets) const;
|
||
|
||
/** Неконстантные массивы константных значений - довольно редкое явление.
|
||
* Большинство функций не умеет с ними работать, и не создаёт такие столбцы в качестве результата.
|
||
* Исключение - функция replicate (см. FunctionsMiscellaneous.h), которая имеет служебное значение для реализации лямбда-функций.
|
||
* Только ради неё сделана реализация метода replicate для ColumnArray(ColumnConst).
|
||
*/
|
||
ColumnPtr replicateConst(const Offsets_t & replicate_offsets) const;
|
||
|
||
|
||
/// Специализации для функции filter.
|
||
template <typename T>
|
||
ColumnPtr filterNumber(const Filter & filt, ssize_t result_size_hint) const;
|
||
|
||
ColumnPtr filterString(const Filter & filt, ssize_t result_size_hint) const;
|
||
ColumnPtr filterGeneric(const Filter & filt, ssize_t result_size_hint) const;
|
||
};
|
||
|
||
|
||
}
|