parent
ad8f12c9fd
commit
f83438b693
@ -1,34 +0,0 @@ |
|||||||
package main |
|
||||||
|
|
||||||
import ( |
|
||||||
"context" |
|
||||||
"errors" |
|
||||||
"log" |
|
||||||
|
|
||||||
"github.com/go-kratos/kratos/v2" |
|
||||||
"github.com/go-kratos/kratos/v2/transport/http" |
|
||||||
"github.com/go-kratos/kratos/v2/transport/http/health" |
|
||||||
) |
|
||||||
|
|
||||||
func main() { |
|
||||||
handler := health.NewHandler() |
|
||||||
handler.AddChecker("mysql", func(ctx context.Context) error { |
|
||||||
return nil |
|
||||||
}) |
|
||||||
handler.AddObserver("redis", func(ctx context.Context) error { |
|
||||||
return errors.New("connection refused") |
|
||||||
}) |
|
||||||
|
|
||||||
httpSrv := http.NewServer(http.Address(":8000")) |
|
||||||
httpSrv.Handle("/healthz", handler) |
|
||||||
|
|
||||||
app := kratos.New( |
|
||||||
kratos.Name("mux"), |
|
||||||
kratos.Server( |
|
||||||
httpSrv, |
|
||||||
), |
|
||||||
) |
|
||||||
if err := app.Run(); err != nil { |
|
||||||
log.Println(err) |
|
||||||
} |
|
||||||
} |
|
@ -1,66 +0,0 @@ |
|||||||
package health |
|
||||||
|
|
||||||
import ( |
|
||||||
"context" |
|
||||||
"encoding/json" |
|
||||||
"net/http" |
|
||||||
) |
|
||||||
|
|
||||||
// CheckerFunc wraps the CheckHealth method.
|
|
||||||
//
|
|
||||||
// CheckHealth returns nil if the resource is healthy, or a non-nil
|
|
||||||
// error if the resource is not healthy. CheckHealth must be safe to
|
|
||||||
// call from multiple goroutines.
|
|
||||||
type CheckerFunc func(ctx context.Context) error |
|
||||||
|
|
||||||
// Handler is an HTTP handler that reports on the success of an
|
|
||||||
// aggregate of Checkers. The zero value is always healthy.
|
|
||||||
type Handler struct { |
|
||||||
checkers map[string]CheckerFunc |
|
||||||
observers map[string]CheckerFunc |
|
||||||
} |
|
||||||
|
|
||||||
// NewHandler new a health handler.
|
|
||||||
func NewHandler() *Handler { |
|
||||||
return &Handler{ |
|
||||||
checkers: make(map[string]CheckerFunc), |
|
||||||
observers: make(map[string]CheckerFunc), |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
// AddChecker adds a new check to the handler.
|
|
||||||
func (h *Handler) AddChecker(name string, c CheckerFunc) { |
|
||||||
h.checkers[name] = c |
|
||||||
} |
|
||||||
|
|
||||||
// AddObserver adds a new check to the handler but it does not fail the entire status.
|
|
||||||
func (h *Handler) AddObserver(name string, c CheckerFunc) { |
|
||||||
h.observers[name] = c |
|
||||||
} |
|
||||||
|
|
||||||
// ServeHTTP returns 200 if it is healthy, 500 otherwise.
|
|
||||||
func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) { |
|
||||||
code := http.StatusOK |
|
||||||
results := make(map[string]string, len(h.checkers)) |
|
||||||
|
|
||||||
for name, checker := range h.checkers { |
|
||||||
if err := checker(r.Context()); err != nil { |
|
||||||
code = http.StatusInternalServerError |
|
||||||
results[name] = err.Error() |
|
||||||
} else { |
|
||||||
results[name] = "OK" |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
for name, checker := range h.observers { |
|
||||||
if err := checker(r.Context()); err != nil { |
|
||||||
results[name] = err.Error() |
|
||||||
} else { |
|
||||||
results[name] = "OK" |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
w.WriteHeader(code) |
|
||||||
w.Header().Set("Content-Type", "application/json; charset=utf-8") |
|
||||||
json.NewEncoder(w).Encode(results) |
|
||||||
} |
|
Loading…
Reference in new issue