ClickHouse/dbms/include/DB/Interpreters/InterpreterSelectQuery.h

124 lines
5.5 KiB
C
Raw Normal View History

2011-08-28 05:13:24 +00:00
#pragma once
2012-05-09 13:12:38 +00:00
#include <DB/Core/QueryProcessingStage.h>
2011-08-28 05:13:24 +00:00
#include <DB/Interpreters/Context.h>
#include <DB/Interpreters/ExpressionAnalyzer.h>
2011-08-28 05:13:24 +00:00
#include <DB/DataStreams/IBlockInputStream.h>
2012-05-09 13:12:38 +00:00
#include <DB/Parsers/ASTSelectQuery.h>
#include <DB/TableFunctions/ITableFunction.h>
2011-08-28 05:13:24 +00:00
namespace DB
{
2012-05-09 13:12:38 +00:00
/** Интерпретирует запрос SELECT. Возвращает поток блоков с результатами выполнения запроса до стадии to_stage.
2011-08-28 05:13:24 +00:00
*/
class InterpreterSelectQuery
{
public:
/** to_stage
* - стадия, до которой нужно выполнить запрос. По-умолчанию - до конца.
* Можно выполнить до промежуточного состояния агрегации, которые объединяются с разных серверов при распределённой обработке запроса.
*
* subquery_depth
* - для контроля ограничений на глубину вложенности подзапросов. Для подзапросов передаётся значение, увеличенное на единицу.
*
* input
* - если задан - читать не из таблицы, указанной в запросе, а из готового источника.
*
* required_column_names
* - удалить из запроса все столбцы кроме указанных - используется для удаления ненужных столбцов из подзапросов.
*
* table_column_names
* - поместить в контекст в качестве известных столбцов только указанные столбцы, а не все столбцы таблицы.
* Используется, например, совместно с указанием input.
*/
2014-06-12 23:21:38 +00:00
InterpreterSelectQuery(
ASTPtr query_ptr_,
const Context & context_,
QueryProcessingStage::Enum to_stage_ = QueryProcessingStage::Complete,
size_t subquery_depth_ = 0,
BlockInputStreamPtr input = nullptr);
2014-06-12 23:21:38 +00:00
InterpreterSelectQuery(
ASTPtr query_ptr_,
const Context & context_,
const Names & required_column_names,
2014-06-12 23:21:38 +00:00
QueryProcessingStage::Enum to_stage_ = QueryProcessingStage::Complete,
size_t subquery_depth_ = 0,
BlockInputStreamPtr input = nullptr);
2011-08-28 05:13:24 +00:00
2014-06-12 23:21:38 +00:00
InterpreterSelectQuery(
ASTPtr query_ptr_,
const Context & context_,
const Names & required_column_names,
2014-06-12 23:21:38 +00:00
const NamesAndTypesList & table_column_names,
QueryProcessingStage::Enum to_stage_ = QueryProcessingStage::Complete,
size_t subquery_depth_ = 0,
BlockInputStreamPtr input = nullptr);
2011-10-30 05:19:41 +00:00
/// Выполнить запрос, получить поток блоков для чтения
2011-08-28 05:13:24 +00:00
BlockInputStreamPtr execute();
2011-10-30 05:19:41 +00:00
/** Выполнить запрос, записать результат в нужном формате в buf.
* BlockInputStreamPtr возвращается, чтобы можно было потом получить информацию о плане выполнения запроса.
*/
BlockInputStreamPtr executeAndFormat(WriteBuffer & buf);
2011-08-28 05:13:24 +00:00
DataTypes getReturnTypes();
2011-10-30 05:19:41 +00:00
Block getSampleBlock();
2011-08-28 05:13:24 +00:00
private:
typedef Poco::SharedPtr<ExpressionAnalyzer> ExpressionAnalyzerPtr;
2014-06-12 23:21:38 +00:00
void init(BlockInputStreamPtr input, const NamesAndTypesList & table_column_names = NamesAndTypesList());
/** Из какой таблицы читать. JOIN-ы не поддерживаются.
*/
2012-08-20 19:21:04 +00:00
void getDatabaseAndTableNames(String & database_name, String & table_name);
2014-06-12 23:21:38 +00:00
2012-05-09 13:12:38 +00:00
/** Выбрать из списка столбцов какой-нибудь, лучше - минимального размера.
*/
String getAnyColumn();
/// Разные стадии выполнения запроса.
2012-05-22 18:32:45 +00:00
/// Вынимает данные из таблицы. Возвращает стадию, до которой запрос был обработан в Storage.
QueryProcessingStage::Enum executeFetchColumns(BlockInputStreams & streams);
void executeWhere( BlockInputStreams & streams, ExpressionActionsPtr expression);
2014-02-27 12:49:21 +00:00
void executeAggregation( BlockInputStreams & streams, ExpressionActionsPtr expression,
bool overflow_row, bool final);
void executeMergeAggregated( BlockInputStreams & streams, bool overflow_row, bool final);
void executeTotalsAndHaving( BlockInputStreams & streams, bool has_having, ExpressionActionsPtr expression,
bool overflow_row);
void executeHaving( BlockInputStreams & streams, ExpressionActionsPtr expression);
void executeExpression( BlockInputStreams & streams, ExpressionActionsPtr expression);
void executeOrder( BlockInputStreams & streams);
void executePreLimit( BlockInputStreams & streams);
void executeUnion( BlockInputStreams & streams);
void executeLimit( BlockInputStreams & streams);
void executeProjection( BlockInputStreams & streams, ExpressionActionsPtr expression);
void executeDistinct( BlockInputStreams & streams, bool before_order, Names columns);
void executeSubqueriesInSetsAndJoins(BlockInputStreams & streams, SubqueriesForSets & subqueries_for_sets);
2012-05-09 13:12:38 +00:00
2011-08-28 05:13:24 +00:00
ASTPtr query_ptr;
2012-05-09 13:12:38 +00:00
ASTSelectQuery & query;
2011-08-28 05:13:24 +00:00
Context context;
2012-08-02 17:33:31 +00:00
Settings settings;
2012-05-09 13:12:38 +00:00
QueryProcessingStage::Enum to_stage;
size_t subquery_depth;
ExpressionAnalyzerPtr query_analyzer;
BlockInputStreams streams;
/// Таблица, откуда читать данные, если не подзапрос.
StoragePtr storage;
IStorage::TableStructureReadLockPtr table_lock;
2012-06-25 03:56:45 +00:00
Logger * log;
2011-08-28 05:13:24 +00:00
};
}