nghttp2/nghttp2
1
0
Fork 0
mirror of https://github.com/nghttp2/nghttp2.git synced 2025-12-07 02:28:53 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #2382 from nghttp2/buffer-ranges

Browse Source 
buffer: Adopt std::ranges
This commit is contained in:
Tatsuhiro Tsujikawa
2025-05-17 19:39:03 +09:00
committed by GitHub
parent 10ebb3825a 09bb8350e8
commit fef9e8fb20
1 changed files with 15 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/buffer.h
View File

@@ -27,49 +27,46 @@
#include "nghttp2_config.h"
#include <cstring>
#include <algorithm>
#include <array>
namespace nghttp2 {
template <size_t N> struct Buffer {
Buffer() : pos(std::begin(buf)), last(pos) {}
constexpr Buffer() noexcept : pos(buf), last(pos) {}
// Returns the number of bytes to read.
size_t rleft() const { return last - pos; }
constexpr size_t rleft() const noexcept { return last - pos; }
// Returns the number of bytes this buffer can store.
size_t wleft() const { return std::end(buf) - last; }
constexpr size_t wleft() const noexcept { return &buf[N] - last; }
// Writes up to min(wleft(), |count|) bytes from buffer pointed by
// |src|. Returns number of bytes written.
size_t write(const void *src, size_t count) {
constexpr size_t write(const void *src, size_t count) {
count = std::min(count, wleft());
auto p = static_cast<const uint8_t *>(src);
last = std::copy_n(p, count, last);
last = std::ranges::copy_n(p, count, last).out;
return count;
}
size_t write(size_t count) {
constexpr size_t write(size_t count) {
count = std::min(count, wleft());
last += count;
return count;
}
// Drains min(rleft(), |count|) bytes from start of the buffer.
size_t drain(size_t count) {
constexpr size_t drain(size_t count) {
count = std::min(count, rleft());
pos += count;
return count;
}
size_t drain_reset(size_t count) {
constexpr size_t drain_reset(size_t count) {
count = std::min(count, rleft());
std::copy(pos + count, last, std::begin(buf));
last = std::begin(buf) + (last - (pos + count));
pos = std::begin(buf);
last = std::ranges::copy(pos + count, last, buf).out;
pos = buf;
return count;
}
void reset() { pos = last = std::begin(buf); }
uint8_t *begin() { return std::begin(buf); }
uint8_t &operator[](size_t n) { return buf[n]; }
const uint8_t &operator[](size_t n) const { return buf[n]; }
std::array<uint8_t, N> buf;
constexpr void reset() noexcept { pos = last = buf; }
constexpr uint8_t *begin() noexcept { return buf; }
constexpr uint8_t &operator[](size_t n) { return buf[n]; }
constexpr const uint8_t &operator[](size_t n) const { return buf[n]; }
uint8_t buf[N];
uint8_t *pos, *last;
};