use different handler for healthcheck

knative · Oct 27, 2021 · b1b343a · b1b343a
1 parent be57d2d
commit b1b343a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cmd/queue/main.go b/cmd/queue/main.go
@@ -313,8 +313,14 @@ func buildServer(ctx context.Context, env config, drainer *pkghandler.Drainer, p
 		composedHandler = requestLogHandler(logger, composedHandler, env)
 	}
 
+	var healthcheckHandler http.Handler
+	healthcheckHandler = health.ProbeHandler(probeContainer, tracingEnabled, healthcheckHandler)
+	if env.ServingEnableRequestLog {
+		healthcheckHandler = requestLogHandler(logger, healthcheckHandler, env)
+	}
+
 	drainer.Inner = composedHandler
-	drainer.HealthCheck = composedHandler.ServeHTTP
+	drainer.HealthCheck = healthcheckHandler.ServeHTTP
 
 	return pkgnet.NewServer(":"+env.QueueServingPort, drainer)
 }