ClickHouse/src/Interpreters/ClusterProxy/SelectStreamFactory.h
2021-07-31 15:45:39 +08:00

42 lines
911 B
C++

#pragma once
#include <Core/QueryProcessingStage.h>
#include <Interpreters/ClusterProxy/IStreamFactory.h>
#include <Interpreters/StorageID.h>
#include <Storages/IStorage_fwd.h>
namespace DB
{
namespace ClusterProxy
{
class SelectStreamFactory final : public IStreamFactory
{
public:
SelectStreamFactory(
const Block & header_,
QueryProcessingStage::Enum processed_stage_,
bool has_virtual_shard_num_column_);
void createForShard(
const Cluster::ShardInfo & shard_info,
const ASTPtr & query_ast,
const StorageID & main_table,
const ASTPtr & table_func_ptr,
ContextPtr context,
std::vector<QueryPlanPtr> & local_plans,
Shards & remote_shards,
UInt32 shard_count) override;
private:
const Block header;
QueryProcessingStage::Enum processed_stage;
bool has_virtual_shard_num_column = false;
};
}
}