ParserRolesOrUsersSet.h 1.3 KB
Newer Older
1 2 3 4 5 6 7 8
#pragma once

#include <Parsers/IParserBase.h>


namespace DB
{
/** Parses a string like this:
9 10
  * {user_name | role_name | CURRENT_USER | ALL | NONE} [,...]
  * [EXCEPT {user_name | role_name | CURRENT_USER | ALL | NONE} [,...]]
11 12 13 14 15
  */
class ParserRolesOrUsersSet : public IParserBase
{
public:
    ParserRolesOrUsersSet & allowAll(bool allow_all_ = true) { allow_all = allow_all_; return *this; }
16
    ParserRolesOrUsersSet & allowAny(bool allow_any_ = true) { allow_any = allow_any_; return *this; }
17
    ParserRolesOrUsersSet & allowUsers(bool allow_users_ = true) { allow_users = allow_users_; return *this; }
18
    ParserRolesOrUsersSet & allowCurrentUser(bool allow_current_user_ = true) { allow_current_user = allow_current_user_; return *this; }
19
    ParserRolesOrUsersSet & allowRoles(bool allow_roles_ = true) { allow_roles = allow_roles_; return *this; }
20 21 22 23 24 25 26 27
    ParserRolesOrUsersSet & useIDMode(bool id_mode_ = true) { id_mode = id_mode_; return *this; }

protected:
    const char * getName() const override { return "RolesOrUsersSet"; }
    bool parseImpl(Pos & pos, ASTPtr & node, Expected & expected) override;

private:
    bool allow_all = false;
28
    bool allow_any = false;
29
    bool allow_users = false;
30
    bool allow_current_user = false;
31
    bool allow_roles = false;
32 33 34 35
    bool id_mode = false;
};

}