C++ 字节流 CByteBuffer

最新推荐文章于 2024-05-10 16:25:13 发布

转载最新推荐文章于 2024-05-10 16:25:13 发布 · 1.1k 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/JsonBlog/p/4153431.html

文章标签：

#c/c++ #网络

本文详细介绍了CByteBuffer类的实现及其如何用于高效地管理内存，包括如何将基本类型的数据写入缓冲区以及如何自动调整缓冲区大小以适应数据增长。

CByteBuffer使用

static void Main()
{
    BYTE a;
    short b;
    int c;
    CByteBuffer buffer;
    buffer<<a<<b<<c;
}


class CByteBuffer
{
public:
    BYTE* m_lpBuf;
    int nCurrentIndex;
    int nCapacity;
    int nLen;
    CByteBuffer()
    {
        Init(128);
    }

    CByteBuffer(int nCapacityIn)
    {
        Init(nCapacityIn);
    }

    ~CByteBuffer()
    {
        delete [] m_lpBuf;
        m_lpBuf=NULL;
    }

    void GetMoreCapacity()
    {
        BYTE *pBytesMore=new BYTE[nCapacity*2]; //两倍
        memcpy(pBytesMore,m_lpBuf,nCapacity*sizeof(BYTE));
        memset(pBytesMore+nCapacity,0x00,nCapacity*sizeof(BYTE));
        delete [] m_lpBuf;
        m_lpBuf=pBytesMore;
        nCapacity*=2;
    }
    
    template<typename T>
    CByteBuffer& operator <<(T dw)
    {
        int nSizeNew=nCurrentIndex + sizeof(T);
        while(nSizeNew>nCapacity)
        {
            GetMoreCapacity(); //缓冲区空间不够
        }
        *(T*)(&m_lpBuf[nCurrentIndex])=dw; //添入缓冲区
        nCurrentIndex += sizeof(T);     //移动当前索引
        nLen+=sizeof(T);
        return *this;
    }

private:
    void Init(int nCapatityIn)
    {
        nCapacity=nCapatityIn;
        m_lpBuf=new BYTE[nCapatityIn];
        memset(m_lpBuf,0x00,nCapacity);
        nCurrentIndex=0;
        nLen=0;
    }
};