diff --git a/Dockerfile b/Dockerfile index 1b24728..86574cc 100644 --- a/Dockerfile +++ b/Dockerfile @@ -64,5 +64,5 @@ COPY --from=builder --chown=wgui:wgui /build/wg-ui /app RUN chmod +x wg-ui EXPOSE 5000/tcp -HEALTHCHECK CMD ["wget","--output-document=-","--quiet","--tries=1","http://127.0.0.1:5000/login"] +HEALTHCHECK CMD ["wget","--output-document=-","--quiet","--tries=1","http://127.0.0.1:5000/_health"] ENTRYPOINT ["./wg-ui"] diff --git a/handler/routes.go b/handler/routes.go index db837db..156e701 100644 --- a/handler/routes.go +++ b/handler/routes.go @@ -24,6 +24,13 @@ import ( "github.com/ngoduykhanh/wireguard-ui/util" ) +// Health check handler +func Health() echo.HandlerFunc { + return func(c echo.Context) error { + return c.String(http.StatusOK, "ok") + } +} + // LoginPage handler func LoginPage() echo.HandlerFunc { return func(c echo.Context) error { diff --git a/main.go b/main.go index ef376d6..8cf5dc9 100644 --- a/main.go +++ b/main.go @@ -105,6 +105,7 @@ func main() { sendmail := emailer.NewSendgridApiMail(util.SendgridApiKey, util.EmailFromName, util.EmailFrom) + app.GET("/_health", handler.Health()) app.GET("/logout", handler.Logout(), handler.ValidSession) app.POST("/new-client", handler.NewClient(db), handler.ValidSession) app.POST("/update-client", handler.UpdateClient(db), handler.ValidSession)