7215d4bfe9
these unused includes were identifier by clang-include-cleaner. after auditing these source files, all of the reports have been confirmed. please note, because quite a few source files relied on `utils/to_string.hh` to pull in the specialization of `fmt::formatter<std::optional<T>>`, after removing `#include <fmt/std.h>` from `utils/to_string.hh`, we have to include `fmt/std.h` directly. Signed-off-by: Kefu Chai <kefu.chai@scylladb.com>
137 lines
3.8 KiB
C++
137 lines
3.8 KiB
C++
/*
|
|
* Copyright (C) 2014-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
|
|
*/
|
|
|
|
#ifndef UTILS_DATA_INPUT_HH_
|
|
#define UTILS_DATA_INPUT_HH_
|
|
|
|
#include "bytes_fwd.hh"
|
|
#include <seastar/net/byteorder.hh>
|
|
#include <concepts>
|
|
|
|
class data_input {
|
|
public:
|
|
data_input(const bytes_view& v)
|
|
: _view(v) {
|
|
}
|
|
data_input(bytes_view&& v)
|
|
: _view(std::move(v)) {
|
|
}
|
|
data_input(const bytes& b)
|
|
: data_input(bytes_view(b)) {
|
|
}
|
|
data_input(const bytes& b, size_t off, size_t n = bytes::npos)
|
|
: data_input(
|
|
bytes_view(b.c_str() + off, std::min(b.size() - off, n))) {
|
|
if (off > b.size()) {
|
|
throw std::out_of_range("Offset out of range");
|
|
}
|
|
}
|
|
template<typename T>
|
|
data_input(const std::basic_string_view<T>& view)
|
|
: data_input(
|
|
bytes_view(reinterpret_cast<const int8_t *>(view.data()),
|
|
view.size() * sizeof(T))) {
|
|
}
|
|
template<typename T>
|
|
data_input(const temporary_buffer<T>& buf)
|
|
: data_input(
|
|
std::basic_string_view<T>(buf.get(), buf.size())) {
|
|
}
|
|
data_input(data_input&&) = default;
|
|
data_input(const data_input&) = default;
|
|
|
|
size_t avail() const {
|
|
return _view.size();
|
|
}
|
|
bool has_next() const {
|
|
return !_view.empty();
|
|
}
|
|
void ensure(size_t s) const {
|
|
if (avail() < s) {
|
|
throw std::out_of_range("Buffer underflow");
|
|
}
|
|
}
|
|
template<typename T> T peek() const;
|
|
template<typename T> T read();
|
|
|
|
bytes_view read_view(size_t len) {
|
|
ensure(len);
|
|
bytes_view v(_view.begin(), len);
|
|
_view.remove_prefix(len);
|
|
return v;
|
|
}
|
|
|
|
template <typename SizeType>
|
|
bytes_view read_view_to_blob() {
|
|
auto len = read<SizeType>();
|
|
return read_view(len);
|
|
}
|
|
|
|
void skip(size_t s) {
|
|
ensure(s);
|
|
_view.remove_prefix(s);
|
|
}
|
|
private:
|
|
template<typename T> size_t ssize(const T &) const;
|
|
template<std::integral T>
|
|
inline T peek_primitive() const {
|
|
ensure(sizeof(T));
|
|
T t;
|
|
std::copy_n(_view.begin(), sizeof(T), reinterpret_cast<char *>(&t));
|
|
return net::ntoh(t);
|
|
}
|
|
|
|
bytes_view _view;
|
|
};
|
|
|
|
template<> inline sstring data_input::peek<sstring>() const {
|
|
auto len = peek<uint16_t>();
|
|
ensure(sizeof(uint16_t) + len);
|
|
return sstring(reinterpret_cast<const char*>(_view.data()) + sizeof(uint16_t), len);
|
|
}
|
|
template<> inline size_t data_input::ssize<sstring>(const sstring & s) const {
|
|
return sizeof(uint16_t) + s.size();
|
|
}
|
|
template<> inline bytes data_input::peek<bytes>() const {
|
|
auto len = peek<uint32_t>();
|
|
ensure(sizeof(uint32_t) + len);
|
|
return bytes(_view.data() + sizeof(uint32_t), len);
|
|
}
|
|
template<> inline size_t data_input::ssize<bytes>(const bytes & s) const {
|
|
return sizeof(uint32_t) + s.size();
|
|
}
|
|
template<> inline bytes_view data_input::peek<bytes_view>() const {
|
|
auto len = peek<uint32_t>();
|
|
ensure(sizeof(uint32_t) + len);
|
|
return bytes_view(_view.data() + sizeof(uint32_t), len);
|
|
}
|
|
template<> inline size_t data_input::ssize<bytes_view>(const bytes_view& v) const {
|
|
return sizeof(uint32_t) + v.size();
|
|
}
|
|
template<> inline size_t data_input::ssize(const bool &) const {
|
|
return sizeof(uint8_t);
|
|
}
|
|
template<> inline bool data_input::peek<bool>() const {
|
|
return peek<uint8_t>() != 0;
|
|
}
|
|
template<typename T>
|
|
inline T data_input::peek() const {
|
|
return peek_primitive<T>();
|
|
}
|
|
template<typename T> inline T data_input::read() {
|
|
auto t = peek<T>();
|
|
_view.remove_prefix(ssize(t));
|
|
return std::move(t);
|
|
}
|
|
template<typename T> inline size_t data_input::ssize(const T &) const {
|
|
return sizeof(T);
|
|
}
|
|
|
|
#endif /* UTILS_DATA_INPUT_HH_ */
|
|
|