57 lines
1.5 KiB
C#
57 lines
1.5 KiB
C#
using System;
|
||
/// <summary>
|
||
/// Socket传输过程的缓冲区，尝试拆包获得数据
|
||
/// </summary>
|
||
public class DataBuffer
|
||
{
|
||
// 缓存区长度
|
||
private const int MIN_BUFF_LEN = 1024;
|
||
|
||
private byte[] _buff;
|
||
private int _buffLength = 0;
|
||
|
||
public DataBuffer(int minBuffLen = MIN_BUFF_LEN)
|
||
{
|
||
if (minBuffLen <= 0)
|
||
{
|
||
minBuffLen = MIN_BUFF_LEN;
|
||
}
|
||
_buff = new byte[minBuffLen];
|
||
}
|
||
|
||
/// <summary>
|
||
/// 添加缓存数据
|
||
/// </summary>
|
||
public void AddBuffer(byte[] data, int len)
|
||
{
|
||
byte[] buff = new byte[len];
|
||
Array.Copy(data, buff, len);
|
||
if (len > _buff.Length - _buffLength) //超过当前缓存
|
||
{
|
||
byte[] temp = new byte[_buffLength + len];
|
||
Array.Copy(_buff, 0, temp, 0, _buffLength);
|
||
Array.Copy(buff, 0, temp, _buffLength, len);
|
||
_buff = temp;
|
||
}
|
||
else
|
||
{
|
||
Array.Copy(data, 0, _buff, _buffLength, len);
|
||
}
|
||
_buffLength += len;//修改当前数据标记
|
||
}
|
||
|
||
public bool TryUnpack(out SocketDataPack dataPack)
|
||
{
|
||
dataPack = SocketDataPack.Unpack(_buff);
|
||
if (dataPack == null)
|
||
{
|
||
return false;
|
||
}
|
||
// 清理旧缓存
|
||
_buffLength -= dataPack.BuffLength;
|
||
byte[] temp = new byte[_buffLength < MIN_BUFF_LEN ? MIN_BUFF_LEN : _buffLength];
|
||
Array.Copy(_buff, dataPack.BuffLength, temp, 0, _buffLength);
|
||
_buff = temp;
|
||
return true;
|
||
}
|
||
} |