feat: Adding ping of redis into healthcheck

Signed-off-by: Vincent Boutour <bob@vibioh.fr>
ViBiOh · Apr 10, 2021 · 18df8a3 · 18df8a3
1 parent f609fa4
commit 18df8a3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/cmd/ketchup/api.go b/cmd/ketchup/api.go
@@ -93,7 +93,9 @@ func main() {
 		}
 	}()
 
-	healthApp := health.New(healthConfig, ketchupDb.Ping)
+	tokenApp := token.New(tokenConfig)
+
+	healthApp := health.New(healthConfig, ketchupDb.Ping, tokenApp.Ping)
 
 	authServiceApp, authMiddlewareApp := initAuth(ketchupDb)
 
@@ -109,8 +111,6 @@ func main() {
 	publicRendererApp, err := renderer.New(rendererConfig, content, ketchup.FuncMap)
 	logger.Fatal(err)
 
-	tokenApp := token.New(tokenConfig)
-
 	notifierApp := notifier.New(notifierConfig, repositoryServiceApp, ketchupServiceApp, mailerApp)
 	schedulerApp := scheduler.New(schedulerConfig, notifierApp)
 	ketchupApp := ketchup.New(publicRendererApp, ketchupServiceApp, userServiceApp, repositoryServiceApp, tokenApp)

diff --git a/pkg/token/token.go b/pkg/token/token.go
@@ -18,6 +18,7 @@ type App interface {
 	Store(context.Context, string, time.Duration) (string, error)
 	Load(context.Context, string) (string, error)
 	Delete(context.Context, string) error
+	Ping() error
 }
 
 // Config of package
@@ -53,6 +54,10 @@ func New(config Config) App {
 	}
 }
 
+func (a app) Ping() error {
+	return a.redisClient.Ping(context.Background()).Err()
+}
+
 func (a app) Store(ctx context.Context, value string, duration time.Duration) (string, error) {
 	token := uuid()