add timeout to Ping check

2 months ago · 62f9656e19
parent 663c957880
commit 62f9656e19
1 changed files with 4 additions and 1 deletions
--- a/cmd/api/main.go
+++ b/cmd/api/main.go
@ -72,7 +72,10 @@ func run(ctx context.Context, w io.Writer, args []string) error {
 		pool.Close()
 	}()

-	if err := pool.Ping(appCtx); err != nil {
+	timoutCtx, close := context.WithTimeout(appCtx, 5*time.Second)
+	defer close()
+
+	if err := pool.Ping(timoutCtx); err != nil {
 		slog.Error("Error in attempting first postgres database connection")
 		return err
 	}