2019-12-01 22:01:05 +00:00
|
|
|
#include <Interpreters/InterpreterDropAccessEntityQuery.h>
|
|
|
|
#include <Parsers/ASTDropAccessEntityQuery.h>
|
|
|
|
#include <Interpreters/Context.h>
|
|
|
|
#include <Access/AccessControlManager.h>
|
|
|
|
#include <Access/Quota.h>
|
2019-11-29 17:22:56 +00:00
|
|
|
#include <Access/RowPolicy.h>
|
|
|
|
#include <boost/range/algorithm/transform.hpp>
|
2019-12-01 22:01:05 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
BlockIO InterpreterDropAccessEntityQuery::execute()
|
|
|
|
{
|
|
|
|
const auto & query = query_ptr->as<const ASTDropAccessEntityQuery &>();
|
|
|
|
auto & access_control = context.getAccessControlManager();
|
|
|
|
using Kind = ASTDropAccessEntityQuery::Kind;
|
|
|
|
|
|
|
|
switch (query.kind)
|
|
|
|
{
|
|
|
|
case Kind::QUOTA:
|
|
|
|
{
|
2020-01-24 16:20:36 +00:00
|
|
|
context.checkAccess(AccessType::DROP_QUOTA);
|
2019-12-01 22:01:05 +00:00
|
|
|
if (query.if_exists)
|
|
|
|
access_control.tryRemove(access_control.find<Quota>(query.names));
|
|
|
|
else
|
|
|
|
access_control.remove(access_control.getIDs<Quota>(query.names));
|
|
|
|
return {};
|
|
|
|
}
|
2019-11-29 17:22:56 +00:00
|
|
|
case Kind::ROW_POLICY:
|
|
|
|
{
|
2020-01-24 16:20:36 +00:00
|
|
|
context.checkAccess(AccessType::DROP_POLICY);
|
2019-11-29 17:22:56 +00:00
|
|
|
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
|
|
|
}
|
|
|
|
|
|
|
|
__builtin_unreachable();
|
|
|
|
}
|
|
|
|
}
|