tokyo tyrant源码分析-编程小技巧

----------------

1、内存分配

----------------

 

char stack[TTIOBUFSIZ]; //65536

char *buf = (rsiz < TTIOBUFSIZ) ? stack : tcmalloc(rsiz + 1);

 

...

free((buf == stack) ? NULL : buf);

 

rsiz是运行时才能确定的值。这种方法的思想是尽可能使用静态分配，因为动态分配有损效率。因为rsiz的未知性，它不可能也不必保证100%是静态分配，它只须保证一个高的命中率即可。你也可以使用预分配和内存池，但那意味着更复杂的编码和管理。这种做法在好几个方面都是一种折中(静态分配空间越大，命中率越高，同时可能造成的空间浪费也越高；用内存池可能性能更高，但编码会更复杂...)。

tyrant中大量使用到了这种小技巧。tcpip bsd实现里也用到了这种小技巧。

 

 

------------------

2、sock io封装

------------------

底层封装：

 

int ttsockgetc(TTSOCK *sock){ /* return a char */ assert(sock); if(sock->rp < sock->ep) return *(unsigned char *)(sock->rp++); /* buf not empty */ do { int rv = recv(sock->fd, sock->buf, TTIOBUFSIZ, 0); /* 65536 */ if(rv > 0){ sock->rp = sock->buf + 1; sock->ep = sock->buf + rv; return *(unsigned char *)sock->buf; } else if(rv == 0){ /* the other peer shutdown */ sock->end = true; return -1; } } while((errno == EINTR || errno == EAGAIN || errno == EWOULDBLOCK) && tctime() <= sock->dl); /* if interrupted */ sock->end = true; return -1; } 

它使用自己的buffer,一次recv大块数据(最多65536字节)放入buffer。每次对外返回一个char，buffer为空时才再次recv。

 

------------------------

各种提供给应用的封装：

 

返回指定size：

bool ttsockrecv(TTSOCK *sock, char *buf, int size){ assert(sock && buf && size >= 0); if(sock->rp + size <= sock->ep){ /* enough data in buffer */ memcpy(buf, sock->rp, size); sock->rp += size; return true; } bool err = false; char *wp = buf; while(size > 0){ /* interesting */ int c = ttsockgetc(sock); if(c == -1){ err = true; break; } *(wp++) = c; size--; } return !err; } 

 

返回64bits:

uint64_t ttsockgetint64(TTSOCK *sock){ assert(sock); uint64_t num; ttsockrecv(sock, (char *)&num, sizeof(num)); return TTNTOHLL(num); } 

 

等等... 还有一些类似的封装。

 

封装的比较漂亮，使用和阅读都很便利。ttsockrecv里过多的函数调用开销可能会给性能带来一定损耗。但其实事实上不一定，因为还有一个命中率的问题，分析可以知道执行下面那个while循环的情况是较少的。而且如果再增加一些代码复杂度，可以把它封装的更漂亮。