mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-21 17:10:35 +00:00
Metadata id was introduced in CQLv5 to make metadata of prepared statement consistent between driver and database. This commit introduces a protocol extension that allows to use the same mechanism in CQLv4. This change: - Introduce SCYLLA_USE_METADATA_ID protocol extension for CQLv4 - Introduce METADATA_CHANGED flag in RESULT. The flag cames directly from CQLv5 binary protocol. In CQLv4, the bit was never used, so we assume it is safe to reuse it. - Implement handling of metadata_id and METADATA_CHANGED in RESULT rows - Implement returning metadata_id in RESULT prepared - Implement reading metadata_id from EXECUTE - Added description of SCYLLA_USE_METADATA_ID in documentation Metadata_id is wrapped in cql_metadata_id_wrapper because we need to distinguish the following situations: - Metadata_id is not supported by the protocol (e.g. CQLv4 without the extension is used) - Metadata_id is supported by the protocol but not set - e.g. PREPARE query is being handled: it doesn't contain metadata_id in the request but the reply (RESULT prepared) must contain metadata_id - Metadata_id is supported by the protocol and set, any number of bytes >= 0 is allowed, according to the CQLv5 protocol specification Fixes scylladb/scylladb#20860
82 lines
2.6 KiB
C++
82 lines
2.6 KiB
C++
/*
|
|
* Copyright (C) 2014-present ScyllaDB
|
|
*
|
|
* Modified by ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: (LicenseRef-ScyllaDB-Source-Available-1.0 and Apache-2.0)
|
|
*/
|
|
|
|
#include "parsed_statement.hh"
|
|
|
|
#include "cql3/statements/prepared_statement.hh"
|
|
#include "cql3/column_specification.hh"
|
|
|
|
#include "cql3/cql_statement.hh"
|
|
#include "cql3/result_set.hh"
|
|
|
|
namespace cql3 {
|
|
|
|
namespace statements {
|
|
|
|
namespace raw {
|
|
|
|
parsed_statement::~parsed_statement()
|
|
{ }
|
|
|
|
prepare_context& parsed_statement::get_prepare_context() {
|
|
return _prepare_ctx;
|
|
}
|
|
|
|
const prepare_context& parsed_statement::get_prepare_context() const {
|
|
return _prepare_ctx;
|
|
}
|
|
|
|
// Used by the parser and preparable statement
|
|
void parsed_statement::set_bound_variables(const std::vector<::shared_ptr<column_identifier>>& bound_names) {
|
|
_prepare_ctx.set_bound_variables(bound_names);
|
|
}
|
|
|
|
}
|
|
|
|
prepared_statement::prepared_statement(
|
|
audit::audit_info_ptr&& audit_info,
|
|
::shared_ptr<cql_statement> statement_, std::vector<lw_shared_ptr<column_specification>> bound_names_,
|
|
std::vector<uint16_t> partition_key_bind_indices, std::vector<sstring> warnings)
|
|
: statement(std::move(statement_))
|
|
, bound_names(std::move(bound_names_))
|
|
, partition_key_bind_indices(std::move(partition_key_bind_indices))
|
|
, warnings(std::move(warnings))
|
|
, _metadata_id(bytes{})
|
|
{
|
|
statement->set_audit_info(std::move(audit_info));
|
|
}
|
|
|
|
prepared_statement::prepared_statement(
|
|
audit::audit_info_ptr&& audit_info,
|
|
::shared_ptr<cql_statement> statement_, const prepare_context& ctx,
|
|
const std::vector<uint16_t>& partition_key_bind_indices, std::vector<sstring> warnings)
|
|
: prepared_statement(std::move(audit_info), statement_, ctx.get_variable_specifications(), partition_key_bind_indices, std::move(warnings))
|
|
{ }
|
|
|
|
prepared_statement::prepared_statement(audit::audit_info_ptr&& audit_info, ::shared_ptr<cql_statement> statement_, prepare_context&& ctx, std::vector<uint16_t>&& partition_key_bind_indices)
|
|
: prepared_statement(std::move(audit_info), statement_, std::move(ctx).get_variable_specifications(), std::move(partition_key_bind_indices))
|
|
{ }
|
|
|
|
prepared_statement::prepared_statement(audit::audit_info_ptr&& audit_info, ::shared_ptr<cql_statement>&& statement_, std::vector<sstring> warnings)
|
|
: prepared_statement(std::move(audit_info), statement_, std::vector<lw_shared_ptr<column_specification>>(), std::vector<uint16_t>(), std::move(warnings))
|
|
{ }
|
|
|
|
void prepared_statement::calculate_metadata_id() {
|
|
_metadata_id = statement->get_result_metadata()->calculate_metadata_id();
|
|
}
|
|
|
|
cql_metadata_id_type prepared_statement::get_metadata_id() const {
|
|
return _metadata_id;
|
|
}
|
|
|
|
}
|
|
|
|
}
|