#pragma once #include #include #include #include #include #include #include #include #include #include namespace DB { /// A ColumnVector for Decimals template class ColumnDecimal final : public COWHelper> { private: using Self = ColumnDecimal; friend class COWHelper; public: using ValueType = T; using NativeT = typename T::NativeType; using Container = PaddedPODArray; private: ColumnDecimal(const size_t n, UInt32 scale_) : data(n), scale(scale_) {} ColumnDecimal(const ColumnDecimal & src) : data(src.data.begin(), src.data.end()), scale(src.scale) {} public: const char * getFamilyName() const override { return TypeName.data(); } TypeIndex getDataType() const override { return TypeToTypeIndex; } bool isNumeric() const override { return false; } bool canBeInsideNullable() const override { return true; } bool isFixedAndContiguous() const final { return true; } size_t sizeOfValueIfFixed() const override { return sizeof(T); } size_t size() const override { return data.size(); } size_t byteSize() const override { return data.size() * sizeof(data[0]); } size_t byteSizeAt(size_t) const override { return sizeof(data[0]); } size_t allocatedBytes() const override { return data.allocated_bytes(); } void protect() override { data.protect(); } void reserve(size_t n) override { data.reserve(n); } void insertFrom(const IColumn & src, size_t n) override { data.push_back(static_cast(src).getData()[n]); } void insertData(const char * src, size_t /*length*/) override; void insertDefault() override { data.push_back(T()); } void insertManyDefaults(size_t length) override { data.resize_fill(data.size() + length); } void insert(const Field & x) override { data.push_back(x.get()); } void insertRangeFrom(const IColumn & src, size_t start, size_t length) override; void popBack(size_t n) override { data.resize_assume_reserved(data.size() - n); } std::string_view getRawData() const override { return {reinterpret_cast(data.data()), byteSize()}; } StringRef getDataAt(size_t n) const override { return StringRef(reinterpret_cast(&data[n]), sizeof(data[n])); } Float64 getFloat64(size_t n) const final { return DecimalUtils::convertTo(data[n], scale); } StringRef serializeValueIntoArena(size_t n, Arena & arena, char const *& begin) const override; const char * deserializeAndInsertFromArena(const char * pos) override; const char * skipSerializedInArena(const char * pos) const override; void updateHashWithValue(size_t n, SipHash & hash) const override; void updateWeakHash32(WeakHash32 & hash) const override; void updateHashFast(SipHash & hash) const override; int compareAt(size_t n, size_t m, const IColumn & rhs_, int nan_direction_hint) const override; void compareColumn(const IColumn & rhs, size_t rhs_row_num, PaddedPODArray * row_indexes, PaddedPODArray & compare_results, int direction, int nan_direction_hint) const override; bool hasEqualValues() const override; void getPermutation(IColumn::PermutationSortDirection direction, IColumn::PermutationSortStability stability, size_t limit, int nan_direction_hint, IColumn::Permutation & res) const override; void updatePermutation(IColumn::PermutationSortDirection direction, IColumn::PermutationSortStability stability, size_t limit, int, IColumn::Permutation & res, EqualRanges& equal_ranges) const override; MutableColumnPtr cloneResized(size_t size) const override; Field operator[](size_t n) const override { return DecimalField(data[n], scale); } void get(size_t n, Field & res) const override { res = (*this)[n]; } bool getBool(size_t n) const override { return bool(data[n].value); } Int64 getInt(size_t n) const override { return Int64(data[n].value); } UInt64 get64(size_t n) const override; bool isDefaultAt(size_t n) const override { return data[n].value == 0; } ColumnPtr filter(const IColumn::Filter & filt, ssize_t result_size_hint) const override; void expand(const IColumn::Filter & mask, bool inverted) override; ColumnPtr permute(const IColumn::Permutation & perm, size_t limit) const override; ColumnPtr index(const IColumn & indexes, size_t limit) const override; template ColumnPtr indexImpl(const PaddedPODArray & indexes, size_t limit) const; ColumnPtr replicate(const IColumn::Offsets & offsets) const override; void getExtremes(Field & min, Field & max) const override; MutableColumns scatter(IColumn::ColumnIndex num_columns, const IColumn::Selector & selector) const override { return this->template scatterImpl(num_columns, selector); } void gather(ColumnGathererStream & gatherer_stream) override; bool structureEquals(const IColumn & rhs) const override { if (auto rhs_concrete = typeid_cast *>(&rhs)) return scale == rhs_concrete->scale; return false; } double getRatioOfDefaultRows(double sample_ratio) const override { return this->template getRatioOfDefaultRowsImpl(sample_ratio); } UInt64 getNumberOfDefaultRows() const override { return this->template getNumberOfDefaultRowsImpl(); } void getIndicesOfNonDefaultRows(IColumn::Offsets & indices, size_t from, size_t limit) const override { return this->template getIndicesOfNonDefaultRowsImpl(indices, from, limit); } ColumnPtr compress() const override; void insertValue(const T value) { data.push_back(value); } Container & getData() { return data; } const Container & getData() const { return data; } const T & getElement(size_t n) const { return data[n]; } T & getElement(size_t n) { return data[n]; } UInt32 getScale() const { return scale; } protected: Container data; UInt32 scale; }; template class ColumnVector; template struct ColumnVectorOrDecimalT { using Col = ColumnVector; }; template struct ColumnVectorOrDecimalT { using Col = ColumnDecimal; }; template using ColumnVectorOrDecimal = typename ColumnVectorOrDecimalT::Col; template template ColumnPtr ColumnDecimal::indexImpl(const PaddedPODArray & indexes, size_t limit) const { assert(limit <= indexes.size()); auto res = this->create(limit, scale); typename Self::Container & res_data = res->getData(); for (size_t i = 0; i < limit; ++i) res_data[i] = data[indexes[i]]; return res; } /// Prevent implicit template instantiation of ColumnDecimal for common decimal types extern template class ColumnDecimal; extern template class ColumnDecimal; extern template class ColumnDecimal; extern template class ColumnDecimal; extern template class ColumnDecimal; }