Add method for re-using connection (#6)

go.mod

@@ -4,7 +4,7 @@ go 1.19
 
 require (
 	github.com/galeone/fitbit v1.0.3
-	github.com/galeone/igor v1.0.8
+	github.com/galeone/igor v1.0.9
 	github.com/joho/godotenv v1.5.1
 )
 

go.sum

@@ -1,7 +1,7 @@
 github.com/galeone/fitbit v1.0.3 h1:iYC3japYbCsWc6SyMz3JXTwKnCnG3y8oLgZ9uy4yQKM=
 github.com/galeone/fitbit v1.0.3/go.mod h1:osb+HFcVMMkAP/NH2+HVB2UAYP7A1kOpKYmkfbtL2fY=
-github.com/galeone/igor v1.0.8 h1:necystLPnQ1slRNfsp5n9Z1dtgee5PJCndDHem5CNpU=
-github.com/galeone/igor v1.0.8/go.mod h1:YhORY+h+kG1n6RyuvoEh82h8Dkc/KXHZzaAkVPWtcSc=
+github.com/galeone/igor v1.0.9 h1:nGS0WhMkzlwC0hHOUiRNJEFCHOInm+fTSiopy5FzyKc=
+github.com/galeone/igor v1.0.9/go.mod h1:YhORY+h+kG1n6RyuvoEh82h8Dkc/KXHZzaAkVPWtcSc=
 github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
 github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=

pgdb.go

@@ -11,6 +11,8 @@ package fitbit_pgdb
 // [2]: https://github.com/galeone/fitbit
 
 import (
+	"database/sql"
+
 	"github.com/galeone/fitbit/types"
 	"github.com/galeone/igor"
 
@@ -26,7 +28,7 @@ type PGDB struct {
 // You must provide the connection string to use for connecting to a
 // running PostgreSQL instance.
 //
-// It implements the `fitbit.Storage` interface.
+// *PGDB implements the `fitbit.Storage` interface.
 func NewPGDB(connectionString string) *PGDB {
 	var err error
 	var db *igor.Database
@@ -39,6 +41,22 @@ func NewPGDB(connectionString string) *PGDB {
 	}
 }
 
+// NewPGDBFromConnection creates a new instance of PGDB without creating
+// a new connection. It re-uses the database connection provided as input.
+//
+// *PGDB implements the `fitbit.Storage` interface.
+func NewPGDBFromConnection(connection *sql.DB) *PGDB {
+	var err error
+	var db *igor.Database
+	if db, err = igor.Wrap(connection); err != nil {
+		panic(err.Error())
+	}
+
+	return &PGDB{
+		db,
+	}
+}
+
 // InsertAuhorizingUser executes a CREATE query on the PostgreSQL database
 // inserting the authorizing user in the associated table.
 func (s *PGDB) InsertAuhorizingUser(authorizing *types.AuthorizingUser) error {

pgdb_test.go

@@ -5,6 +5,7 @@
 package fitbit_pgdb_test
 
 import (
+	"database/sql"
 	"fmt"
 	"os"
 	"testing"
@@ -67,6 +68,21 @@ func init() {
 
 var db fitbit.Storage
 
+func TestPGDBNew(t *testing.T) {
+	// Test connection re-use
+	if connection, err := sql.Open("postgres", _connectionString); err != nil {
+		if db = pgdb.NewPGDBFromConnection(connection); db == nil {
+			t.Errorf("Unable to re-use db connection when creating PGDB object")
+		}
+	}
+
+	// Test connection creation
+	if db = pgdb.NewPGDB(_connectionString); db == nil {
+		t.Errorf("Unable to create a PGDB object creating a connection")
+	}
+
+}
+
 func TestAuthorizingUser(t *testing.T) {
 	db = pgdb.NewPGDB(_connectionString)
 	var err error