55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
|
|
"tanabata/internal/domain"
|
|
)
|
|
|
|
// Initialize PostgreSQL database driver
|
|
func New(dbURL string) (*pgxpool.Pool, error) {
|
|
poolConfig, err := pgxpool.ParseConfig(dbURL)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to parse connection string: %w", err)
|
|
}
|
|
|
|
poolConfig.MaxConns = 100
|
|
poolConfig.MinConns = 0
|
|
poolConfig.MaxConnLifetime = time.Hour
|
|
poolConfig.HealthCheckPeriod = 30 * time.Second
|
|
|
|
ctx := context.Background()
|
|
db, err := pgxpool.NewWithConfig(ctx, poolConfig)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to initialize DB connections pool: %w", err)
|
|
}
|
|
if err = db.Ping(ctx); err != nil {
|
|
return nil, fmt.Errorf("failed to ping database: %w", err)
|
|
}
|
|
return db, nil
|
|
}
|
|
|
|
// Transaction wrapper
|
|
func transaction(ctx context.Context, db *pgxpool.Pool, handler func(context.Context, pgx.Tx) *domain.DomainError) (domainErr *domain.DomainError) {
|
|
tx, err := db.Begin(ctx)
|
|
if err != nil {
|
|
domainErr = domain.NewErrorUnexpected().Wrap(err)
|
|
return
|
|
}
|
|
domainErr = handler(ctx, tx)
|
|
if domainErr != nil {
|
|
tx.Rollback(ctx)
|
|
return
|
|
}
|
|
err = tx.Commit(ctx)
|
|
if err != nil {
|
|
domainErr = domain.NewErrorUnexpected().Wrap(err)
|
|
}
|
|
return
|
|
}
|