mirror of
https://github.com/Dvorinka/SEEN.git
synced 2026-06-04 12:33:02 +00:00
38 lines
926 B
Go
38 lines
926 B
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
"github.com/tdvorak/seen/backend/internal/config"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func NewPool(ctx context.Context, cfg config.PostgresConfig, log *zap.Logger) (*pgxpool.Pool, error) {
|
|
poolConfig, err := pgxpool.ParseConfig(cfg.URL)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("parse postgres url: %w", err)
|
|
}
|
|
|
|
poolConfig.MaxConns = cfg.MaxConns
|
|
poolConfig.MinConns = cfg.MinConns
|
|
poolConfig.MaxConnIdleTime = 5 * time.Minute
|
|
poolConfig.HealthCheckPeriod = 30 * time.Second
|
|
|
|
pool, err := pgxpool.NewWithConfig(ctx, poolConfig)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("connect postgres: %w", err)
|
|
}
|
|
|
|
if err := pool.Ping(ctx); err != nil {
|
|
pool.Close()
|
|
return nil, fmt.Errorf("ping postgres: %w", err)
|
|
}
|
|
|
|
log.Info("postgres connected", zap.Int32("max_conns", cfg.MaxConns), zap.Int32("min_conns", cfg.MinConns))
|
|
|
|
return pool, nil
|
|
}
|