mirror of
https://github.com/coredns/coredns.git
synced 2025-11-03 10:43:20 -05:00
48059a6c3e
* plugin/health: add 'overloaded metrics' Query our on health endpoint and record (and export as a metric) the time it takes. The Get has a 5s timeout, that, when reached, will set the metric duration to 5s. The actually call "I'm I overloaded" is left to an external entity. * README * golint and govet * and the tests
78 lines
1.2 KiB
Go
78 lines
1.2 KiB
Go
// Package health implements an HTTP handler that responds to health checks.
|
|
package health
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"sync"
|
|
)
|
|
|
|
var once sync.Once
|
|
|
|
type health struct {
|
|
Addr string
|
|
|
|
ln net.Listener
|
|
mux *http.ServeMux
|
|
|
|
// A slice of Healthers that the health plugin will poll every second for their health status.
|
|
h []Healther
|
|
sync.RWMutex
|
|
ok bool // ok is the global boolean indicating an all healthy plugin stack
|
|
|
|
stop chan bool
|
|
}
|
|
|
|
func (h *health) OnStartup() error {
|
|
if h.Addr == "" {
|
|
h.Addr = defAddr
|
|
}
|
|
|
|
once.Do(func() {
|
|
ln, err := net.Listen("tcp", h.Addr)
|
|
if err != nil {
|
|
log.Printf("[ERROR] Failed to start health handler: %s", err)
|
|
return
|
|
}
|
|
|
|
h.ln = ln
|
|
|
|
h.mux = http.NewServeMux()
|
|
|
|
h.mux.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
|
|
if h.Ok() {
|
|
w.WriteHeader(http.StatusOK)
|
|
io.WriteString(w, ok)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusServiceUnavailable)
|
|
})
|
|
|
|
go func() {
|
|
http.Serve(h.ln, h.mux)
|
|
}()
|
|
go func() {
|
|
h.overloaded()
|
|
}()
|
|
})
|
|
return nil
|
|
}
|
|
|
|
func (h *health) OnShutdown() error {
|
|
if h.ln != nil {
|
|
return h.ln.Close()
|
|
}
|
|
|
|
h.stop <- true
|
|
|
|
return nil
|
|
}
|
|
|
|
const (
|
|
ok = "OK"
|
|
defAddr = ":8080"
|
|
path = "/health"
|
|
)
|