nghttpx: Refactor graceful shutdown in Http2Upstream

Instead of using bool flag, just stop prepare watcher.
This commit is contained in:
Tatsuhiro Tsujikawa 2016-06-04 12:43:17 +09:00
parent 9237d30e34
commit d0bf247419
2 changed files with 16 additions and 17 deletions

View File

@ -787,23 +787,25 @@ void Http2Upstream::submit_goaway() {
void Http2Upstream::check_shutdown() {
int rv;
if (shutdown_handled_) {
return;
}
auto worker = handler_->get_worker();
if (worker->get_graceful_shutdown()) {
shutdown_handled_ = true;
rv = nghttp2_submit_shutdown_notice(session_);
if (rv != 0) {
ULOG(FATAL, this) << "nghttp2_submit_shutdown_notice() failed: "
<< nghttp2_strerror(rv);
return;
}
handler_->signal_write();
ev_timer_start(handler_->get_loop(), &shutdown_timer_);
if (!worker->get_graceful_shutdown()) {
return;
}
ev_prepare_stop(handler_->get_loop(), &prep_);
rv = nghttp2_submit_shutdown_notice(session_);
if (rv != 0) {
ULOG(FATAL, this) << "nghttp2_submit_shutdown_notice() failed: "
<< nghttp2_strerror(rv);
return;
}
handler_->signal_write();
ev_timer_start(handler_->get_loop(), &shutdown_timer_);
}
nghttp2_session_callbacks *create_http2_upstream_callbacks() {
@ -870,9 +872,7 @@ Http2Upstream::Http2Upstream(ClientHandler *handler)
downstream_queue_(downstream_queue_size(handler->get_worker()),
!get_config()->http2_proxy),
handler_(handler),
session_(nullptr),
shutdown_handled_(false) {
session_(nullptr) {
int rv;
auto &http2conf = get_config()->http2;

View File

@ -132,7 +132,6 @@ private:
ClientHandler *handler_;
nghttp2_session *session_;
bool flow_control_;
bool shutdown_handled_;
};
nghttp2_session_callbacks *create_http2_upstream_callbacks();