mirror of
https://github.com/renorris/openfsd
synced 2026-04-18 17:05:33 +08:00
readability changes
This commit is contained in:
@@ -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 {
|
||||||
|
|||||||
Reference in New Issue
Block a user