package healthcheck import ( "errors" "fmt" "net/http" "os" "time" "github.com/containous/traefik/v2/pkg/config/static" "github.com/traefik/paerser/cli" ) // NewCmd builds a new HealthCheck command. func NewCmd(traefikConfiguration *static.Configuration, loaders []cli.ResourceLoader) *cli.Command { return &cli.Command{ Name: "healthcheck", Description: `Calls Traefik /ping endpoint (disabled by default) to check the health of Traefik.`, Configuration: traefikConfiguration, Run: runCmd(traefikConfiguration), Resources: loaders, } } func runCmd(traefikConfiguration *static.Configuration) func(_ []string) error { return func(_ []string) error { traefikConfiguration.SetEffectiveConfiguration() resp, errPing := Do(*traefikConfiguration) if resp != nil { resp.Body.Close() } if errPing != nil { fmt.Printf("Error calling healthcheck: %s\n", errPing) os.Exit(1) } if resp.StatusCode != http.StatusOK { fmt.Printf("Bad healthcheck status: %s\n", resp.Status) os.Exit(1) } fmt.Printf("OK: %s\n", resp.Request.URL) os.Exit(0) return nil } } // Do try to do a healthcheck. func Do(staticConfiguration static.Configuration) (*http.Response, error) { if staticConfiguration.Ping == nil { return nil, errors.New("please enable `ping` to use health check") } ep := staticConfiguration.Ping.EntryPoint if ep == "" { ep = "traefik" } pingEntryPoint, ok := staticConfiguration.EntryPoints[ep] if !ok { return nil, fmt.Errorf("ping: missing %s entry point", ep) } client := &http.Client{Timeout: 5 * time.Second} protocol := "http" // FIXME Handle TLS on ping etc... // if pingEntryPoint.TLS != nil { // protocol = "https" // tr := &http.Transport{ // TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, // } // client.Transport = tr // } path := "/" return client.Head(protocol + "://" + pingEntryPoint.GetAddress() + path + "ping") }