add Use() to use an existing *sql.DB handle (#14)

lqs · Aug 5, 2022 · df26da7 · df26da7
1 parent 630b1fc
commit df26da7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git 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 {