mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 19:02:04 +00:00
74 lines
2.1 KiB
C++
74 lines
2.1 KiB
C++
#include <Parsers/ASTSetQuery.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Interpreters/InterpreterSetQuery.h>
|
|
#include <Common/typeid_cast.h>
|
|
#include <Common/FieldVisitors.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int READONLY;
|
|
extern const int QUERY_IS_PROHIBITED;
|
|
}
|
|
|
|
|
|
BlockIO InterpreterSetQuery::execute()
|
|
{
|
|
const ASTSetQuery & ast = typeid_cast<const ASTSetQuery &>(*query_ptr);
|
|
|
|
checkAccess(ast);
|
|
|
|
Context & target = context.getSessionContext();
|
|
for (const auto & change : ast.changes)
|
|
target.setSetting(change.name, change.value);
|
|
|
|
return {};
|
|
}
|
|
|
|
void InterpreterSetQuery::checkAccess(const ASTSetQuery & ast)
|
|
{
|
|
/** The `readonly` value is understood as follows:
|
|
* 0 - everything allowed.
|
|
* 1 - only read queries can be made; you can not change the settings.
|
|
* 2 - You can only do read queries and you can change the settings, except for the `readonly` setting.
|
|
*/
|
|
|
|
const Settings & settings = context.getSettingsRef();
|
|
auto readonly = settings.readonly;
|
|
auto allow_ddl = settings.allow_ddl;
|
|
|
|
for (const auto & change : ast.changes)
|
|
{
|
|
String value;
|
|
/// Setting isn't checked if value wasn't changed.
|
|
if (!settings.tryGet(change.name, value) || applyVisitor(FieldVisitorToString(), change.value) != value)
|
|
{
|
|
|
|
if (!allow_ddl && change.name == "allow_ddl")
|
|
throw Exception("Cannot modify 'allow_ddl' setting when DDL queries are prohibited for the user", ErrorCodes::QUERY_IS_PROHIBITED);
|
|
|
|
if (readonly == 1)
|
|
throw Exception("Cannot execute SET query in readonly mode", ErrorCodes::READONLY);
|
|
|
|
if (readonly > 1 && change.name == "readonly")
|
|
throw Exception("Cannot modify 'readonly' setting in readonly mode", ErrorCodes::READONLY);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void InterpreterSetQuery::executeForCurrentContext()
|
|
{
|
|
const ASTSetQuery & ast = typeid_cast<const ASTSetQuery &>(*query_ptr);
|
|
|
|
checkAccess(ast);
|
|
|
|
for (const auto & change : ast.changes)
|
|
context.setSetting(change.name, change.value);
|
|
}
|
|
|
|
|
|
}
|