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>
86 lines
2.7 KiB
C++
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();
|
|
}
|
|
|
|
}
|