From f2da5f54256377e604c76b74805381b2acf38f50 Mon Sep 17 00:00:00 2001 From: Tom Clegg Date: Fri, 7 Aug 2020 15:00:22 -0400 Subject: [PATCH] 16306: Add "ping" health check to keep-balance. Arvados-DCO-1.1-Signed-off-by: Tom Clegg --- services/keep-balance/main.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/services/keep-balance/main.go b/services/keep-balance/main.go index 65bd8d4cf0..8b4ee84c71 100644 --- a/services/keep-balance/main.go +++ b/services/keep-balance/main.go @@ -9,13 +9,13 @@ import ( "flag" "fmt" "io" - "net/http" "os" "git.arvados.org/arvados.git/lib/config" "git.arvados.org/arvados.git/lib/service" "git.arvados.org/arvados.git/sdk/go/arvados" "git.arvados.org/arvados.git/sdk/go/ctxlog" + "git.arvados.org/arvados.git/sdk/go/health" "github.com/prometheus/client_golang/prometheus" "github.com/sirupsen/logrus" ) @@ -83,7 +83,6 @@ func runCommand(prog string, args []string, stdin io.Reader, stdout, stderr io.W } srv := &Server{ - Handler: http.NotFoundHandler(), Cluster: cluster, ArvClient: ac, RunOptions: options, @@ -91,6 +90,11 @@ func runCommand(prog string, args []string, stdin io.Reader, stdout, stderr io.W Logger: options.Logger, Dumper: options.Dumper, } + srv.Handler = &health.Handler{ + Token: cluster.ManagementToken, + Prefix: "/_health/", + Routes: health.Routes{"ping": srv.CheckHealth}, + } go srv.run() return srv -- 2.30.2