scottk/vault
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/vault.git synced 2025-11-01 19:17:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

postgresql: Configurable max open connections to the database

Browse Source
This commit is contained in:
vishalnayak
2015-10-01 20:11:24 -04:00
parent 614d1e804b
commit bc5ad114e4
3 changed files with 34 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
builtin/logical/postgresql/backend.go
View File

@@ -6,9 +6,9 @@ import (
"strings"
"sync"
"github.com/lib/pq"
"github.com/hashicorp/vault/logical"
"github.com/hashicorp/vault/logical/framework"
"github.com/lib/pq"
)
func Factory(conf *logical.BackendConfig) (logical.Backend, error) {
@@ -70,11 +70,13 @@ func (b *backend) DB(s logical.Storage) (*sql.DB, error) {
fmt.Errorf("configure the DB connection with config/connection first")
}
var conn string
if err := entry.DecodeJSON(&conn); err != nil {
var connConfig connectionConfig
if err := entry.DecodeJSON(&connConfig); err != nil {
return nil, err
}
conn := connConfig.ConnectionString
// Ensure timezone is set to UTC for all the conenctions
if strings.HasPrefix(conn, "postgres://") || strings.HasPrefix(conn, "postgresql://") {
var err error
@@ -92,7 +94,7 @@ func (b *backend) DB(s logical.Storage) (*sql.DB, error) {
// Set some connection pool settings. We don't need much of this,
// since the request rate shouldn't be high.
b.db.SetMaxOpenConns(2)
b.db.SetMaxOpenConns(connConfig.MaxOpenConnections)
return b.db, nil
}