2019-01-26 13:12:13 +00:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <Storages/MergeTree/MergeTreeIndices.h>
|
|
|
|
#include <Storages/MergeTree/MergeTreeData.h>
|
|
|
|
|
2019-01-27 18:02:15 +00:00
|
|
|
#include <Interpreters/Set.h>
|
|
|
|
|
2019-01-26 13:12:13 +00:00
|
|
|
#include <memory>
|
2019-01-29 17:26:45 +00:00
|
|
|
#include <set>
|
2019-01-26 13:12:13 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2019-02-06 20:05:50 +00:00
|
|
|
class MergeTreeSetSkippingIndex;
|
2019-01-26 13:12:13 +00:00
|
|
|
|
2019-02-06 20:05:50 +00:00
|
|
|
struct MergeTreeSetIndexGranule : public IMergeTreeIndexGranule
|
2019-01-26 13:12:13 +00:00
|
|
|
{
|
2019-02-06 20:05:50 +00:00
|
|
|
explicit MergeTreeSetIndexGranule(const MergeTreeSetSkippingIndex & index);
|
2019-01-26 13:12:13 +00:00
|
|
|
|
|
|
|
void serializeBinary(WriteBuffer & ostr) const override;
|
|
|
|
void deserializeBinary(ReadBuffer & istr) override;
|
|
|
|
|
2019-01-27 18:02:15 +00:00
|
|
|
size_t size() const { return set->getTotalRowCount(); }
|
|
|
|
bool empty() const override { return !size(); }
|
2019-01-26 13:12:13 +00:00
|
|
|
|
|
|
|
void update(const Block & block, size_t * pos, size_t limit) override;
|
2019-01-27 18:23:08 +00:00
|
|
|
Block getElementsBlock() const;
|
2019-01-26 13:12:13 +00:00
|
|
|
|
2019-02-06 20:05:50 +00:00
|
|
|
~MergeTreeSetIndexGranule() override = default;
|
2019-01-26 13:12:13 +00:00
|
|
|
|
2019-02-06 20:05:50 +00:00
|
|
|
const MergeTreeSetSkippingIndex & index;
|
2019-01-27 18:02:15 +00:00
|
|
|
std::unique_ptr<Set> set;
|
2019-01-26 13:12:13 +00:00
|
|
|
};
|
|
|
|
|
2019-01-27 18:23:08 +00:00
|
|
|
|
2019-02-06 20:05:50 +00:00
|
|
|
class SetIndexCondition : public IIndexCondition
|
2019-01-26 13:12:13 +00:00
|
|
|
{
|
|
|
|
public:
|
2019-02-06 20:05:50 +00:00
|
|
|
SetIndexCondition(
|
2019-01-26 13:12:13 +00:00
|
|
|
const SelectQueryInfo & query,
|
|
|
|
const Context & context,
|
2019-02-06 20:05:50 +00:00
|
|
|
const MergeTreeSetSkippingIndex & index);
|
2019-01-26 13:12:13 +00:00
|
|
|
|
|
|
|
bool alwaysUnknownOrTrue() const override;
|
|
|
|
|
|
|
|
bool mayBeTrueOnGranule(MergeTreeIndexGranulePtr idx_granule) const override;
|
|
|
|
|
2019-02-06 20:05:50 +00:00
|
|
|
~SetIndexCondition() override = default;
|
2019-01-26 13:12:13 +00:00
|
|
|
private:
|
2019-01-29 13:13:18 +00:00
|
|
|
void traverseAST(ASTPtr & node) const;
|
|
|
|
bool atomFromAST(ASTPtr & node) const;
|
2019-01-28 08:15:13 +00:00
|
|
|
bool operatorFromAST(ASTPtr & node) const;
|
2019-01-27 20:24:33 +00:00
|
|
|
|
2019-02-05 13:52:45 +00:00
|
|
|
bool checkASTUseless(const ASTPtr &node, bool atomic = false) const;
|
2019-01-29 16:53:44 +00:00
|
|
|
|
2019-02-06 20:05:50 +00:00
|
|
|
const MergeTreeSetSkippingIndex & index;
|
2019-01-27 20:24:33 +00:00
|
|
|
|
2019-01-29 16:53:44 +00:00
|
|
|
bool useless;
|
2019-01-29 17:26:45 +00:00
|
|
|
std::set<String> key_columns;
|
2019-01-28 08:15:13 +00:00
|
|
|
ASTPtr expression_ast;
|
2019-01-27 20:24:33 +00:00
|
|
|
ExpressionActionsPtr actions;
|
2019-01-26 13:12:13 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
|
2019-02-06 20:05:50 +00:00
|
|
|
class MergeTreeSetSkippingIndex : public IMergeTreeIndex
|
2019-01-26 13:12:13 +00:00
|
|
|
{
|
|
|
|
public:
|
2019-02-06 20:05:50 +00:00
|
|
|
MergeTreeSetSkippingIndex(
|
2019-02-05 16:43:52 +00:00
|
|
|
String name_,
|
|
|
|
ExpressionActionsPtr expr_,
|
|
|
|
const Names & columns_,
|
|
|
|
const DataTypes & data_types_,
|
|
|
|
const Block & header_,
|
|
|
|
size_t granularity_,
|
|
|
|
size_t max_rows_)
|
2019-02-06 07:49:18 +00:00
|
|
|
: IMergeTreeIndex(std::move(name_), std::move(expr_), columns_, data_types_, header_, granularity_), max_rows(max_rows_) {}
|
2019-01-26 13:12:13 +00:00
|
|
|
|
2019-02-06 20:05:50 +00:00
|
|
|
~MergeTreeSetSkippingIndex() override = default;
|
2019-01-26 13:12:13 +00:00
|
|
|
|
|
|
|
MergeTreeIndexGranulePtr createIndexGranule() const override;
|
|
|
|
|
|
|
|
IndexConditionPtr createIndexCondition(
|
|
|
|
const SelectQueryInfo & query, const Context & context) const override;
|
|
|
|
|
2019-01-26 15:45:38 +00:00
|
|
|
size_t max_rows = 0;
|
2019-01-26 13:12:13 +00:00
|
|
|
};
|
|
|
|
|
2019-01-29 17:28:13 +00:00
|
|
|
}
|