2019-12-01 22:01:05 +00:00
|
|
|
#include <Interpreters/InterpreterDropAccessEntityQuery.h>
|
|
|
|
#include <Parsers/ASTDropAccessEntityQuery.h>
|
|
|
|
#include <Interpreters/Context.h>
|
2020-04-05 23:03:20 +00:00
|
|
|
#include <Interpreters/DDLWorker.h>
|
2019-12-01 22:01:05 +00:00
|
|
|
#include <Access/AccessControlManager.h>
|
2020-01-26 09:49:53 +00:00
|
|
|
#include <Access/AccessFlags.h>
|
2020-02-17 02:59:56 +00:00
|
|
|
#include <Access/User.h>
|
|
|
|
#include <Access/Role.h>
|
2019-12-01 22:01:05 +00:00
|
|
|
#include <Access/Quota.h>
|
2019-11-29 17:22:56 +00:00
|
|
|
#include <Access/RowPolicy.h>
|
2020-03-18 14:11:44 +00:00
|
|
|
#include <Access/SettingsProfile.h>
|
2019-11-29 17:22:56 +00:00
|
|
|
#include <boost/range/algorithm/transform.hpp>
|
2019-12-01 22:01:05 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
2020-03-18 14:11:44 +00:00
|
|
|
namespace
|
2019-12-01 22:01:05 +00:00
|
|
|
{
|
|
|
|
using Kind = ASTDropAccessEntityQuery::Kind;
|
|
|
|
|
2020-03-18 14:11:44 +00:00
|
|
|
std::type_index getType(Kind kind)
|
2019-12-01 22:01:05 +00:00
|
|
|
{
|
2020-03-18 14:11:44 +00:00
|
|
|
switch (kind)
|
2020-02-04 22:37:04 +00:00
|
|
|
{
|
2020-03-18 14:11:44 +00:00
|
|
|
case Kind::USER: return typeid(User);
|
|
|
|
case Kind::ROLE: return typeid(Role);
|
|
|
|
case Kind::QUOTA: return typeid(Quota);
|
|
|
|
case Kind::ROW_POLICY: return typeid(RowPolicy);
|
|
|
|
case Kind::SETTINGS_PROFILE: return typeid(SettingsProfile);
|
2020-02-17 02:59:56 +00:00
|
|
|
}
|
2020-03-18 14:11:44 +00:00
|
|
|
__builtin_unreachable();
|
|
|
|
}
|
2020-02-17 02:59:56 +00:00
|
|
|
|
2020-03-18 14:11:44 +00:00
|
|
|
AccessType getRequiredAccessType(Kind kind)
|
|
|
|
{
|
|
|
|
switch (kind)
|
2020-02-17 02:59:56 +00:00
|
|
|
{
|
2020-03-18 14:11:44 +00:00
|
|
|
case Kind::USER: return AccessType::DROP_USER;
|
|
|
|
case Kind::ROLE: return AccessType::DROP_ROLE;
|
|
|
|
case Kind::QUOTA: return AccessType::DROP_QUOTA;
|
2020-04-02 18:31:59 +00:00
|
|
|
case Kind::ROW_POLICY: return AccessType::DROP_ROW_POLICY;
|
2020-03-18 14:11:44 +00:00
|
|
|
case Kind::SETTINGS_PROFILE: return AccessType::DROP_SETTINGS_PROFILE;
|
2020-02-04 22:37:04 +00:00
|
|
|
}
|
2020-03-18 14:11:44 +00:00
|
|
|
__builtin_unreachable();
|
|
|
|
}
|
|
|
|
}
|
2020-02-04 22:37:04 +00:00
|
|
|
|
2020-03-18 14:11:44 +00:00
|
|
|
BlockIO InterpreterDropAccessEntityQuery::execute()
|
|
|
|
{
|
|
|
|
const auto & query = query_ptr->as<const ASTDropAccessEntityQuery &>();
|
|
|
|
auto & access_control = context.getAccessControlManager();
|
2020-02-04 22:37:04 +00:00
|
|
|
|
2020-03-18 14:11:44 +00:00
|
|
|
std::type_index type = getType(query.kind);
|
|
|
|
context.checkAccess(getRequiredAccessType(query.kind));
|
|
|
|
|
2020-04-05 23:03:20 +00:00
|
|
|
if (!query.cluster.empty())
|
|
|
|
return executeDDLQueryOnCluster(query_ptr, context);
|
|
|
|
|
2020-03-18 14:11:44 +00:00
|
|
|
if (query.kind == Kind::ROW_POLICY)
|
|
|
|
{
|
|
|
|
Strings full_names;
|
|
|
|
boost::range::transform(
|
|
|
|
query.row_policies_names, std::back_inserter(full_names),
|
|
|
|
[this](const RowPolicy::FullNameParts & row_policy_name) { return row_policy_name.getFullName(context); });
|
|
|
|
if (query.if_exists)
|
|
|
|
access_control.tryRemove(access_control.find<RowPolicy>(full_names));
|
|
|
|
else
|
|
|
|
access_control.remove(access_control.getIDs<RowPolicy>(full_names));
|
|
|
|
return {};
|
2019-12-01 22:01:05 +00:00
|
|
|
}
|
|
|
|
|
2020-03-18 14:11:44 +00:00
|
|
|
if (query.if_exists)
|
|
|
|
access_control.tryRemove(access_control.find(type, query.names));
|
|
|
|
else
|
|
|
|
access_control.remove(access_control.getIDs(type, query.names));
|
|
|
|
return {};
|
2019-12-01 22:01:05 +00:00
|
|
|
}
|
2020-03-18 14:11:44 +00:00
|
|
|
|
2019-12-01 22:01:05 +00:00
|
|
|
}
|