Add special handling for postgresql-specific migrations

See miniflux#1925
jvoisin · Dec 22, 2024 · c63e8bf · c63e8bf
1 parent d345c87
commit c63e8bf
Show file tree

Hide file tree

Showing 2 changed files with 161 additions and 146 deletions.
diff --git a/internal/database/database.go b/internal/database/database.go
@@ -10,7 +10,7 @@ import (
 	"time"
 
 	// Postgresql driver import
-	_ "github.com/lib/pq"
+	pq "github.com/lib/pq"
 )
 
 // NewConnectionPool configures the database connection pool.
@@ -32,6 +32,12 @@ func Migrate(db *sql.DB) error {
 	var currentVersion int
 	db.QueryRow(`SELECT version FROM schema_version`).Scan(&currentVersion)
 
+	driver := ""
+	switch db.Driver().(type) {
+	case *pq.Driver:
+		driver = "postgresql"
+	}
+
 	slog.Info("Running database migrations",
 		slog.Int("current_version", currentVersion),
 		slog.Int("latest_version", schemaVersion),
@@ -45,7 +51,7 @@ func Migrate(db *sql.DB) error {
 			return fmt.Errorf("[Migration v%d] %v", newVersion, err)
 		}
 
-		if err := migrations[version](tx); err != nil {
+		if err := migrations[version](tx, driver); err != nil {
 			tx.Rollback()
 			return fmt.Errorf("[Migration v%d] %v", newVersion, err)
 		}