pgx.Conn -> pgxpool.Pool

server/cmd/serve.go

@@ -1,7 +1,6 @@
 package cmd
 
 import (
-	"context"
 	"fmt"
 	"log"
 	"net/http"
@@ -36,7 +35,7 @@ func runServe() error {
 		return err
 	}
 	db, err := postgresql.NewDB(dbURL)
-	defer db.Close(context.Background())
+	defer db.Close()
 	if err != nil {
 		return err
 	}

server/go.mod

@@ -13,9 +13,11 @@ require (
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/jackc/pgpassfile v1.0.0 // indirect
 	github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
+	github.com/jackc/puddle/v2 v2.2.2 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
 	golang.org/x/crypto v0.27.0 // indirect
 	golang.org/x/net v0.28.0 // indirect
+	golang.org/x/sync v0.8.0 // indirect
 	golang.org/x/sys v0.25.0 // indirect
 	golang.org/x/text v0.18.0 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20240814211410-ddb44dafa142 // indirect

server/pkg/storage/postgresql/db.go

@@ -4,13 +4,15 @@ import (
 	"context"
 	"fmt"
 
-	"github.com/jackc/pgx/v5"
+	// "github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgxpool"
 )
 
-func NewDB(dbURL string) (*pgx.Conn, error) {
-	conn, err := pgx.Connect(context.Background(), dbURL)
+func NewDB(dbURL string) (*pgxpool.Pool, error) {
+	// conn, err := pgx.Connect(context.Background(), dbURL)
+	dbpool, err := pgxpool.New(context.Background(), dbURL)
 	if err != nil {
 		return nil, fmt.Errorf("failed to connect to database: %w", err)
 	}
-	return conn, nil
+	return dbpool, nil
 }

server/pkg/storage/postgresql/repository/imdb.go

@@ -4,15 +4,16 @@ import (
 	"context"
 
 	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgxpool"
 
 	"github.com/aykhans/movier/server/pkg/dto"
 )
 
 type IMDbRepository struct {
-	db *pgx.Conn
+	db *pgxpool.Pool
 }
 
-func NewIMDbRepository(db *pgx.Conn) *IMDbRepository {
+func NewIMDbRepository(db *pgxpool.Pool) *IMDbRepository {
 	return &IMDbRepository{
 		db: db,
 	}