diff --git a/health.go b/health.go new file mode 100644 index 0000000..708bbb7 --- /dev/null +++ b/health.go @@ -0,0 +1,34 @@ +package main + +import ( + "github.com/fasthttp/router" + "github.com/valyala/fasthttp" + "go.uber.org/atomic" +) + +const ( + healthyState = "NeoFS HTTP Gateway is " + defaultContentType = "text/plain; charset=utf-8" +) + +func attachHealthy(r *router.Router, e *atomic.Error) { + r.GET("/-/ready", func(ctx *fasthttp.RequestCtx) { + ctx.SetStatusCode(fasthttp.StatusOK) + ctx.SetBodyString(healthyState + "ready") + }) + + r.GET("/-/healthy", func(c *fasthttp.RequestCtx) { + code := fasthttp.StatusOK + msg := "healthy" + + if err := e.Load(); err != nil { + msg = "unhealthy: " + err.Error() + code = fasthttp.StatusBadRequest + } + + c.Response.Reset() + c.SetStatusCode(code) + c.SetContentType(defaultContentType) + c.SetBodyString(healthyState + msg) + }) +}