feat(byte_buffer): add append() method

This commit is contained in:
Marcus Holland-Moritz 2025-04-09 14:30:52 +02:00
parent 19ff269602
commit cb52a2387b
3 changed files with 25 additions and 0 deletions

View File

@ -78,6 +78,8 @@ class mutable_byte_buffer_interface : public byte_buffer_interface {
// that would reallocate the buffer will throw.
virtual void freeze_location() = 0;
virtual void append(void const* data, size_t size) = 0;
virtual internal::malloc_buffer& raw_buffer() = 0;
};
@ -168,6 +170,13 @@ class mutable_byte_buffer {
void freeze_location() { bb_->freeze_location(); }
void append(void const* data, size_t size) { bb_->append(data, size); }
template <detail::byte_range T>
void append(T const& data) {
append(data.data(), data.size());
}
internal::malloc_buffer& raw_buffer() { return bb_->raw_buffer(); }
void swap(mutable_byte_buffer& other) noexcept { std::swap(bb_, other.bb_); }

View File

@ -94,6 +94,13 @@ class malloc_byte_buffer_impl : public mutable_byte_buffer_interface {
frozen_.store(true, std::memory_order_release);
}
void append(void const* data, size_t size) override {
if (frozen() && data_.size() + size > data_.capacity()) {
frozen_error("append beyond capacity");
}
data_.append(data, size);
}
internal::malloc_buffer& raw_buffer() override { return data_; }
private:

View File

@ -27,6 +27,7 @@
*/
#include <cassert>
#include <cstring>
#include <stdexcept>
#include <system_error>
@ -138,6 +139,14 @@ class mmap_byte_buffer_impl : public mutable_byte_buffer_interface {
// always frozen
}
void append(void const* data, size_t size) override {
if (size_ + size > data_.size()) {
frozen_error("append beyond capacity");
}
std::memcpy(data_.data() + size_, data, size);
size_ += size;
}
internal::malloc_buffer& raw_buffer() override {
throw std::runtime_error(
"operation not allowed on mmap buffer: raw_buffer");