lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-24 03:53:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

expand buffer faster

Browse Source
This commit is contained in:
Cloud Wu
2021-02-24 11:33:35 +08:00
parent 04dd921c5f
commit 6a4a2482fc
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
skynet-src/socket_server.c
View File

@@ -1381,9 +1381,10 @@ read_socket(struct socket *s, struct stream_buffer *buffer) {
int sz = s->p.size;
buffer->buf = NULL;
buffer->sz = 0;
int rsz = sz;
for (;;) {
char *buf = reserve_buffer(buffer, sz);
int n = (int)read(s->fd, buf, sz);
char *buf = reserve_buffer(buffer, rsz);
int n = (int)read(s->fd, buf, rsz);
if (n <= 0) {
if (buffer->sz == 0) {
// read nothing
@@ -1397,10 +1398,11 @@ read_socket(struct socket *s, struct stream_buffer *buffer) {
}
}
buffer->sz += n;
if (n < sz) {
if (n < rsz) {
break;
}
// n == sz, read again
// n == rsz, read again ( and read more )
rsz *= 2;
}
int r = buffer->sz;
if (r > sz) {