mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-21 17:10:35 +00:00
3d38708434
To prepare a user-defined type, we need to look up its name in the keyspace. While we get the keyspace name as an argument to prepare(), it is useless without the database instance. Fix the problem by passing a database reference along with the keyspace. This precolates through the class structure, so most cql3 raw types end up receiving this treatment. Origin gets along without it by using a singleton. We can't do this due to sharding (we could use a thread-local instance, but that's ugly too). Hopefully the transition to a visitor will clean this up.
218 lines
9.6 KiB
C++
218 lines
9.6 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 Cloudius Systems, Ltd.
|
|
*/
|
|
|
|
#include "operation.hh"
|
|
#include "operation_impl.hh"
|
|
#include "maps.hh"
|
|
#include "sets.hh"
|
|
#include "lists.hh"
|
|
|
|
namespace cql3 {
|
|
|
|
|
|
shared_ptr<operation>
|
|
operation::set_element::prepare(database& db, const sstring& keyspace, const column_definition& receiver) {
|
|
using exceptions::invalid_request_exception;
|
|
auto rtype = dynamic_pointer_cast<collection_type_impl>(receiver.type);
|
|
if (!rtype) {
|
|
throw invalid_request_exception(sprint("Invalid operation (%s) for non collection column %s", receiver, receiver.name()));
|
|
} else if (!rtype->is_multi_cell()) {
|
|
throw invalid_request_exception(sprint("Invalid operation (%s) for frozen collection column %s", receiver, receiver.name()));
|
|
}
|
|
|
|
if (&rtype->_kind == &collection_type_impl::kind::list) {
|
|
auto&& idx = _selector->prepare(db, keyspace, lists::index_spec_of(receiver.column_specification));
|
|
auto&& lval = _value->prepare(db, keyspace, lists::value_spec_of(receiver.column_specification));
|
|
return make_shared<lists::setter_by_index>(receiver, idx, lval);
|
|
} else if (&rtype->_kind == &collection_type_impl::kind::set) {
|
|
throw invalid_request_exception(sprint("Invalid operation (%s) for set column %s", receiver, receiver.name()));
|
|
} else if (&rtype->_kind == &collection_type_impl::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(shared_ptr<raw_update> other) {
|
|
// 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_pointer_cast<set_value>(std::move(other));
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::addition::prepare(database& db, const sstring& keyspace, const column_definition& receiver) {
|
|
auto v = _value->prepare(db, keyspace, receiver.column_specification);
|
|
|
|
auto ctype = dynamic_pointer_cast<collection_type_impl>(receiver.type);
|
|
if (!ctype) {
|
|
fail(unimplemented::cause::COUNTERS);
|
|
// FIXME: implelement
|
|
#if 0
|
|
if (!(receiver.type instanceof CounterColumnType))
|
|
throw new InvalidRequestException(String.format("Invalid operation (%s) for non counter column %s", toString(receiver), receiver.name));
|
|
return new Constants.Adder(receiver, v);
|
|
#endif
|
|
} else if (!ctype->is_multi_cell()) {
|
|
throw exceptions::invalid_request_exception(sprint("Invalid operation (%s) for frozen collection column %s", receiver, receiver.name()));
|
|
}
|
|
|
|
if (&ctype->_kind == &collection_type_impl::kind::list) {
|
|
warn(unimplemented::cause::COLLECTIONS);
|
|
// FIXME:
|
|
// return new Lists.Appender(receiver, v);
|
|
throw exceptions::invalid_request_exception("unimplemented, bye");
|
|
} else if (&ctype->_kind == &collection_type_impl::kind::set) {
|
|
return make_shared<sets::adder>(receiver, v);
|
|
} else if (&ctype->_kind == &collection_type_impl::kind::map) {
|
|
return make_shared<maps::putter>(receiver, v);
|
|
} else {
|
|
abort();
|
|
}
|
|
}
|
|
|
|
bool
|
|
operation::addition::is_compatible_with(shared_ptr<raw_update> other) {
|
|
return !dynamic_pointer_cast<set_value>(other);
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::subtraction::prepare(database& db, const sstring& keyspace, const column_definition& receiver) {
|
|
auto ctype = dynamic_pointer_cast<collection_type_impl>(receiver.type);
|
|
if (!ctype) {
|
|
fail(unimplemented::cause::COUNTERS);
|
|
#if 0
|
|
if (!(receiver.type instanceof CounterColumnType))
|
|
throw new InvalidRequestException(String.format("Invalid operation (%s) for non counter column %s", toString(receiver), receiver.name));
|
|
return new Constants.Substracter(receiver, value.prepare(keyspace, receiver));
|
|
#endif
|
|
}
|
|
if (!ctype->is_multi_cell()) {
|
|
throw exceptions::invalid_request_exception(
|
|
sprint("Invalid operation (%s) for frozen collection column %s", receiver, receiver.name()));
|
|
}
|
|
|
|
if (&ctype->_kind == &collection_type_impl::kind::list) {
|
|
return make_shared<lists::discarder>(receiver, _value->prepare(db, keyspace, receiver.column_specification));
|
|
} else if (&ctype->_kind == &collection_type_impl::kind::set) {
|
|
return make_shared<sets::discarder>(receiver, _value->prepare(db, keyspace, receiver.column_specification));
|
|
} else if (&ctype->_kind == &collection_type_impl::kind::map) {
|
|
auto&& mtype = dynamic_pointer_cast<map_type_impl>(ctype);
|
|
// The value for a map subtraction is actually a set
|
|
auto&& vr = make_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(shared_ptr<raw_update> other) {
|
|
return !dynamic_pointer_cast<set_value>(other);
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::prepend::prepare(database& db, const sstring& keyspace, const column_definition& receiver) {
|
|
warn(unimplemented::cause::COLLECTIONS);
|
|
throw exceptions::invalid_request_exception("unimplemented, go away");
|
|
// FIXME:
|
|
#if 0
|
|
auto v = _value->prepare(keyspace, receiver.column_specification);
|
|
|
|
if (!(receiver.type instanceof ListType))
|
|
throw new InvalidRequestException(String.format("Invalid operation (%s) for non list column %s", toString(receiver), receiver.name));
|
|
else if (!(receiver.type.isMultiCell()))
|
|
throw new InvalidRequestException(String.format("Invalid operation (%s) for frozen list column %s", toString(receiver), receiver.name));
|
|
|
|
return new Lists.Prepender(receiver, v);
|
|
#endif
|
|
}
|
|
|
|
bool
|
|
operation::prepend::is_compatible_with(shared_ptr<raw_update> other) {
|
|
return !dynamic_pointer_cast<set_value>(other);
|
|
}
|
|
|
|
|
|
::shared_ptr <operation>
|
|
operation::set_value::prepare(database& db, const sstring& keyspace, const column_definition& receiver) {
|
|
auto v = _value->prepare(db, keyspace, receiver.column_specification);
|
|
|
|
if (receiver.type->is_counter()) {
|
|
throw exceptions::invalid_request_exception(sprint("Cannot set the value of counter column %s (counters can only be incremented/decremented, not set)", receiver.name_as_text()));
|
|
}
|
|
|
|
if (!receiver.type->is_collection()) {
|
|
return ::make_shared<constants::setter>(receiver, v);
|
|
}
|
|
|
|
auto& k = static_pointer_cast<collection_type_impl>(receiver.type)->_kind;
|
|
if (&k == &collection_type_impl::kind::list) {
|
|
return make_shared<lists::setter>(receiver, v);
|
|
} else if (&k == &collection_type_impl::kind::set) {
|
|
return make_shared<sets::setter>(receiver, v);
|
|
} else if (&k == &collection_type_impl::kind::map) {
|
|
return make_shared<maps::setter>(receiver, v);
|
|
} else {
|
|
abort();
|
|
}
|
|
}
|
|
|
|
bool
|
|
operation::set_value::is_compatible_with(::shared_ptr <raw_update> other) {
|
|
// 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;
|
|
}
|
|
|
|
shared_ptr<column_identifier::raw>
|
|
operation::element_deletion::affected_column() {
|
|
return _id;
|
|
}
|
|
|
|
shared_ptr<operation>
|
|
operation::element_deletion::prepare(database& db, const sstring& keyspace, const column_definition& receiver) {
|
|
if (!receiver.type->is_collection()) {
|
|
throw exceptions::invalid_request_exception(sprint("Invalid deletion operation for non collection column %s", receiver.name()));
|
|
} else if (!receiver.type->is_multi_cell()) {
|
|
throw exceptions::invalid_request_exception(sprint("Invalid deletion operation for frozen collection column %s", receiver.name()));
|
|
}
|
|
auto ctype = static_pointer_cast<collection_type_impl>(receiver.type);
|
|
if (&ctype->_kind == &collection_type_impl::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->_kind == &collection_type_impl::kind::set) {
|
|
auto&& elt = _element->prepare(db, keyspace, sets::value_spec_of(receiver.column_specification));
|
|
return make_shared<sets::discarder>(receiver, std::move(elt));
|
|
} else if (&ctype->_kind == &collection_type_impl::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();
|
|
}
|
|
|
|
}
|