set max pool size

PeerDB-io · Dec 10, 2023 · 6d42fa2 · 6d42fa2
1 parent 350bb86
commit 6d42fa2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/flow/connectors/postgres/client.go b/flow/connectors/postgres/client.go
@@ -217,7 +217,6 @@ func (c *PostgresConnector) GetSlotInfo(slotName string) ([]*protos.SlotInfo, er
     FROM
         pg_control_checkpoint(),
         pg_replication_slots`+specificSlotClause+`;`)
-
 	if err != nil {
 		return nil, err
 	}

diff --git a/flow/connectors/postgres/postgres.go b/flow/connectors/postgres/postgres.go
@@ -47,6 +47,9 @@ func NewPostgresConnector(ctx context.Context, pgConfig *protos.PostgresConfig)
 	runtimeParams["idle_in_transaction_session_timeout"] = "0"
 	runtimeParams["statement_timeout"] = "0"
 
+	// set pool size to 3 to avoid connection pool exhaustion
+	connConfig.MaxConns = 3
+
 	pool, err := pgxpool.NewWithConfig(ctx, connConfig)
 	if err != nil {
 		return nil, fmt.Errorf("failed to create connection pool: %w", err)