mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-29 13:10:48 +00:00
157 lines
4.3 KiB
C++
157 lines
4.3 KiB
C++
#include <DB/Parsers/formatAST.h>
|
||
|
||
#include <DB/DataStreams/BlockIO.h>
|
||
#include <DB/Interpreters/executeQuery.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
|
||
static void checkLimits(const IAST & ast, const Limits & limits)
|
||
{
|
||
if (limits.max_ast_depth)
|
||
ast.checkDepth(limits.max_ast_depth);
|
||
if (limits.max_ast_elements)
|
||
ast.checkSize(limits.max_ast_elements);
|
||
}
|
||
|
||
|
||
void executeQuery(
|
||
ReadBuffer & istr,
|
||
WriteBuffer & ostr,
|
||
Context & context,
|
||
BlockInputStreamPtr & query_plan,
|
||
bool internal,
|
||
QueryProcessingStage::Enum stage)
|
||
{
|
||
ParserQuery parser;
|
||
ASTPtr ast;
|
||
std::string expected;
|
||
|
||
std::vector<char> parse_buf;
|
||
const char * begin;
|
||
const char * end;
|
||
|
||
/// Если в istr ещё ничего нет, то считываем кусок данных
|
||
if (istr.buffer().size() == 0)
|
||
istr.next();
|
||
|
||
size_t max_query_size = context.getSettings().max_query_size;
|
||
|
||
if (istr.buffer().end() - istr.position() >= static_cast<ssize_t>(max_query_size))
|
||
{
|
||
/// Если оставшийся размер буфера istr достаточен, чтобы распарсить запрос до max_query_size, то парсим прямо в нём
|
||
begin = istr.position();
|
||
end = istr.buffer().end();
|
||
istr.position() += end - begin;
|
||
}
|
||
else
|
||
{
|
||
/// Если нет - считываем достаточное количество данных в parse_buf
|
||
parse_buf.resize(max_query_size);
|
||
parse_buf.resize(istr.read(&parse_buf[0], max_query_size));
|
||
begin = &parse_buf[0];
|
||
end = begin + parse_buf.size();
|
||
}
|
||
|
||
const char * pos = begin;
|
||
|
||
bool parse_res = parser.parse(pos, end, ast, expected);
|
||
|
||
/// Распарсенный запрос должен заканчиваться на конец входных данных или на точку с запятой.
|
||
if (!parse_res || (pos != end && *pos != ';'))
|
||
throw Exception("Syntax error: failed at position "
|
||
+ toString(pos - begin) + ": "
|
||
+ std::string(pos, std::min(SHOW_CHARS_ON_SYNTAX_ERROR, end - pos))
|
||
+ ", expected " + (parse_res ? "end of query" : expected) + ".",
|
||
ErrorCodes::SYNTAX_ERROR);
|
||
|
||
String query(begin, pos - begin);
|
||
|
||
LOG_DEBUG(&Logger::get("executeQuery"), query);
|
||
|
||
/// Положим запрос в список процессов. Но запрос SHOW PROCESSLIST класть не будем.
|
||
ProcessList::EntryPtr process_list_entry;
|
||
if (!internal && NULL == dynamic_cast<const ASTShowProcesslistQuery *>(&*ast))
|
||
process_list_entry = context.getProcessList().insert(query);
|
||
|
||
/// Проверка ограничений.
|
||
checkLimits(*ast, context.getSettingsRef().limits);
|
||
|
||
QuotaForIntervals & quota = context.getQuota();
|
||
time_t current_time = time(0);
|
||
|
||
quota.checkExceeded(current_time);
|
||
|
||
try
|
||
{
|
||
InterpreterQuery interpreter(ast, context, stage);
|
||
interpreter.execute(ostr, &istr, query_plan);
|
||
}
|
||
catch (...)
|
||
{
|
||
quota.addError(current_time);
|
||
throw;
|
||
}
|
||
|
||
quota.addQuery(current_time);
|
||
}
|
||
|
||
|
||
BlockIO executeQuery(
|
||
const String & query,
|
||
Context & context,
|
||
bool internal,
|
||
QueryProcessingStage::Enum stage)
|
||
{
|
||
ParserQuery parser;
|
||
ASTPtr ast;
|
||
std::string expected;
|
||
|
||
const char * begin = query.data();
|
||
const char * end = begin + query.size();
|
||
const char * pos = begin;
|
||
|
||
bool parse_res = parser.parse(pos, end, ast, expected);
|
||
|
||
/// Распарсенный запрос должен заканчиваться на конец входных данных или на точку с запятой.
|
||
if (!parse_res || (pos != end && *pos != ';'))
|
||
throw Exception("Syntax error: failed at position "
|
||
+ toString(pos - begin) + ": "
|
||
+ std::string(pos, std::min(SHOW_CHARS_ON_SYNTAX_ERROR, end - pos))
|
||
+ ", expected " + (parse_res ? "end of query" : expected) + ".",
|
||
ErrorCodes::SYNTAX_ERROR);
|
||
|
||
/// Проверка ограничений.
|
||
checkLimits(*ast, context.getSettingsRef().limits);
|
||
|
||
QuotaForIntervals & quota = context.getQuota();
|
||
time_t current_time = time(0);
|
||
|
||
quota.checkExceeded(current_time);
|
||
|
||
BlockIO res;
|
||
|
||
try
|
||
{
|
||
InterpreterQuery interpreter(ast, context, stage);
|
||
res = interpreter.execute();
|
||
}
|
||
catch (...)
|
||
{
|
||
quota.addError(current_time);
|
||
throw;
|
||
}
|
||
|
||
/// Положим запрос в список процессов. Но запрос SHOW PROCESSLIST класть не будем.
|
||
if (!internal && NULL == dynamic_cast<const ASTShowProcesslistQuery *>(&*ast))
|
||
res.process_list_entry = context.getProcessList().insert(query);
|
||
|
||
quota.addQuery(current_time);
|
||
return res;
|
||
}
|
||
|
||
|
||
}
|