68 lines
1.1 KiB
Go
68 lines
1.1 KiB
Go
|
package driver
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
_ "github.com/jackc/pgconn"
|
||
|
|
_ "github.com/jackc/pgx/v5"
|
||
|
|
_ "github.com/jackc/pgx/v5/stdlib"
|
||
|
|
)
|
||
|
|
|
||
|
|
// DB holds the database connection pool
|
||
|
|
type DB struct {
|
||
|
|
SQL *sql.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
var dbConn = &DB{}
|
||
|
|
|
||
|
|
const (
|
||
|
|
maxOpenDbConn = 10
|
||
|
|
maxIdleDbConn = 5
|
||
|
|
maxDbLifetime = 5 * time.Minute
|
||
|
|
)
|
||
|
|
|
||
|
|
// ConnectSQL creates SQL pool for Postgres
|
||
|
|
func ConnectSQL(dsn string) (*DB, error) {
|
||
|
|
d, err := NewDatabase(dsn)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
d.SetMaxOpenConns(maxOpenDbConn)
|
||
|
|
d.SetMaxIdleConns(maxIdleDbConn)
|
||
|
|
d.SetConnMaxLifetime(maxDbLifetime)
|
||
|
|
|
||
|
|
dbConn.SQL = d
|
||
|
|
|
||
|
|
err = testDB(d)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return dbConn, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// testDB tries to ping the database
|
||
|
|
func testDB(d *sql.DB) error {
|
||
|
|
err := d.Ping()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewDatabase creates a new datavase for the application
|
||
|
|
func NewDatabase(dsn string) (*sql.DB, error) {
|
||
|
|
db, err := sql.Open("pgx", dsn)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Ping test the conn
|
||
|
|
if err = db.Ping(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return db, err
|
||
|
|
}
|