2019-01-26 13:12:13 +00:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <Storages/MergeTree/MergeTreeIndices.h>
|
|
|
|
#include <Storages/MergeTree/MergeTreeData.h>
|
|
|
|
|
|
|
|
#include <memory>
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
class MergeTreeUniqueIndex;
|
|
|
|
|
|
|
|
struct MergeTreeUniqueGranule : public MergeTreeIndexGranule
|
|
|
|
{
|
|
|
|
explicit MergeTreeUniqueGranule(const MergeTreeUniqueIndex & index);
|
|
|
|
|
|
|
|
void serializeBinary(WriteBuffer & ostr) const override;
|
|
|
|
void deserializeBinary(ReadBuffer & istr) override;
|
|
|
|
|
|
|
|
String toString() const override;
|
|
|
|
bool empty() const override { return !block.rows(); }
|
|
|
|
|
|
|
|
void update(const Block & block, size_t * pos, size_t limit) override;
|
|
|
|
|
|
|
|
~MergeTreeUniqueGranule() override = default;
|
|
|
|
|
|
|
|
const MergeTreeUniqueIndex & index;
|
|
|
|
Block block;
|
|
|
|
};
|
|
|
|
|
|
|
|
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:
|
|
|
|
const MergeTreeUniqueIndex & index;
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
class MergeTreeUniqueIndex : public MergeTreeIndex
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
MergeTreeUniqueIndex(
|
|
|
|
String name,
|
|
|
|
ExpressionActionsPtr expr,
|
|
|
|
const Names & columns,
|
|
|
|
const DataTypes & data_types,
|
2019-01-26 15:45:38 +00:00
|
|
|
size_t granularity,
|
|
|
|
size_t _max_rows)
|
|
|
|
: MergeTreeIndex(name, expr, columns, data_types, 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(
|
|
|
|
const MergeTreeData & data, std::shared_ptr<ASTIndexDeclaration> node, const Context & context);
|
|
|
|
|
|
|
|
}
|