Files
scylladb/cql3/statements/drop_service_level_statement.cc
2024-06-06 13:26:34 +02:00

45 lines
1.5 KiB
C++

/*
* Copyright (C) 2021-present ScyllaDB
*/
/*
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
#include "seastarx.hh"
#include "cql3/statements/drop_service_level_statement.hh"
#include "service/qos/service_level_controller.hh"
#include "service/client_state.hh"
#include "service/query_state.hh"
namespace cql3 {
namespace statements {
drop_service_level_statement::drop_service_level_statement(sstring service_level, bool if_exists) :
_service_level(service_level), _if_exists(if_exists) {}
std::unique_ptr<cql3::statements::prepared_statement>
cql3::statements::drop_service_level_statement::prepare(
data_dictionary::database db, cql_stats &stats) {
return std::make_unique<prepared_statement>(::make_shared<drop_service_level_statement>(*this));
}
future<> drop_service_level_statement::check_access(query_processor& qp, const service::client_state &state) const {
return state.ensure_has_permission(auth::command_desc{.permission = auth::permission::DROP, .resource = auth::root_service_level_resource()});
}
future<::shared_ptr<cql_transport::messages::result_message>>
drop_service_level_statement::execute(query_processor& qp,
service::query_state &state,
const query_options &,
std::optional<service::group0_guard> guard) const {
service::group0_batch mc{std::move(guard)};
auto& sl = state.get_service_level_controller();
co_await sl.drop_distributed_service_level(_service_level, _if_exists, mc);
co_await sl.commit_mutations(std::move(mc));
co_return nullptr;
}
}
}