From b96a0c781e5f85515ecfbede2e782bdaf254b603 Mon Sep 17 00:00:00 2001
From: Carlos Alexandro Becker <caarlos0@gmail.com>
Date: Tue, 8 Sep 2015 21:39:11 -0300
Subject: [PATCH] Allowing the use of an existing database/sql pool

---
 pgstore.go | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/pgstore.go b/pgstore.go
index 687b860..e3be094 100644
--- a/pgstore.go
+++ b/pgstore.go
@@ -31,9 +31,19 @@ type Session struct {
 	ExpiresOn  time.Time `db:"expires_on"`
 }
 
-// NewPGStore creates a new PGStore instance
+// NewPGStore creates a new PGStore instance and a new database/sql pool
 func NewPGStore(dbURL string, keyPairs ...[]byte) *PGStore {
 	db, err := sql.Open("postgres", dbURL)
+	if err != nil {
+		// Ignore and return nil
+		return nil
+	}
+	return NewPGStoreFromPool(db, keyPairs...)
+}
+
+// NewPGStoreFromPool creates a new PGStore instance from an existing
+// database/sql pool
+func NewPGStoreFromPool(db *sql.DB, keyPairs ...[]byte) *PGStore {
 	dbmap := &gorp.DbMap{Db: db, Dialect: gorp.PostgresDialect{}}
 
 	dbStore := &PGStore{
@@ -45,14 +55,9 @@ func NewPGStore(dbURL string, keyPairs ...[]byte) *PGStore {
 		DbMap: dbmap,
 	}
 
-	if err != nil {
-		// Ignore and return nil
-		return nil
-	}
-
 	// Create table if it doesn't exist
 	dbmap.AddTableWithName(Session{}, "http_sessions").SetKeys(true, "Id")
-	err = dbmap.CreateTablesIfNotExists()
+	err := dbmap.CreateTablesIfNotExists()
 
 	if err != nil {
 		// Ignore and return nil