Files
scylladb/cql3/values.cc
Avi Kivity eec441d365 cql3: values: add operator==() for raw_value
This is useful for implementing operator==() for expressions, which in
turn require comparing constants, which contain raw_values.

Note that this is not CQL comparison (that would be implemented
in cql3::expr::evaluate() and would return a CQL boolean, not a C++
boolean, but a traditional C++ value comparison.
2022-07-22 20:13:49 +03:00

82 lines
2.6 KiB
C++

/*
* Copyright (C) 2020-present ScyllaDB
*/
/*
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
#include "cql3/values.hh"
namespace cql3 {
std::ostream& operator<<(std::ostream& os, const raw_value_view& value) {
seastar::visit(value._data, [&] (FragmentedView auto v) {
os << "{ value: ";
for (bytes_view frag : fragment_range(v)) {
os << frag;
}
os << " }";
}, [&] (null_value) {
os << "{ null }";
}, [&] (unset_value) {
os << "{ unset }";
});
return os;
}
raw_value_view raw_value::view() const {
switch (_data.index()) {
case 0: return raw_value_view::make_value(managed_bytes_view(bytes_view(std::get<bytes>(_data))));
case 1: return raw_value_view::make_value(managed_bytes_view(std::get<managed_bytes>(_data)));
case 2: return raw_value_view::make_null();
default: return raw_value_view::make_unset_value();
}
}
raw_value raw_value::make_value(const raw_value_view& view) {
switch (view._data.index()) {
case 0: return raw_value::make_value(managed_bytes(std::get<fragmented_temporary_buffer::view>(view._data)));
case 1: return raw_value::make_value(managed_bytes(std::get<managed_bytes_view>(view._data)));
case 2: return raw_value::make_null();
default: return raw_value::make_unset_value();
}
}
raw_value_view raw_value_view::make_temporary(raw_value&& value) {
switch (value._data.index()) {
case 0: return raw_value_view(managed_bytes(std::get<bytes>(value._data)));
case 1: return raw_value_view(std::move(std::get<managed_bytes>(value._data)));
case 2: return raw_value_view::make_null();
case 3: return raw_value_view::make_unset_value();
default: throw std::runtime_error(fmt::format("raw_value_view::make_temporary bad index: {}", value._data.index()));
}
}
raw_value_view::raw_value_view(managed_bytes&& tmp) {
_temporary_storage = make_lw_shared<managed_bytes>(std::move(tmp));
_data = managed_bytes_view(*_temporary_storage);
}
bool operator==(const raw_value& v1, const raw_value& v2) {
if (v1.is_unset_value() && v2.is_unset_value()) {
return true;
}
if (v1.is_null() && v2.is_null()) {
return true; // note: this is not CQL comparison which would return NULL here
}
if (v1.is_value() && v2.is_value()) {
return v1.view().with_value([&] (const FragmentedView auto& v1v) {
return v2.view().with_value([&] (const FragmentedView auto& v2v) {
return compare_unsigned(v1v, v2v) == 0;
});
});
}
return false;
}
}