Merge pull request #81 from xmtp/pg-max-conns

Fixes "too many clients" issue from PG by setting max open connections
xmtp · May 9, 2023 · b13f040 · b13f040
2 parents dcf0717 + e4637c5
commit b13f040
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/pkg/store/postgres/db.go b/pkg/store/postgres/db.go
@@ -8,6 +8,11 @@ import (
 	"github.com/xmtp/xmtpd/pkg/zap"
 )
 
+const (
+	// Default in postgres is typically configured to be 100.
+	maxOpenConnections = 80
+)
+
 type DB struct {
 	*sql.DB
 	DSN string
@@ -18,6 +23,7 @@ func NewDB(opts *Options) (*DB, error) {
 	if err != nil {
 		return nil, err
 	}
+	db.SetMaxOpenConns(maxOpenConnections)
 	return &DB{
 		DB:  db,
 		DSN: opts.DSN,

diff --git a/pkg/store/postgres/store.go b/pkg/store/postgres/store.go
@@ -278,6 +278,7 @@ func (s *Store) heads(ctx context.Context, tx *sql.Tx) ([]multihash.Multihash, e
 	if err != nil {
 		return nil, err
 	}
+	defer rows.Close()
 
 	var cids []multihash.Multihash
 	for rows.Next() {