Files
scylladb/service/query_state.hh
Gleb Natapov 4ffc39d885 cql3: Extend the scope of group0_guard during DDL statement execution
Currently we hold group0_guard only during DDL statement's execute()
function, but unfortunately some statements access underlying schema
state also during check_access() and validate() calls which are called
by the query_processor before it calls execute. We need to cover those
calls with group0_guard as well and also move retry loop up. This patch
does it by introducing new function to cql_statement class take_guard().
Schema altering statements return group0 guard while others do not
return any guard. Query processor takes this guard at the beginning of a
statement execution and retries if service::group0_concurrent_modification
is thrown. The guard is passed to the execute in query_state structure.

Fixes: #13942

Message-ID: <ZNsynXayKim2XAFr@scylladb.com>
2023-08-17 15:52:48 +03:00

78 lines
1.7 KiB
C++

/*
* Copyright (C) 2015-present ScyllaDB
*/
/*
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
#ifndef SERVICE_QUERY_STATE_HH
#define SERVICE_QUERY_STATE_HH
#include "service/client_state.hh"
#include "tracing/tracing.hh"
#include "service_permit.hh"
#include "cql3/cql_statement.hh"
namespace qos {
class service_level_controller;
}
namespace service {
class query_state final {
private:
client_state& _client_state;
tracing::trace_state_ptr _trace_state_ptr;
service_permit _permit;
public:
query_state(client_state& client_state, service_permit permit)
: _client_state(client_state)
, _trace_state_ptr(tracing::trace_state_ptr())
, _permit(std::move(permit))
{}
query_state(client_state& client_state, tracing::trace_state_ptr trace_state_ptr, service_permit permit)
: _client_state(client_state)
, _trace_state_ptr(std::move(trace_state_ptr))
, _permit(std::move(permit))
{ }
const tracing::trace_state_ptr& get_trace_state() const {
return _trace_state_ptr;
}
tracing::trace_state_ptr& get_trace_state() {
return _trace_state_ptr;
}
client_state& get_client_state() {
return _client_state;
}
const client_state& get_client_state() const {
return _client_state;
}
api::timestamp_type get_timestamp() {
return _client_state.get_timestamp();
}
service_permit get_permit() const& {
return _permit;
}
service_permit&& get_permit() && {
return std::move(_permit);
}
qos::service_level_controller& get_service_level_controller() const {
return _client_state.get_service_level_controller();
}
};
}
#endif