35 lines
708 B
Go
35 lines
708 B
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
||
|
|
)
|
||
|
|
|
||
|
|
// NewPool creates a new pgx connection pool and verifies connectivity with a ping.
|
||
|
|
func NewPool(databaseURL string) (*pgxpool.Pool, error) {
|
||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
pool, err := pgxpool.New(ctx, databaseURL)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to create db pool: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := pool.Ping(ctx); err != nil {
|
||
|
|
pool.Close()
|
||
|
|
return nil, fmt.Errorf("failed to ping db: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return pool, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// ClosePool gracefully closes the connection pool.
|
||
|
|
func ClosePool(pool *pgxpool.Pool) {
|
||
|
|
if pool != nil {
|
||
|
|
pool.Close()
|
||
|
|
}
|
||
|
|
}
|