ClickHouse/src/Storages/IStorageCluster.h

48 lines
1.4 KiB
C++
Raw Normal View History

#pragma once
#include <Storages/IStorage.h>
#include <Interpreters/Cluster.h>
#include <QueryPipeline/RemoteQueryExecutor.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(
String cluster_name_,
const StorageID & table_id_,
Poco::Logger * log_,
size_t max_function_arguments_,
bool structure_argument_was_provided_);
Pipe read(const Names &, const StorageSnapshotPtr &, SelectQueryInfo &,
ContextPtr, QueryProcessingStage::Enum, 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(ASTPtr query, 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 &) {}
private:
Poco::Logger * log;
String cluster_name;
size_t max_function_arguments;
bool structure_argument_was_provided;
};
}