Commit 2fa3d34a authored by Tatsuhiro Tsujikawa's avatar Tatsuhiro Tsujikawa

nghttpx: Use Connection::again_rt() in MemcachedConnection

parent fa3452ec
...@@ -42,6 +42,10 @@ void timeoutcb(struct ev_loop *loop, ev_timer *w, int revents) { ...@@ -42,6 +42,10 @@ void timeoutcb(struct ev_loop *loop, ev_timer *w, int revents) {
auto conn = static_cast<Connection *>(w->data); auto conn = static_cast<Connection *>(w->data);
auto mconn = static_cast<MemcachedConnection *>(conn->data); auto mconn = static_cast<MemcachedConnection *>(conn->data);
if (w == &conn->rt && !conn->expired_rt()) {
return;
}
if (LOG_ENABLED(INFO)) { if (LOG_ENABLED(INFO)) {
MCLOG(INFO, mconn) << "Time out"; MCLOG(INFO, mconn) << "Time out";
} }
...@@ -225,6 +229,8 @@ int MemcachedConnection::connected() { ...@@ -225,6 +229,8 @@ int MemcachedConnection::connected() {
ev_set_cb(&conn_.wev, writecb); ev_set_cb(&conn_.wev, writecb);
if (conn_.tls.ssl) { if (conn_.tls.ssl) {
conn_.again_rt();
do_read_ = &MemcachedConnection::tls_handshake; do_read_ = &MemcachedConnection::tls_handshake;
do_write_ = &MemcachedConnection::tls_handshake; do_write_ = &MemcachedConnection::tls_handshake;
...@@ -249,7 +255,7 @@ int MemcachedConnection::on_read() { return do_read_(*this); } ...@@ -249,7 +255,7 @@ int MemcachedConnection::on_read() { return do_read_(*this); }
int MemcachedConnection::tls_handshake() { int MemcachedConnection::tls_handshake() {
ERR_clear_error(); ERR_clear_error();
ev_timer_again(conn_.loop, &conn_.rt); conn_.last_read = ev_now(conn_.loop);
auto rv = conn_.tls_handshake(); auto rv = conn_.tls_handshake();
if (rv == SHRPX_ERR_INPROGRESS) { if (rv == SHRPX_ERR_INPROGRESS) {
...@@ -306,6 +312,8 @@ int MemcachedConnection::write_tls() { ...@@ -306,6 +312,8 @@ int MemcachedConnection::write_tls() {
return 0; return 0;
} }
conn_.last_read = ev_now(conn_.loop);
std::array<struct iovec, MAX_WR_IOVCNT> iov; std::array<struct iovec, MAX_WR_IOVCNT> iov;
std::array<uint8_t, 16_k> buf; std::array<uint8_t, 16_k> buf;
...@@ -343,9 +351,7 @@ int MemcachedConnection::read_tls() { ...@@ -343,9 +351,7 @@ int MemcachedConnection::read_tls() {
return 0; return 0;
} }
if (ev_is_active(&conn_.rt)) { conn_.last_read = ev_now(conn_.loop);
ev_timer_again(conn_.loop, &conn_.rt);
}
for (;;) { for (;;) {
auto nread = conn_.read_tls(recvbuf_.last, recvbuf_.wleft()); auto nread = conn_.read_tls(recvbuf_.last, recvbuf_.wleft());
...@@ -380,6 +386,8 @@ int MemcachedConnection::write_clear() { ...@@ -380,6 +386,8 @@ int MemcachedConnection::write_clear() {
return 0; return 0;
} }
conn_.last_read = ev_now(conn_.loop);
std::array<struct iovec, MAX_WR_IOVCNT> iov; std::array<struct iovec, MAX_WR_IOVCNT> iov;
for (; !sendq_.empty();) { for (; !sendq_.empty();) {
...@@ -406,9 +414,7 @@ int MemcachedConnection::read_clear() { ...@@ -406,9 +414,7 @@ int MemcachedConnection::read_clear() {
return 0; return 0;
} }
if (ev_is_active(&conn_.rt)) { conn_.last_read = ev_now(conn_.loop);
ev_timer_again(conn_.loop, &conn_.rt);
}
for (;;) { for (;;) {
auto nread = conn_.read_clear(recvbuf_.last, recvbuf_.wleft()); auto nread = conn_.read_clear(recvbuf_.last, recvbuf_.wleft());
...@@ -669,7 +675,7 @@ void MemcachedConnection::drain_send_queue(size_t nwrite) { ...@@ -669,7 +675,7 @@ void MemcachedConnection::drain_send_queue(size_t nwrite) {
if (recvq_.empty()) { if (recvq_.empty()) {
ev_timer_stop(conn_.loop, &conn_.rt); ev_timer_stop(conn_.loop, &conn_.rt);
} else if (!ev_is_active(&conn_.rt)) { } else if (!ev_is_active(&conn_.rt)) {
ev_timer_again(conn_.loop, &conn_.rt); conn_.again_rt();
} }
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment