022d5f6498
These are not shared anywhere and so can be easily changed to be stored in std::unique_ptr instead of shared_ptr's. Tests: unit(dev, debug) Signed-off-by: Pavel Solodovnikov <pa.solodovnikov@scylladb.com> Message-Id: <20200531201051.946432-1-pa.solodovnikov@scylladb.com>
392 lines
17 KiB
C++
392 lines
17 KiB
C++
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
/*
|
|
* Copyright (C) 2015 ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* This file is part of Scylla.
|
|
*
|
|
* Scylla is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Scylla is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Scylla. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#include <utility>
|
|
|
|
#include "operation.hh"
|
|
#include "operation_impl.hh"
|
|
#include "maps.hh"
|
|
#include "sets.hh"
|
|
#include "lists.hh"
|
|
#include "user_types.hh"
|
|
#include "types/tuple.hh"
|
|
#include "types/map.hh"
|
|
#include "types/list.hh"
|
|
#include "types/set.hh"
|
|
#include "types/user.hh"
|
|
|
|
namespace cql3 {
|
|
|
|
sstring
|
|
operation::set_element::to_string(const column_definition& receiver) const {
|
|
return format("{}[{}] = {}", receiver.name_as_text(), *_selector, *_value);
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::set_element::prepare(database& db, const sstring& keyspace, const column_definition& receiver) const {
|
|
using exceptions::invalid_request_exception;
|
|
auto rtype = dynamic_pointer_cast<const collection_type_impl>(receiver.type);
|
|
if (!rtype) {
|
|
throw invalid_request_exception(format("Invalid operation ({}) for non collection column {}", to_string(receiver), receiver.name()));
|
|
} else if (!rtype->is_multi_cell()) {
|
|
throw invalid_request_exception(format("Invalid operation ({}) for frozen collection column {}", to_string(receiver), receiver.name()));
|
|
}
|
|
|
|
if (rtype->get_kind() == abstract_type::kind::list) {
|
|
auto&& lval = _value->prepare(db, keyspace, lists::value_spec_of(*receiver.column_specification));
|
|
if (_by_uuid) {
|
|
auto&& idx = _selector->prepare(db, keyspace, lists::uuid_index_spec_of(*receiver.column_specification));
|
|
return make_shared<lists::setter_by_uuid>(receiver, idx, lval);
|
|
} else {
|
|
auto&& idx = _selector->prepare(db, keyspace, lists::index_spec_of(*receiver.column_specification));
|
|
return make_shared<lists::setter_by_index>(receiver, idx, lval);
|
|
}
|
|
} else if (rtype->get_kind() == abstract_type::kind::set) {
|
|
throw invalid_request_exception(format("Invalid operation ({}) for set column {}", to_string(receiver), receiver.name()));
|
|
} else if (rtype->get_kind() == abstract_type::kind::map) {
|
|
auto key = _selector->prepare(db, keyspace, maps::key_spec_of(*receiver.column_specification));
|
|
auto mval = _value->prepare(db, keyspace, maps::value_spec_of(*receiver.column_specification));
|
|
return make_shared<maps::setter_by_key>(receiver, key, mval);
|
|
}
|
|
abort();
|
|
}
|
|
|
|
bool
|
|
operation::set_element::is_compatible_with(const std::unique_ptr<raw_update>& other) const {
|
|
// TODO: we could check that the other operation is not setting the same element
|
|
// too (but since the index/key set may be a bind variables we can't always do it at this point)
|
|
return !dynamic_cast<const set_value*>(other.get());
|
|
}
|
|
|
|
sstring
|
|
operation::set_field::to_string(const column_definition& receiver) const {
|
|
return format("{}.{} = {}", receiver.name_as_text(), *_field, *_value);
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::set_field::prepare(database& db, const sstring& keyspace, const column_definition& receiver) const {
|
|
if (!receiver.type->is_user_type()) {
|
|
throw exceptions::invalid_request_exception(
|
|
format("Invalid operation ({}) for non-UDT column {}", to_string(receiver), receiver.name_as_text()));
|
|
} else if (!receiver.type->is_multi_cell()) {
|
|
throw exceptions::invalid_request_exception(
|
|
format("Invalid operation ({}) for frozen UDT column {}", to_string(receiver), receiver.name_as_text()));
|
|
}
|
|
|
|
auto& type = static_cast<const user_type_impl&>(*receiver.type);
|
|
auto idx = type.idx_of_field(_field->name());
|
|
if (!idx) {
|
|
throw exceptions::invalid_request_exception(
|
|
format("UDT column {} does not have a field named {}", receiver.name_as_text(), *_field));
|
|
}
|
|
|
|
auto val = _value->prepare(db, keyspace, user_types::field_spec_of(*receiver.column_specification, *idx));
|
|
return make_shared<user_types::setter_by_field>(receiver, *idx, std::move(val));
|
|
}
|
|
|
|
bool
|
|
operation::set_field::is_compatible_with(const std::unique_ptr<raw_update>& other) const {
|
|
auto x = dynamic_cast<const set_field*>(other.get());
|
|
if (x) {
|
|
return _field != x->_field;
|
|
}
|
|
|
|
return !dynamic_cast<const set_value*>(other.get());
|
|
}
|
|
|
|
const column_identifier::raw&
|
|
operation::field_deletion::affected_column() const {
|
|
return *_id;
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::field_deletion::prepare(database& db, const sstring& keyspace, const column_definition& receiver) const {
|
|
if (!receiver.type->is_user_type()) {
|
|
throw exceptions::invalid_request_exception(
|
|
format("Invalid deletion operation for non-UDT column {}", receiver.name_as_text()));
|
|
} else if (!receiver.type->is_multi_cell()) {
|
|
throw exceptions::invalid_request_exception(
|
|
format("Frozen UDT column {} does not support field deletions", receiver.name_as_text()));
|
|
}
|
|
|
|
auto type = static_cast<const user_type_impl*>(receiver.type.get());
|
|
auto idx = type->idx_of_field(_field->name());
|
|
if (!idx) {
|
|
throw exceptions::invalid_request_exception(
|
|
format("UDT column {} does not have a field named {}", receiver.name_as_text(), *_field));
|
|
}
|
|
|
|
return make_shared<user_types::deleter_by_field>(receiver, *idx);
|
|
}
|
|
|
|
sstring
|
|
operation::addition::to_string(const column_definition& receiver) const {
|
|
return format("{} = {} + {}", receiver.name_as_text(), receiver.name_as_text(), *_value);
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::addition::prepare(database& db, const sstring& keyspace, const column_definition& receiver) const {
|
|
auto v = _value->prepare(db, keyspace, receiver.column_specification);
|
|
|
|
auto ctype = dynamic_pointer_cast<const collection_type_impl>(receiver.type);
|
|
if (!ctype) {
|
|
if (!receiver.is_counter()) {
|
|
throw exceptions::invalid_request_exception(format("Invalid operation ({}) for non counter column {}", to_string(receiver), receiver.name()));
|
|
}
|
|
return make_shared<constants::adder>(receiver, v);
|
|
} else if (!ctype->is_multi_cell()) {
|
|
throw exceptions::invalid_request_exception(format("Invalid operation ({}) for frozen collection column {}", to_string(receiver), receiver.name()));
|
|
}
|
|
|
|
if (ctype->get_kind() == abstract_type::kind::list) {
|
|
return make_shared<lists::appender>(receiver, v);
|
|
} else if (ctype->get_kind() == abstract_type::kind::set) {
|
|
return make_shared<sets::adder>(receiver, v);
|
|
} else if (ctype->get_kind() == abstract_type::kind::map) {
|
|
return make_shared<maps::putter>(receiver, v);
|
|
} else {
|
|
abort();
|
|
}
|
|
}
|
|
|
|
bool
|
|
operation::addition::is_compatible_with(const std::unique_ptr<raw_update>& other) const {
|
|
return !dynamic_cast<const set_value*>(other.get());
|
|
}
|
|
|
|
sstring
|
|
operation::subtraction::to_string(const column_definition& receiver) const {
|
|
return format("{} = {} - {}", receiver.name_as_text(), receiver.name_as_text(), *_value);
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::subtraction::prepare(database& db, const sstring& keyspace, const column_definition& receiver) const {
|
|
auto ctype = dynamic_pointer_cast<const collection_type_impl>(receiver.type);
|
|
if (!ctype) {
|
|
if (!receiver.is_counter()) {
|
|
throw exceptions::invalid_request_exception(format("Invalid operation ({}) for non counter column {}", to_string(receiver), receiver.name()));
|
|
}
|
|
auto v = _value->prepare(db, keyspace, receiver.column_specification);
|
|
return make_shared<constants::subtracter>(receiver, v);
|
|
}
|
|
if (!ctype->is_multi_cell()) {
|
|
throw exceptions::invalid_request_exception(
|
|
format("Invalid operation ({}) for frozen collection column {}", to_string(receiver), receiver.name()));
|
|
}
|
|
|
|
if (ctype->get_kind() == abstract_type::kind::list) {
|
|
return make_shared<lists::discarder>(receiver, _value->prepare(db, keyspace, receiver.column_specification));
|
|
} else if (ctype->get_kind() == abstract_type::kind::set) {
|
|
return make_shared<sets::discarder>(receiver, _value->prepare(db, keyspace, receiver.column_specification));
|
|
} else if (ctype->get_kind() == abstract_type::kind::map) {
|
|
auto&& mtype = dynamic_pointer_cast<const map_type_impl>(ctype);
|
|
// The value for a map subtraction is actually a set
|
|
auto&& vr = make_lw_shared<column_specification>(
|
|
receiver.column_specification->ks_name,
|
|
receiver.column_specification->cf_name,
|
|
receiver.column_specification->name,
|
|
set_type_impl::get_instance(mtype->get_keys_type(), false));
|
|
return ::make_shared<sets::discarder>(receiver, _value->prepare(db, keyspace, std::move(vr)));
|
|
}
|
|
abort();
|
|
}
|
|
|
|
bool
|
|
operation::subtraction::is_compatible_with(const std::unique_ptr<raw_update>& other) const {
|
|
return !dynamic_cast<const set_value*>(other.get());
|
|
}
|
|
|
|
sstring
|
|
operation::prepend::to_string(const column_definition& receiver) const {
|
|
return format("{} = {} + {}", receiver.name_as_text(), *_value, receiver.name_as_text());
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::prepend::prepare(database& db, const sstring& keyspace, const column_definition& receiver) const {
|
|
auto v = _value->prepare(db, keyspace, receiver.column_specification);
|
|
|
|
if (!dynamic_cast<const list_type_impl*>(receiver.type.get())) {
|
|
throw exceptions::invalid_request_exception(format("Invalid operation ({}) for non list column {}", to_string(receiver), receiver.name()));
|
|
} else if (!receiver.type->is_multi_cell()) {
|
|
throw exceptions::invalid_request_exception(format("Invalid operation ({}) for frozen list column {}", to_string(receiver), receiver.name()));
|
|
}
|
|
|
|
return make_shared<lists::prepender>(receiver, std::move(v));
|
|
}
|
|
|
|
bool
|
|
operation::prepend::is_compatible_with(const std::unique_ptr<raw_update>& other) const {
|
|
return !dynamic_cast<const set_value*>(other.get());
|
|
}
|
|
|
|
|
|
::shared_ptr <operation>
|
|
operation::set_value::prepare(database& db, const sstring& keyspace, const column_definition& receiver) const {
|
|
auto v = _value->prepare(db, keyspace, receiver.column_specification);
|
|
|
|
if (receiver.type->is_counter()) {
|
|
throw exceptions::invalid_request_exception(format("Cannot set the value of counter column {} (counters can only be incremented/decremented, not set)", receiver.name_as_text()));
|
|
}
|
|
|
|
if (receiver.type->is_collection()) {
|
|
auto k = receiver.type->get_kind();
|
|
if (k == abstract_type::kind::list) {
|
|
return make_shared<lists::setter>(receiver, v);
|
|
} else if (k == abstract_type::kind::set) {
|
|
return make_shared<sets::setter>(receiver, v);
|
|
} else if (k == abstract_type::kind::map) {
|
|
return make_shared<maps::setter>(receiver, v);
|
|
} else {
|
|
abort();
|
|
}
|
|
}
|
|
|
|
if (receiver.type->is_user_type()) {
|
|
return make_shared<user_types::setter>(receiver, v);
|
|
}
|
|
|
|
return ::make_shared<constants::setter>(receiver, v);
|
|
}
|
|
|
|
::shared_ptr <operation>
|
|
operation::set_counter_value_from_tuple_list::prepare(database& db, const sstring& keyspace, const column_definition& receiver) const {
|
|
static thread_local const data_type counter_tuple_type = tuple_type_impl::get_instance({int32_type, uuid_type, long_type, long_type});
|
|
static thread_local const data_type counter_tuple_list_type = list_type_impl::get_instance(counter_tuple_type, true);
|
|
|
|
if (!receiver.type->is_counter()) {
|
|
throw exceptions::invalid_request_exception(format("Column {} is not a counter", receiver.name_as_text()));
|
|
}
|
|
|
|
// We need to fake a column of list<tuple<...>> to prepare the value term
|
|
auto & os = receiver.column_specification;
|
|
auto spec = make_lw_shared<cql3::column_specification>(os->ks_name, os->cf_name, os->name, counter_tuple_list_type);
|
|
auto v = _value->prepare(db, keyspace, spec);
|
|
|
|
// Will not be used elsewhere, so make it local.
|
|
class counter_setter : public operation {
|
|
public:
|
|
using operation::operation;
|
|
|
|
bool is_raw_counter_shard_write() const override {
|
|
return true;
|
|
}
|
|
void execute(mutation& m, const clustering_key_prefix& prefix, const update_parameters& params) override {
|
|
const auto& value = _t->bind(params._options);
|
|
auto&& list_value = dynamic_pointer_cast<lists::value>(value);
|
|
|
|
if (!list_value) {
|
|
throw std::invalid_argument("Invalid input data to counter set");
|
|
}
|
|
|
|
counter_id last(utils::UUID(0, 0));
|
|
counter_cell_builder ccb(list_value->_elements.size());
|
|
for (auto& bo : list_value->_elements) {
|
|
// lexical etc cast fails should be enough type checking here.
|
|
auto tuple = value_cast<tuple_type_impl::native_type>(counter_tuple_type->deserialize(*bo));
|
|
auto shard = value_cast<int>(tuple[0]);
|
|
auto id = counter_id(value_cast<utils::UUID>(tuple[1]));
|
|
auto clock = value_cast<int64_t>(tuple[2]);
|
|
auto value = value_cast<int64_t>(tuple[3]);
|
|
|
|
using namespace std::rel_ops;
|
|
|
|
if (id <= last) {
|
|
throw marshal_exception(
|
|
format("invalid counter id order, {} <= {}",
|
|
id.to_uuid().to_sstring(),
|
|
last.to_uuid().to_sstring()));
|
|
}
|
|
last = id;
|
|
// TODO: maybe allow more than global values to propagate,
|
|
// though we don't (yet at least) in sstable::partition so...
|
|
switch (shard) {
|
|
case 'g':
|
|
ccb.add_shard(counter_shard(id, value, clock));
|
|
break;
|
|
case 'l':
|
|
throw marshal_exception("encountered a local shard in a counter cell");
|
|
case 'r':
|
|
throw marshal_exception("encountered remote shards in a counter cell");
|
|
default:
|
|
throw marshal_exception(format("encountered unknown shard {:d} in a counter cell", shard));
|
|
}
|
|
}
|
|
// Note. this is a counter value cell, not an update.
|
|
// see counters.cc, we need to detect this.
|
|
m.set_cell(prefix, column, ccb.build(params.timestamp()));
|
|
}
|
|
};
|
|
|
|
return make_shared<counter_setter>(receiver, v);
|
|
};
|
|
|
|
bool
|
|
operation::set_value::is_compatible_with(const std::unique_ptr<raw_update>& other) const {
|
|
// We don't allow setting multiple time the same column, because 1)
|
|
// it's stupid and 2) the result would seem random to the user.
|
|
return false;
|
|
}
|
|
|
|
const column_identifier::raw&
|
|
operation::element_deletion::affected_column() const {
|
|
return *_id;
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::element_deletion::prepare(database& db, const sstring& keyspace, const column_definition& receiver) const {
|
|
if (!receiver.type->is_collection()) {
|
|
throw exceptions::invalid_request_exception(format("Invalid deletion operation for non collection column {}", receiver.name()));
|
|
} else if (!receiver.type->is_multi_cell()) {
|
|
throw exceptions::invalid_request_exception(format("Invalid deletion operation for frozen collection column {}", receiver.name()));
|
|
}
|
|
auto ctype = static_pointer_cast<const collection_type_impl>(receiver.type);
|
|
if (ctype->get_kind() == abstract_type::kind::list) {
|
|
auto&& idx = _element->prepare(db, keyspace, lists::index_spec_of(*receiver.column_specification));
|
|
return make_shared<lists::discarder_by_index>(receiver, std::move(idx));
|
|
} else if (ctype->get_kind() == abstract_type::kind::set) {
|
|
auto&& elt = _element->prepare(db, keyspace, sets::value_spec_of(*receiver.column_specification));
|
|
return make_shared<sets::element_discarder>(receiver, std::move(elt));
|
|
} else if (ctype->get_kind() == abstract_type::kind::map) {
|
|
auto&& key = _element->prepare(db, keyspace, maps::key_spec_of(*receiver.column_specification));
|
|
return make_shared<maps::discarder_by_key>(receiver, std::move(key));
|
|
}
|
|
abort();
|
|
}
|
|
|
|
}
|