有两个线程，一个在读一个BUF，另一个在写这个BUF。
最开始的做法是这样的：
定义一个尾指针，每次写的时候都往尾指针里面写，读的时候就从头读，读完之后把剩下没读完的重新拷贝到头，并修改尾指针。
考虑到每次读的时候都需要使用到内存拷贝(memcpy)，于是想了办法，用环型内存来处理。
需要两个指针，一个叫小偷(thief_ptr)，另一个叫警察(police_ptr)，其中小偷扮演写的角色，警察则扮演读的角色。那么实际上就是在一个环型跑道上警察追小偷的过程，小偷永远在警察前面，但是不能超过警察一圈以上。
这里还有个细节问题需要处理，就是当小偷跑到起跑线的时候（内存边界）当步长超过余下的内存的时候，即
step_lenth > max_buffer_len - thief_ptr
也就是说要写入数据的长度大于剩下的空间的时候，这个时候，跑道的大小是可以改变的，我让这一圈跑道的长度设置为小偷在临界区时候的步长，这样就可以有效的解决这种问题。
经测试，该方法的效率比每次都memcpy的效率要高10%左右，由于是网络层引擎的比较核心的地方，所以10%的效率能带来非常大的改变。
以下是实现伪代码：
// 设置为双倍大小，有利于每一圈跑道的变长
unsigned char buf[BUF_SIZE * 2];
unsigned char *thief_ptr;
unsigned char *police_ptr;
int racetrack_len;
// 判断警察和小偷是否在同一圈内，界线是起跑线
bool InCatchCycle;
void Init()
{
// 警察和小偷在同一起跑线上
thief_ptr = police_ptr = buf;
// 警察和小偷一开始都在同一圈内
InCatchCycle = true;
}
bool CheckWriteable(int len)
{
//如果不在同一圈内，小偷就不能超过警察
if((!InCatchCycle && (police_ptr - thief_ptr < len)) )
return false;
return true;
}
int SendMsg(const unsigned char *msg, const int &len)
{
if(!CheckWriteable(len) || len > BUF_SIZE)
return 0;
// 此处要接收消息，无奈，只能使用memcpy
memcpy(thief_ptr, msg, len);
// 警察和小偷不在同一圈内
if(!InCatchCycle || (BUF_SIZE - (thief_ptr - buf) >= len))
{
//小偷跑len步长
thief_ptr += len;
return 1;
}
// 修正这一圈的长度
racetrack_len = (thief_ptr - WBuf) + len;
// 跑完一圈，回到终点
thief_ptr = buf;
// 小偷这个时候和警察肯定不在同一圈内
InCatchCycle = false;
return 1;
}
int ReadMsg()
{
int len;
//如果警察和小偷在同一圈内，则计算警察和小偷之间差多远
if(InCatchCycle) len = thief_ptr - police_ptr;
//否则则算警察离修正过的终点还差多远
else len = racetrack_len - (police_ptr - buf);
if(len <= 0) return 0;
int n;
// 因为很多因素，n大多数情况下不等于len
// 特别是在网络情况下。
n = get_msg(police_ptr, len);
//警察跑n步长
police_ptr += n;
if(n == len && !InCatchCycle){
//如果警察跑完一圈，回到起点
police_ptr = WBuf;
//这个时候警察和小偷肯定在同一圈了
InCatchCycle = true;
}
return n;
}
|
本类阅读TOP10