ClickHouse/dbms/include/DB/Interpreters/ClusterProxy/SelectQueryConstructor.h

#pragma once

#include <DB/Interpreters/ClusterProxy/IQueryConstructor.h>
#include <DB/Core/QueryProcessingStage.h>
#include <DB/Storages/IStorage.h>

namespace DB
{

namespace ClusterProxy
{

class SelectQueryConstructor final : public IQueryConstructor
{
public:
	SelectQueryConstructor(const QueryProcessingStage::Enum & processed_stage, const Tables & external_tables);

	BlockInputStreamPtr createLocal(ASTPtr query_ast, const Context & context, const Cluster::Address & address) override;
	BlockInputStreamPtr createRemote(IConnectionPool * pool, const std::string & query,
		const Settings & settings, ThrottlerPtr throttler, const Context & context) override;
	BlockInputStreamPtr createRemote(ConnectionPoolsPtr & pools, const std::string & query,
		const Settings & settings, ThrottlerPtr throttler, const Context & context) override;
	PoolMode getPoolMode() const override;

private:
	const QueryProcessingStage::Enum & processed_stage;
	const Tables & external_tables;
};

}

}