traefik/pkg/ping/ping.go

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))
		})
}