Files
scylladb/sstables/m_format_read_helpers.cc
Avi Kivity f3eade2f62 treewide: relicense to ScyllaDB-Source-Available-1.0
Drop the AGPL license in favor of a source-available license.
See the blog post [1] for details.

[1] https://www.scylladb.com/2024/12/18/why-were-moving-to-a-source-available-license/
2024-12-18 17:45:13 +02:00

85 lines
2.5 KiB
C++

/*
* Copyright (C) 2018-present ScyllaDB
*/
/*
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
*/
#include "vint-serialization.hh"
#include "sstables/m_format_read_helpers.hh"
#include "sstables/exceptions.hh"
#include "sstables/random_access_reader.hh"
#include "sstables/mx/types.hh"
#include <fmt/format.h>
namespace sstables {
static void check_buf_size(temporary_buffer<char>& buf, size_t expected) {
if (buf.size() < expected) {
throw bufsize_mismatch_exception(buf.size(), expected);
}
}
template <typename T>
inline future<T> read_vint_impl(random_access_reader& in) {
using vint_type = std::conditional_t<std::is_unsigned_v<T>, unsigned_vint, signed_vint>;
return in.read_exactly(1).then([&in] (auto&& buf) {
check_buf_size(buf, 1);
vint_size_type len = vint_type::serialized_size_from_first_byte(*buf.begin());
temporary_buffer<char> bytes(len);
std::copy_n(buf.begin(), 1, bytes.get_write());
return in.read_exactly(len - 1).then([len, bytes = std::move(bytes)] (temporary_buffer<char> buf) mutable {
check_buf_size(buf, len - 1);
std::copy_n(buf.begin(), len - 1, bytes.get_write() + 1);
return vint_type::deserialize(
bytes_view(reinterpret_cast<bytes::value_type*>(bytes.get_write()), len));
});
});
}
future<uint64_t> read_unsigned_vint(random_access_reader& in) {
return read_vint_impl<uint64_t>(in);
}
future<int64_t> read_signed_vint(random_access_reader& in) {
return read_vint_impl<int64_t>(in);
}
} // namespace sstables
auto fmt::formatter<sstables::bound_kind_m>::format(sstables::bound_kind_m kind, fmt::format_context& ctx) const
-> decltype(ctx.out()) {
std::string_view name;
switch (kind) {
using enum sstables::bound_kind_m;
case excl_end:
name = "excl_end";
break;
case incl_start:
name = "incl_start";
break;
case excl_end_incl_start:
name = "excl_end_incl_start";
break;
case static_clustering:
name = "static_clustering";
break;
case clustering:
name = "clustering";
break;
case incl_end_excl_start:
name = "incl_end_excl_start";
break;
case incl_end:
name = "incl_end";
break;
case excl_start:
name = "excl_start";
break;
default:
return fmt::format_to(ctx.out(), "{}", underlying(kind));
}
return fmt::format_to(ctx.out(), "{}", name);
}