diff options
author | vitalyisaev <vitalyisaev@ydb.tech> | 2023-11-14 09:58:56 +0300 |
---|---|---|
committer | vitalyisaev <vitalyisaev@ydb.tech> | 2023-11-14 10:20:20 +0300 |
commit | c2b2dfd9827a400a8495e172a56343462e3ceb82 (patch) | |
tree | cd4e4f597d01bede4c82dffeb2d780d0a9046bd0 /contrib/clickhouse/src/Storages/MergeTree/MergeTreeIndexMinMax.h | |
parent | d4ae8f119e67808cb0cf776ba6e0cf95296f2df7 (diff) | |
download | ydb-c2b2dfd9827a400a8495e172a56343462e3ceb82.tar.gz |
YQ Connector: move tests from yql to ydb (OSS)
Перенос папки с тестами на Коннектор из папки yql в папку ydb (синхронизируется с github).
Diffstat (limited to 'contrib/clickhouse/src/Storages/MergeTree/MergeTreeIndexMinMax.h')
-rw-r--r-- | contrib/clickhouse/src/Storages/MergeTree/MergeTreeIndexMinMax.h | 89 |
1 files changed, 89 insertions, 0 deletions
diff --git a/contrib/clickhouse/src/Storages/MergeTree/MergeTreeIndexMinMax.h b/contrib/clickhouse/src/Storages/MergeTree/MergeTreeIndexMinMax.h new file mode 100644 index 0000000000..ec9eca77c1 --- /dev/null +++ b/contrib/clickhouse/src/Storages/MergeTree/MergeTreeIndexMinMax.h @@ -0,0 +1,89 @@ +#pragma once + +#include <Storages/MergeTree/MergeTreeIndices.h> +#include <Storages/MergeTree/MergeTreeData.h> +#include <Storages/MergeTree/KeyCondition.h> + +#include <memory> + + +namespace DB +{ + +struct MergeTreeIndexGranuleMinMax final : public IMergeTreeIndexGranule +{ + MergeTreeIndexGranuleMinMax(const String & index_name_, const Block & index_sample_block_); + MergeTreeIndexGranuleMinMax( + const String & index_name_, + const Block & index_sample_block_, + std::vector<Range> && hyperrectangle_); + + ~MergeTreeIndexGranuleMinMax() override = default; + + void serializeBinary(WriteBuffer & ostr) const override; + void deserializeBinary(ReadBuffer & istr, MergeTreeIndexVersion version) override; + + bool empty() const override { return hyperrectangle.empty(); } + + String index_name; + Block index_sample_block; + std::vector<Range> hyperrectangle; +}; + + +struct MergeTreeIndexAggregatorMinMax final : IMergeTreeIndexAggregator +{ + MergeTreeIndexAggregatorMinMax(const String & index_name_, const Block & index_sample_block); + ~MergeTreeIndexAggregatorMinMax() override = default; + + bool empty() const override { return hyperrectangle.empty(); } + MergeTreeIndexGranulePtr getGranuleAndReset() override; + void update(const Block & block, size_t * pos, size_t limit) override; + + String index_name; + Block index_sample_block; + std::vector<Range> hyperrectangle; +}; + + +class MergeTreeIndexConditionMinMax final : public IMergeTreeIndexCondition +{ +public: + MergeTreeIndexConditionMinMax( + const IndexDescription & index, + const SelectQueryInfo & query_info, + ContextPtr context); + + bool alwaysUnknownOrTrue() const override; + + bool mayBeTrueOnGranule(MergeTreeIndexGranulePtr idx_granule) const override; + + ~MergeTreeIndexConditionMinMax() override = default; +private: + DataTypes index_data_types; + KeyCondition condition; +}; + + +class MergeTreeIndexMinMax : public IMergeTreeIndex +{ +public: + explicit MergeTreeIndexMinMax(const IndexDescription & index_) + : IMergeTreeIndex(index_) + {} + + ~MergeTreeIndexMinMax() override = default; + + MergeTreeIndexGranulePtr createIndexGranule() const override; + MergeTreeIndexAggregatorPtr createIndexAggregator() const override; + + MergeTreeIndexConditionPtr createIndexCondition( + const SelectQueryInfo & query, ContextPtr context) const override; + + bool mayBenefitFromIndexForIn(const ASTPtr & node) const override; + + const char* getSerializedFileExtension() const override { return ".idx2"; } + MergeTreeIndexFormat getDeserializedFormat(const IDataPartStorage & data_part_storage, const std::string & path_prefix) const override; /// NOLINT +}; + +} |