walkero
/
nghttp2
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

Optimize QUIC write

This commit is contained in:
Tatsuhiro Tsujikawa 2019-11-20 23:40:51 +09:00
parent 1d05c6c3c5
commit aa7c580bb1
2 changed files with 24 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/h2load.cc
View File

@ -214,8 +214,7 @@ void readcb(struct ev_loop *loop, ev_io *w, int revents) {
delete client; delete client;
return; return;
} }
writecb(loop, &client->wev, revents); client->signal_write();
// client->disconnect() and client->fail() may be called
} }
} // namespace } // namespace

9
src/h2load_quic.cc
View File

@ -601,7 +601,9 @@ int Client::read_quic() {
sockaddr_union su; sockaddr_union su;
socklen_t addrlen = sizeof(su); socklen_t addrlen = sizeof(su);
int rv; int rv;
size_t pktcnt = 0;
for (;;) {
auto nread = auto nread =
recvfrom(fd, buf.data(), buf.size(), MSG_DONTWAIT, &su.sa, &addrlen); recvfrom(fd, buf.data(), buf.size(), MSG_DONTWAIT, &su.sa, &addrlen);
if (nread == -1) { if (nread == -1) {
@ -622,14 +624,17 @@ int Client::read_quic() {
return -1; return -1;
} }
if (worker->current_phase == Phase::MAIN_DURATION) { if (pktcnt == 10) {
worker->stats.bytes_total += nread; break;
}
} }
return 0; return 0;
} }
int Client::write_quic() { int Client::write_quic() {
ev_io_stop(worker->loop, &wev);
if (quic.close_requested) { if (quic.close_requested) {
return -1; return -1;
} }