38 lines
1.2 KiB
Go
38 lines
1.2 KiB
Go
package ping
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
)
|
|
|
|
// Handler expose ping routes.
|
|
type Handler struct {
|
|
EntryPoint string `description:"EntryPoint" export:"true" json:"entryPoint,omitempty" toml:"entryPoint,omitempty" yaml:"entryPoint,omitempty"`
|
|
ManualRouting bool `description:"Manual routing" json:"manualRouting,omitempty" toml:"manualRouting,omitempty" yaml:"manualRouting,omitempty"`
|
|
TerminatingStatusCode int `description:"Terminating status code" json:"terminatingStatusCode,omitempty" toml:"terminatingStatusCode,omitempty" yaml:"terminatingStatusCode,omitempty"`
|
|
terminating bool
|
|
}
|
|
|
|
// SetDefaults sets the default values.
|
|
func (h *Handler) SetDefaults() {
|
|
h.EntryPoint = "traefik"
|
|
h.TerminatingStatusCode = http.StatusServiceUnavailable
|
|
}
|
|
|
|
// WithContext causes the ping endpoint to serve non 200 responses.
|
|
func (h *Handler) WithContext(ctx context.Context) {
|
|
go func() {
|
|
<-ctx.Done()
|
|
h.terminating = true
|
|
}()
|
|
}
|
|
|
|
func (h *Handler) ServeHTTP(response http.ResponseWriter, request *http.Request) {
|
|
statusCode := http.StatusOK
|
|
if h.terminating {
|
|
statusCode = h.TerminatingStatusCode
|
|
}
|
|
response.WriteHeader(statusCode)
|
|
fmt.Fprint(response, http.StatusText(statusCode))
|
|
}
|