api: add context to ready func

api.go

@@ -8,6 +8,7 @@ package forwarder
 
 import (
 	"bytes"
+	"context"
 	"encoding/json"
 	"net/http"
 	"net/http/pprof"
@@ -24,14 +25,14 @@ import (
 // It provides health and readiness endpoints prometheus metrics, and pprof debug endpoints.
 type APIHandler struct {
 	mux    *http.ServeMux
-	ready  func() bool
+	ready  func(ctx context.Context) bool
 	config string
 	script string
 
 	patterns []string
 }
 
-func NewAPIHandler(r prometheus.Gatherer, ready func() bool, config, pac string) *APIHandler {
+func NewAPIHandler(r prometheus.Gatherer, ready func(ctx context.Context) bool, config, pac string) *APIHandler {
 	m := http.NewServeMux()
 	a := &APIHandler{
 		mux:    m,
@@ -72,7 +73,7 @@ func (h *APIHandler) healthz(w http.ResponseWriter, r *http.Request) {
 }
 
 func (h *APIHandler) readyz(w http.ResponseWriter, r *http.Request) {
-	if h.ready() {
+	if h.ready(r.Context()) {
 		w.WriteHeader(http.StatusOK)
 		w.Header().Set("Content-Type", "text/plain")
 		w.Write([]byte("OK"))

http_proxy.go

@@ -509,6 +509,6 @@ func (hp *HTTPProxy) Addr() string {
 }
 
 // Ready returns true if the server is running and ready to accept requests.
-func (hp *HTTPProxy) Ready() bool {
+func (hp *HTTPProxy) Ready(ctx context.Context) bool {
 	return hp.Addr() != ""
 }

http_server.go

@@ -264,6 +264,6 @@ func (hs *HTTPServer) Addr() string {
 }
 
 // Ready returns true if the server is running and ready to accept requests.
-func (hs *HTTPServer) Ready() bool {
+func (hs *HTTPServer) Ready(ctx context.Context) bool {
 	return hs.Addr() != ""
 }