From df26da71939db937c40212fdc995a28772c13e3d Mon Sep 17 00:00:00 2001
From: lqs <lqs@lqs.me>
Date: Fri, 5 Aug 2022 21:33:13 +0900
Subject: [PATCH] add Use() to use an existing *sql.DB handle (#14)

---
 database.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/database.go b/database.go
index 35e7a18..499e95a 100644
--- a/database.go
+++ b/database.go
@@ -85,11 +85,16 @@ func Open(driverName string, dataSourceName string) (db Database, err error) {
 			return
 		}
 	}
-	db = &database{
+	db = Use(driverName, sqlDB)
+	return
+}
+
+// Use an existing *sql.DB handle
+func Use(driverName string, sqlDB *sql.DB) Database {
+	return &database{
 		dialect: getDialectFromDriverName(driverName),
 		db:      sqlDB,
 	}
-	return
 }
 
 func (d database) GetDB() *sql.DB {