Refactor

1 files changed, 6 insertions, 2 deletions

diff --git a/forged/internal/database/database.go b/forged/internal/database/database.go
index b995adc..093ed8f 100644
--- a/forged/internal/database/database.go
+++ b/forged/internal/database/database.go
@@ -6,6 +6,7 @@ package database
 
 import (
 	"context"
+	"fmt"
 
 	"github.com/jackc/pgx/v5/pgxpool"
 )
@@ -19,7 +20,10 @@ type Database struct {
 // Open opens a new database connection pool using the provided connection
 // string. It returns a Database instance and an error if any occurs.
 // It is run indefinitely in the background.
-func Open(connString string) (Database, error) {
-	db, err := pgxpool.New(context.Background(), connString)
+func Open(ctx context.Context, config Config) (Database, error) {
+	db, err := pgxpool.New(ctx, config.Conn)
+	if err != nil {
+		err = fmt.Errorf("create pgxpool: %w", err)
+	}
 	return Database{db}, err
 }