readability changes

This commit is contained in:
Reese Norris
2025-06-12 12:22:11 -07:00
parent e09ff9e64e
commit a8c8e578f1

View File

@@ -7,7 +7,7 @@ import (
"fmt" "fmt"
"github.com/golang-migrate/migrate/v4" "github.com/golang-migrate/migrate/v4"
"github.com/golang-migrate/migrate/v4/database" "github.com/golang-migrate/migrate/v4/database"
"github.com/golang-migrate/migrate/v4/database/postgres" migratePostgres "github.com/golang-migrate/migrate/v4/database/postgres"
migrateSqlite "github.com/golang-migrate/migrate/v4/database/sqlite" migrateSqlite "github.com/golang-migrate/migrate/v4/database/sqlite"
"github.com/golang-migrate/migrate/v4/source/iofs" "github.com/golang-migrate/migrate/v4/source/iofs"
"github.com/lib/pq" // PostgreSQL driver "github.com/lib/pq" // PostgreSQL driver
@@ -19,26 +19,24 @@ var migrationsFS embed.FS
// Migrate applies database migrations. // Migrate applies database migrations.
func Migrate(db *sql.DB) (err error) { func Migrate(db *sql.DB) (err error) {
var migrationPath string
var driver database.Driver var driver database.Driver
var dbType string var dbType string
var migrationPath string
switch db.Driver().(type) { switch db.Driver().(type) {
case *pq.Driver: case *pq.Driver:
dbType = "postgres" dbType = "postgres"
migrationPath = "migrations/postgres" migrationPath = "migrations/postgres"
if driver, err = postgres.WithInstance(db, &postgres.Config{}); err != nil { driver, err = migratePostgres.WithInstance(db, &migratePostgres.Config{})
return
}
case *sqlite.Driver: case *sqlite.Driver:
dbType = "sqlite" dbType = "sqlite"
migrationPath = "migrations/sqlite" migrationPath = "migrations/sqlite"
if driver, err = migrateSqlite.WithInstance(db, &migrateSqlite.Config{}); err != nil { driver, err = migrateSqlite.WithInstance(db, &migrateSqlite.Config{})
return
}
default: default:
return fmt.Errorf("unsupported database type") return fmt.Errorf("unsupported database type")
} }
if err != nil {
return err
}
d, err := iofs.New(migrationsFS, migrationPath) d, err := iofs.New(migrationsFS, migrationPath)
if err != nil { if err != nil {