/*
+----------------------------------------------------------------------+
| Swoole |
+----------------------------------------------------------------------+
| This source file is subject to version 2.0 of the Apache license, |
| that is bundled with this package in the file LICENSE, and is |
| available through the world-wide-web at the following url: |
| http://www.apache.org/licenses/LICENSE-2.0.html |
| If you did not receive a copy of the Apache2.0 license and are unable|
| to obtain it through the world-wide-web, please send a note to |
| license@php.net so we can mail you a copy immediately. |
+----------------------------------------------------------------------+
| Author: Tianfeng Han <rango@swoole.com> |
+----------------------------------------------------------------------+
*/
#pragma once
#include <queue>
#include <sys/uio.h>
namespace swoole {
struct BufferChunk {
enum Type {
TYPE_DATA,
TYPE_SENDFILE,
TYPE_CLOSE,
};
Type type;
uint32_t length = 0;
uint32_t offset = 0;
union {
char *ptr;
void *object;
struct {
uint32_t val1;
uint32_t val2;
} data;
} value{};
uint32_t size = 0;
BufferChunk(Type type, uint32_t size);
~BufferChunk();
void (*destroy)(BufferChunk *chunk) = nullptr;
};
class Buffer {
private:
// 0: donot use chunk
uint32_t chunk_size;
uint32_t total_length = 0;
std::queue<BufferChunk *> queue_;
public:
explicit Buffer(uint32_t chunk_size);
~Buffer();
BufferChunk *alloc(BufferChunk::Type type, uint32_t size);
BufferChunk *front() {
return queue_.front();
}
void pop();
void append(const void *data, uint32_t size);
void append(const struct iovec *iov, size_t iovcnt, off_t offset);
uint32_t length() const {
return total_length;
}
size_t count() const {
return queue_.size();
}
bool empty() const {
return queue_.empty();
}
static bool empty(Buffer *buffer) {
return buffer == nullptr || buffer->queue_.empty();
}
};
} // namespace swoole
|