Merge pull request #5 from kodefluence/add-eject-method

Adding method to eject real database

db/adapter.go

@@ -105,3 +105,8 @@ func runWithSQLAnalyzer(ktx kontext.Context, executionLevel, function string, f
 	}
 	return nil
 }
+
+// Eject sql.DB out of db adapter
+func (a *Adapter) Eject() *sql.DB {
+	return a.db
+}

db/adapter_test.go

@@ -43,6 +43,16 @@ func TestAdapter(t *testing.T) {
 		})
 	})
 
+	t.Run("Eject", func(t *testing.T) {
+		t.Run("When ejected it return *sql.DB instance", func(t *testing.T) {
+			sqldb, _, _ := sqlmock.New(sqlmock.MonitorPingsOption(true))
+			defer sqldb.Close()
+
+			sql := db.Adapt(sqldb)
+			assert.Equal(t, sqldb, sql.Eject())
+		})
+	})
+
 	t.Run("QueryRowContext", func(t *testing.T) {
 		t.Run("When querying done it will return db.Row and scan the value", func(t *testing.T) {
 

db/db.go

@@ -1,6 +1,8 @@
 package db
 
 import (
+	"database/sql"
+
 	"github.com/kodefluence/monorepo/exception"
 	"github.com/kodefluence/monorepo/kontext"
 )
@@ -12,6 +14,7 @@ type DB interface {
 	Ping(ktx kontext.Context) exception.Exception
 	Transactionable
 	TX
+	Eject() *sql.DB
 }
 
 // Transactionable is wrapper to create transaction process