Rooba
/
odyssey
mirror of https://github.com/yandex/odyssey.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

odissey: limit stream fill by readahead size

This commit is contained in:
Dmitry Simonenko 2018-01-19 16:46:47 +03:00
parent 15761a1186
commit c810d77591
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
sources/frontend.c
View File

@ -427,8 +427,15 @@ od_frontend_setup(od_client_t *client)
return OD_FE_OK;
}
static inline int
od_frontend_stream_hit_limit(od_client_t *client)
{
od_instance_t *instance = client->system->instance;
return shapito_stream_used(&client->stream) >= instance->scheme.readahead;
}
static inline void
od_frontend_reset_stream(od_client_t *client)
od_frontend_stream_reset(od_client_t *client)
{
od_instance_t *instance = client->system->instance;
shapito_stream_t *stream = &client->stream;
@ -506,7 +513,7 @@ od_frontend_remote_client(od_client_t *client)
int request_count = 0;
int terminate = 0;
od_frontend_reset_stream(client);
od_frontend_stream_reset(client);
int rc;
for (;;)
{
@ -571,6 +578,10 @@ od_frontend_remote_client(od_client_t *client)
request_count++;
}
rc = od_frontend_stream_hit_limit(client);
if (rc)
break;
rc = machine_read_pending(client->io);
if (rc < 0 || rc > 0)
continue;
@ -605,7 +616,7 @@ od_frontend_remote_server(od_client_t *client)
shapito_stream_t *stream = &client->stream;
od_server_t *server = client->server;
od_frontend_reset_stream(client);
od_frontend_stream_reset(client);
int rc;
for (;;)
{
@ -692,6 +703,10 @@ od_frontend_remote_server(od_client_t *client)
break;
}
rc = od_frontend_stream_hit_limit(client);
if (rc)
break;
rc = machine_read_pending(server->io);
if (rc < 0 || rc > 0)
continue;