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

66 lines
2.3 KiB
C
Raw Normal View History

2012-08-02 19:03:32 +00:00
#pragma once
#include <DB/Parsers/ASTSetQuery.h>
#include <DB/Interpreters/Context.h>
2015-06-18 02:11:05 +00:00
#include <DB/Interpreters/IInterpreter.h>
2012-08-02 19:03:32 +00:00
namespace DB
{
/** Установить один или несколько параметров, для сессии или глобально... или для текущего запроса.
2012-08-02 19:03:32 +00:00
*/
2015-06-18 02:11:05 +00:00
class InterpreterSetQuery : public IInterpreter
2012-08-02 19:03:32 +00:00
{
public:
InterpreterSetQuery(ASTPtr query_ptr_, Context & context_)
: query_ptr(query_ptr_), context(context_) {}
/** Обычный запрос SET. Задать настройку на сессию или глобальную (если указано GLOBAL).
*/
2015-06-18 02:11:05 +00:00
BlockIO execute() override
2012-08-02 19:03:32 +00:00
{
ASTSetQuery & ast = typeid_cast<ASTSetQuery &>(*query_ptr);
2012-08-02 19:03:32 +00:00
Context & target = ast.global ? context.getGlobalContext() : context.getSessionContext();
executeImpl(ast, target);
2015-06-18 02:11:05 +00:00
return {};
}
/** Задать настроку для текущего контекста (контекста запроса).
* Используется для интерпретации секции SETTINGS в запросе SELECT.
*/
void executeForCurrentContext()
{
ASTSetQuery & ast = typeid_cast<ASTSetQuery &>(*query_ptr);
executeImpl(ast, context);
}
private:
ASTPtr query_ptr;
Context & context;
2012-08-02 19:03:32 +00:00
void executeImpl(ASTSetQuery & ast, Context & target)
{
/** Значение readonly понимается следующим образом:
* 0 - можно всё.
* 1 - можно делать только запросы на чтение; в том числе, нельзя менять настройки.
* 2 - можно делать только запросы на чтение и можно менять настройки, кроме настройки readonly.
*/
if (context.getSettingsRef().limits.readonly == 1)
throw Exception("Cannot execute SET query in readonly mode", ErrorCodes::READONLY);
if (context.getSettingsRef().limits.readonly > 1)
for (ASTSetQuery::Changes::const_iterator it = ast.changes.begin(); it != ast.changes.end(); ++it)
if (it->name == "readonly")
throw Exception("Cannot modify 'readonly' setting in readonly mode", ErrorCodes::READONLY);
2012-08-02 19:03:32 +00:00
for (ASTSetQuery::Changes::const_iterator it = ast.changes.begin(); it != ast.changes.end(); ++it)
target.setSetting(it->name, it->value);
}
};
}