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
|
|
|
|
{
|
|
|
|
|
|
|
|
class MergeTreeUniqueIndex;
|
|
|
|
|
|
|
|
struct MergeTreeUniqueGranule : public MergeTreeIndexGranule
|
|
|
|
{
|
|
|
|
explicit MergeTreeUniqueGranule(const MergeTreeUniqueIndex & index);
|
|
|
|
|
|
|
|
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
|
|
|
|
|
|
|
~MergeTreeUniqueGranule() override = default;
|
|
|
|
|
|
|
|
const MergeTreeUniqueIndex & 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-01-26 13:12:13 +00:00
|
|
|
class UniqueCondition : public IndexCondition
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
UniqueCondition(
|
|
|
|
const SelectQueryInfo & query,
|
|
|
|
const Context & context,
|
|
|
|
const MergeTreeUniqueIndex & index);
|
|
|
|
|
|
|
|
bool alwaysUnknownOrTrue() const override;
|
|
|
|
|
|
|
|
bool mayBeTrueOnGranule(MergeTreeIndexGranulePtr idx_granule) const override;
|
|
|
|
|
|
|
|
~UniqueCondition() override = default;
|
|
|
|
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-01-26 13:12:13 +00:00
|
|
|
const MergeTreeUniqueIndex & 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
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
class MergeTreeUniqueIndex : public MergeTreeIndex
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
MergeTreeUniqueIndex(
|
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_)
|
|
|
|
: MergeTreeIndex(std::move(name_), std::move(expr_), columns_, data_types_, header_, granularity_), max_rows(max_rows_) {}
|
2019-01-26 13:12:13 +00:00
|
|
|
|
|
|
|
~MergeTreeUniqueIndex() override = default;
|
|
|
|
|
|
|
|
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
|
|
|
};
|
|
|
|
|
|
|
|
std::unique_ptr<MergeTreeIndex> MergeTreeUniqueIndexCreator(
|
2019-01-30 10:58:10 +00:00
|
|
|
const NamesAndTypesList & columns, std::shared_ptr<ASTIndexDeclaration> node, const Context & context);
|
2019-01-26 13:12:13 +00:00
|
|
|
|
2019-01-29 17:28:13 +00:00
|
|
|
}
|