ClickHouse/src/Storages/MergeTree/StorageFromMergeTreeDataPart.h

67 lines
2.0 KiB
C++
Raw Normal View History

#pragma once
#include <Storages/IStorage.h>
2019-10-10 16:30:30 +00:00
#include <Storages/MergeTree/IMergeTreeDataPart.h>
#include <Storages/MergeTree/MergeTreeDataSelectExecutor.h>
#include <Core/Defines.h>
#include <ext/shared_ptr_helper.h>
namespace DB
{
/// A Storage that allows reading from a single MergeTree data part.
class StorageFromMergeTreeDataPart final : public ext::shared_ptr_helper<StorageFromMergeTreeDataPart>, public IStorage
{
2019-08-26 19:07:29 +00:00
friend struct ext::shared_ptr_helper<StorageFromMergeTreeDataPart>;
public:
String getName() const override { return "FromMergeTreeDataPart"; }
2020-08-03 11:33:11 +00:00
Pipe read(
const Names & column_names,
const StorageMetadataPtr & metadata_snapshot,
SelectQueryInfo & query_info,
const Context & context,
2018-08-05 07:05:36 +00:00
QueryProcessingStage::Enum /*processed_stage*/,
size_t max_block_size,
unsigned num_streams) override
{
return MergeTreeDataSelectExecutor(part->storage)
.readFromParts({part}, column_names, metadata_snapshot, query_info, context, max_block_size, num_streams);
}
2019-10-01 16:50:08 +00:00
bool supportsIndexForIn() const override { return true; }
bool mayBenefitFromIndexForIn(
const ASTPtr & left_in_operand, const Context & query_context, const StorageMetadataPtr & metadata_snapshot) const override
{
return part->storage.mayBenefitFromIndexForIn(left_in_operand, query_context, metadata_snapshot);
}
NamesAndTypesList getVirtuals() const override
2020-04-27 13:55:30 +00:00
{
return part->storage.getVirtuals();
}
protected:
StorageFromMergeTreeDataPart(const MergeTreeData::DataPartPtr & part_)
2020-04-27 13:55:30 +00:00
: IStorage(getIDFromPart(part_))
2019-12-04 16:06:55 +00:00
, part(part_)
2019-08-24 21:20:20 +00:00
{
setInMemoryMetadata(part_->storage.getInMemoryMetadata());
2019-08-24 21:20:20 +00:00
}
private:
MergeTreeData::DataPartPtr part;
static StorageID getIDFromPart(const MergeTreeData::DataPartPtr & part_)
{
auto table_id = part_->storage.getStorageID();
return StorageID(table_id.database_name, table_id.table_name + " (part " + part_->name + ")");
}
};
}