Buffer模块是一个缓冲区模块,用于实现通信中用户态的接收缓冲区和发送缓冲区功能。
存储数据,取出数据
1.实现换出去得有一块内存空间,采用vector ,vector底层是一个线性的内存空间!
1.默认空间大小
2.当前的读取数据位置!
3.当前的写入数据位置!
class buffer {
private:
std::vector<char> _buffer;
// 位置是一个相对偏移量,而不是绝对地址!
uint64_t _read_idx; // 读位置
uint64_t _write_idx; // 写位置
public:
1. 获取当前写的位置
2. 确保可写空间足够
3. 获取前沿空间大小
4. 获取后沿空间大小
5. 将写入据位置向后移动指定长度
6. 获取当前读取位置的地址!
7. 获取可读空间大小
8. 将读位置向后移动指定长度!
9. clear
#include
#include
#include
#include
#include
#include
using namespace std;
#define BUFFER_SIZE 1024
class Buffer {
private:
std::vector<char> _buffer; // 使用vector进行内存空间管理
uint64_t _read_idx; // 读偏移
uint64_t _write_idx; // 写偏移
public:
Buffer():_read_idx(0),_write_idx(0),_buffer(BUFFER_SIZE) {}
char* begin() {return &*_buffer.begin();}
// 获取当前写入起始地址
char *writePosition() { return begin() + _write_idx;}
// 获取当前读取起始地址
char *readPosition() { return begin() + _read_idx; }
// 获取缓冲区末尾空间大小 —— 写偏移之后的空闲空间,总体大小减去写偏移
uint64_t tailIdleSize() {return _buffer.size() - _write_idx; }
// 获取缓冲区起始空间大小 —— 读偏移之前的空闲空间
uint64_t handIdleSize() {return _read_idx ;}
// 获取可读空间大小 = 写偏移 - 读偏移
uint64_t readAbleSize() {return _write_idx - _read_idx ;}
// 将读偏移向后移动
void moveReadOffset(uint64_t len) {
// 向后移动大小必须小于可读数据大小
assert(len <= readAbleSize());
_read_idx += len;
}
// 将写偏移向后移动
void moveWriteOffset(uint64_t len) {
assert(len <= tailIdleSize());
_write_idx += len;
}
void ensureWriteSpace(uint64_t len) {
// 确保可写空间足够 (整体空间够了就移动数据,否则就扩容!)
if (tailIdleSize() >= len) return;
// 不够的话 ,判断加上起始位置够不够,够了将数据移动到起始位置
if (len <= tailIdleSize() + handIdleSize()) {
uint64_t rsz = readAbleSize(); //帮当前数据大小先保存起来
std::copy(readPosition(),readPosition() + rsz,begin()); // 把可读数据拷贝到起始位置
_read_idx = 0; // 读归为0
_write_idx = rsz; // 可读数据大小是写的偏移量!
}
else { // 总体空间不够!需要扩容,不移动数据,直接给写偏移之后扩容足够空间即可!
_buffer.resize(_write_idx + len);
}
}
// 写入数据
void Write(const void *data,uint64_t len) {
ensureWriteSpace(len);
const char *d = (const char*) data;
std::copy(d,d + len,writePosition());
}
void WriteAndPush(void* data,uint64_t len) {
Write(data,len);
moveWriteOffset(len);
}
void WriteStringAndPush(const std::string &data) {
writeString(data);
moveWriteOffset(data.size());
}
void writeString(const std::string &data) {
return Write(data.c_str(),data.size());
}
void writeBuffer(Buffer &data) {
return Write(data.readPosition(),data.readAbleSize());
}
void writeBufferAndPush(Buffer &data) {
writeBuffer(data);
moveWriteOffset(data.readAbleSize());
}
std::string readAsString (uint64_t len) {
assert(len <= readAbleSize());
std::string str;
str.resize(len);
Read(&str[0],len);
return str;
}
void Read(void *buf,uint64_t len) {
// 读取数据 1. 保证足够的空间 2.拷贝数据进去
// 要求获取的大小必须小于可读数据大小!
assert(len <= readAbleSize());
std::copy(readPosition(),readPosition() + len,(char*)buf);
}
void readAndPop(void *buf,uint64_t len) {
Read(buf,len);
moveReadOffset(len);
}
// 逐步调试!!!!!
std::string ReadAsStringAndPop(uint64_t len) {
assert(len <= readAbleSize());
std::string str = readAsString(len);
moveReadOffset(len);
return str;
}
char* FindCRLF() {
char *res = (char*)memchr(readPosition(),'\n',readAbleSize());
return res;
}
// 通常获取一行数据,这种情况针对是:
std::string getLine() {
char* pos = FindCRLF();
if (pos == NULL) {
return "";
}
// +1 为了把换行数据取出来!
return readAsString(pos - readPosition() + 1);
}
std::string getLineAndPop() {
std::string str = getLine();
moveReadOffset(str.size());
return str;
}
void Clear() { // 清空缓冲区!clear
// 只需要将偏移量归0即可!
_read_idx = 0;
_write_idx = 0;
}
};
#include "server.hpp"
using namespace std;
// 控制打印信息!!!
#define INF 0
#define DBG 1
#define ERR 2
#define LOG_LEVEL INF
#define LOG(level,format,...) do{\
if (level < LOG_LEVEL) break;\
time_t t = time(NULL);\
struct tm *ltm = localtime(&t);\
char tmp[23] = {0};\
strftime(tmp,31,"%H:%M:%S",ltm);\
fprintf(stdout,"[%s,%s:%d] " format "\n",tmp,__FILE__,__LINE__,##__VA_ARGS__);\
}while(0)
#define INF_LOG(format, ...) LOG(INF, format, ##__VA_ARGS__)
#define DBG_LOG(format, ...) LOG(DBG, format, ##__VA_ARGS__)
#define ERR_LOG(format, ...) LOG(ERR, format, ##__VA_ARGS__)
int main() {
Buffer buf;
std::string str = "hello!";
// buf.WriteStringAndPush(str);
// Buffer buf1;
// buf1.writeBufferAndPush(buf);
// std::string tmp;
// tmp = buf1.ReadAsStringAndPop(buf.readAbleSize());
// cout << tmp << endl;
// cout << buf.readAbleSize() << endl;
// cout << buf1.readAbleSize() << endl;
for (int i = 0; i < 300; i ++) {
std::string str = "hello" + std::to_string(i) + '\n';
buf.WriteStringAndPush(str);
}
while(buf.readAbleSize() > 0) {
string line = buf.getLineAndPop();
LOG("hello");
}
// string tmp;
// tmp = buf.ReadAsStringAndPop(buf.readAbleSize());
// cout << tmp << endl;
return 0;
}
中秋快乐!