dnsdist: Count the number of concurrent connections for DoH as well

diff --git a/pdns/dnsdistdist/doh.cc b/pdns/dnsdistdist/doh.cc
index 2f3cd457ecca72cef8300262c072b2e12454dfc3..d0ea42dcf422b92beae0674c0103bd4637d1d806 100644 (file)
--- a/pdns/dnsdistdist/doh.cc
+++ b/pdns/dnsdistdist/doh.cc
@@ -82,7 +82,15 @@ public:
     }
   }
 
+  void decrementConcurrentConnections()
+  {
+    if (d_cs != nullptr) {
+      --d_cs->tcpCurrentConnections;
+    }
+  }
+
   std::map<int, std::string> d_ocspResponses;
+  ClientState* d_cs{nullptr};
 
 private:
   h2o_accept_ctx_t d_h2o_accept_ctx;
@@ -146,6 +154,7 @@ void handleDOHTimeout(DOHUnit* oldDU)
 static void on_socketclose(void *data)
 {
   DOHAcceptContext* ctx = reinterpret_cast<DOHAcceptContext*>(data);
+  ctx->decrementConcurrentConnections();
   ctx->release();
 }
 
@@ -883,6 +892,7 @@ static void on_accept(h2o_socket_t *listener, const char *err)
   auto accept_ctx = dsc->accept_ctx->get();
   sock->on_close.data = dsc->accept_ctx;
   ++dsc->df->d_httpconnects;
+  ++dsc->cs->tcpCurrentConnections;
   h2o_accept(accept_ctx, sock);
 }
 
@@ -1034,7 +1044,7 @@ static void setupAcceptContext(DOHAcceptContext& ctx, DOHServerConfig& dsc, bool
 
     nativeCtx->ssl_ctx = tlsCtx.release();
   }
-
+  ctx.d_cs = dsc.cs;
   ctx.release();
 }