time out on failed PG connections from database provider.

namespacelabs · Aug 14, 2024 · ee57d18 · ee57d18
1 parent 04eb2f9
commit ee57d18
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/library/oss/postgres/prepare/database/main.go b/library/oss/postgres/prepare/database/main.go
@@ -26,6 +26,7 @@ const (
 	providerPkg     = "namespacelabs.dev/foundation/library/oss/postgres"
 	connBackoff     = 1500 * time.Millisecond
 	connIdleTimeout = 15 * time.Minute
+	connTimeout     = 5 * time.Minute
 
 	caCertPath = "/tmp/ca.pem"
 )
@@ -140,6 +141,9 @@ func connect(ctx context.Context, cluster *postgresclass.ClusterInstance, db str
 	}
 	cfg.ConnectTimeout = connBackoff
 
+	ctx, cancel := context.WithTimeout(ctx, connTimeout)
+	defer cancel()
+
 	// Retry until backend is ready.
 	return backoff.RetryWithData(func() (*pgx.Conn, error) {
 		conn, err := pgx.ConnectConfig(ctx, cfg)