mirror of
https://github.com/kazhuravlev/database-gateway.git
synced 2025-06-25 17:39:53 +00:00
57 lines
1.7 KiB
Go
57 lines
1.7 KiB
Go
// Database Gateway provides access to servers with ACL for safe and restricted database interactions.
|
|
// Copyright (C) 2024 Kirill Zhuravlev
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package pgdb
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/kazhuravlev/database-gateway/internal/config"
|
|
"github.com/kazhuravlev/just"
|
|
)
|
|
|
|
func BuildDBDsn(cfg config.PostgresConfig) string { //nolint:gocritic
|
|
return fmt.Sprintf(
|
|
"host=%s port=%d user=%s password=%s dbname=%s sslmode=%s",
|
|
cfg.Host,
|
|
cfg.Port,
|
|
cfg.Username,
|
|
cfg.Password,
|
|
cfg.Database,
|
|
just.If(cfg.UseSSL, "prefer", "disable"),
|
|
)
|
|
}
|
|
|
|
func ConnectToPg(cfg config.PostgresConfig) (*sql.DB, error) { //nolint:gocritic
|
|
postgresDSN := BuildDBDsn(cfg)
|
|
dbConn, err := sql.Open("postgres", postgresDSN)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("connect to postgres: %w", err)
|
|
}
|
|
|
|
dbConn.SetMaxIdleConns(2) //nolint:mnd
|
|
dbConn.SetConnMaxLifetime(5 * time.Minute) //nolint:mnd
|
|
dbConn.SetMaxOpenConns(cfg.MaxPoolSize)
|
|
|
|
if err := dbConn.Ping(); err != nil {
|
|
return nil, fmt.Errorf("ping postgres: %w", err)
|
|
}
|
|
|
|
return dbConn, nil
|
|
}
|