use common state.DB interface

pressly · Oct 10, 2023 · c445b82 · c445b82
1 parent 8ebcb11
commit c445b82
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/migration.go b/migration.go
@@ -11,6 +11,7 @@ import (
 	"time"
 
 	"github.com/pressly/goose/v3/internal/sqlparser"
+	"github.com/pressly/goose/v3/internal/state"
 )
 
 // MigrationRecord struct.
@@ -165,7 +166,7 @@ func runGoMigrationNoTx(
 		}
 	}
 	if recordVersion {
-		return insertOrDeleteVersionNoTx(ctx, db, version, direction)
+		return insertOrDeleteVersion(ctx, db, version, direction)
 	}
 	return nil
 }
@@ -204,14 +205,7 @@ func runGoMigration(
 	return nil
 }
 
-func insertOrDeleteVersion(ctx context.Context, tx *sql.Tx, version int64, direction bool) error {
-	if direction {
-		return globalStorage().InsertVersion(ctx, tx, version)
-	}
-	return globalStorage().DeleteVersion(ctx, tx, version)
-}
-
-func insertOrDeleteVersionNoTx(ctx context.Context, db *sql.DB, version int64, direction bool) error {
+func insertOrDeleteVersion(ctx context.Context, db state.DB, version int64, direction bool) error {
 	if direction {
 		return globalStorage().InsertVersion(ctx, db, version)
 	}