internal/postgres: replace Open with New(db)

The constructor for a postgres.DB takes a *database.DB
instead of a driver and connection string.

Change-Id: Ief58ce676946fd34cbcafc2b634a772663b6f932
Reviewed-on: https://team-review.git.corp.google.com/c/golang/discovery/+/604403
Reviewed-by: Robert Findley <rfindley@google.com>
Reviewed-by: Julie Qiu <julieqiu@google.com>

1 files changed, 3 insertions, 8 deletions

diff --git a/internal/postgres/postgres.go b/internal/postgres/postgres.go
index d437838a..49e4031b 100644
--- a/internal/postgres/postgres.go
+++ b/internal/postgres/postgres.go
@@ -14,14 +14,9 @@ type DB struct {
 	db *database.DB
 }
 
-// Open opens a new postgres DB.
-// TODO(jba): take a *sql.DB.
-func Open(driverName, dbinfo string) (*DB, error) {
-	db, err := database.Open(driverName, dbinfo)
-	if err != nil {
-		return nil, err
-	}
-	return &DB{db}, nil
+// New returns a new postgres DB.
+func New(db *database.DB) *DB {
+	return &DB{db}
 }
 
 // Close closes a DB.