ClickHouse/src/Interpreters/InterpreterSetRoleQuery.h
2021-05-31 17:49:02 +03:00

31 lines
660 B
C++

#pragma once
#include <Interpreters/IInterpreter.h>
#include <Parsers/IAST_fwd.h>
namespace DB
{
class ASTSetRoleQuery;
struct RolesOrUsersSet;
struct User;
class InterpreterSetRoleQuery : public IInterpreter, WithMutableContext
{
public:
InterpreterSetRoleQuery(const ASTPtr & query_ptr_, ContextMutablePtr context_) : WithMutableContext(context_), query_ptr(query_ptr_) {}
BlockIO execute() override;
static void updateUserSetDefaultRoles(User & user, const RolesOrUsersSet & roles_from_query);
private:
void setRole(const ASTSetRoleQuery & query);
void setDefaultRole(const ASTSetRoleQuery & query);
ASTPtr query_ptr;
};
}