mirror of
https://github.com/mhx/dwarfs.git
synced 2025-08-04 02:06:22 -04:00
feat(byte_buffer): add append()
method
This commit is contained in:
parent
19ff269602
commit
cb52a2387b
@ -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_); }
|
||||
|
@ -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:
|
||||
|
@ -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");
|
||||
|
Loading…
x
Reference in New Issue
Block a user