package ping import ( "context" "fmt" "net/http" "github.com/gorilla/mux" ) // Handler expose ping routes. type Handler struct { terminating bool } // SetDefaults sets the default values. func (h *Handler) SetDefaults() { } // WithContext causes the ping endpoint to serve non 200 responses. func (h *Handler) WithContext(ctx context.Context) { go func() { <-ctx.Done() h.terminating = true }() } // Append adds ping routes on a router. func (h *Handler) Append(router *mux.Router) { router.Methods(http.MethodGet, http.MethodHead).Path("/ping"). HandlerFunc(func(response http.ResponseWriter, request *http.Request) { statusCode := http.StatusOK if h.terminating { statusCode = http.StatusServiceUnavailable } response.WriteHeader(statusCode) fmt.Fprint(response, http.StatusText(statusCode)) }) }