nghttpx: Set client_handler to special frontend objects

62411f5a · Tatsuhiro Tsujikawa · e9465f0e · 62411f5a
Commit 62411f5a authored 4 years ago by Tatsuhiro Tsujikawa
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 4 deletions

src/shrpx_client_handler.cc src/shrpx_client_handler.cc +10 -4

No files found.
--- a/src/shrpx_client_handler.cc
+++ b/src/shrpx_client_handler.cc
@@ -902,10 +902,16 @@ ClientHandler::get_downstream_connection(int &err, Downstream *downstream) {
  err = 0;
  switch (faddr_->alt_mode) {
-  case UpstreamAltMode::API:
+  case UpstreamAltMode::API: {
-    return std::make_unique<APIDownstreamConnection>(worker_);
+    auto dconn = std::make_unique<APIDownstreamConnection>(worker_);
-  case UpstreamAltMode::HEALTHMON:
+    dconn->set_client_handler(this);
-    return std::make_unique<HealthMonitorDownstreamConnection>();
+    return dconn;
+  }
+  case UpstreamAltMode::HEALTHMON: {
+    auto dconn = std::make_unique<HealthMonitorDownstreamConnection>();
+    dconn->set_client_handler(this);
+    return dconn;
+  }
  default:
    break;
  }