Files
scylladb/cql3/values.cc
Jan Ciolek 855db49306 cql3: Add ostream operator for raw_value
It's possible to print raw_value_view,
but not raw_value. It would be useful
to be able to print both.

Implement printing raw_value
by creating raw_value_view from it
and printing the view.

Signed-off-by: Jan Ciolek <jan.ciolek@scylladb.com>
2022-10-18 22:48:25 +02:00

86 lines
2.7 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;
}
std::ostream& operator<<(std::ostream& os, const raw_value& value) {
return os << value.view();
}
}