有两个线程,一个在读一个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; }

|