mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 10:22:10 +00:00
56 lines
1.6 KiB
C++
56 lines
1.6 KiB
C++
#pragma once
|
|
|
|
#include <Storages/IStorage.h>
|
|
#include <Interpreters/Cluster.h>
|
|
#include <QueryPipeline/RemoteQueryExecutor.h>
|
|
#include <Parsers/ASTExpressionList.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
|
|
/**
|
|
* Base cluster for Storages used in table functions like s3Cluster and hdfsCluster
|
|
* Needed for code simplification around parallel_distributed_insert_select
|
|
*/
|
|
class IStorageCluster : public IStorage
|
|
{
|
|
public:
|
|
IStorageCluster(
|
|
const String & cluster_name_,
|
|
const StorageID & table_id_,
|
|
Poco::Logger * log_,
|
|
bool structure_argument_was_provided_);
|
|
|
|
void read(
|
|
QueryPlan & query_plan,
|
|
const Names & column_names,
|
|
const StorageSnapshotPtr & storage_snapshot,
|
|
SelectQueryInfo & query_info,
|
|
ContextPtr context,
|
|
QueryProcessingStage::Enum processed_stage,
|
|
size_t /*max_block_size*/,
|
|
size_t /*num_streams*/) override;
|
|
|
|
ClusterPtr getCluster(ContextPtr context) const;
|
|
/// Query is needed for pruning by virtual columns (_file, _path)
|
|
virtual RemoteQueryExecutor::Extension getTaskIteratorExtension(const ActionsDAG::Node * predicate, const ContextPtr & context) const = 0;
|
|
|
|
QueryProcessingStage::Enum getQueryProcessingStage(ContextPtr, QueryProcessingStage::Enum, const StorageSnapshotPtr &, SelectQueryInfo &) const override;
|
|
|
|
bool isRemote() const override { return true; }
|
|
|
|
protected:
|
|
virtual void updateBeforeRead(const ContextPtr &) {}
|
|
|
|
virtual void addColumnsStructureToQuery(ASTPtr & query, const String & structure, const ContextPtr & context) = 0;
|
|
|
|
private:
|
|
Poco::Logger * log;
|
|
String cluster_name;
|
|
bool structure_argument_was_provided;
|
|
};
|
|
|
|
|
|
}
|