64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package migrate
|
|
|
|
import (
|
|
"context"
|
|
"embed"
|
|
"fmt"
|
|
"log"
|
|
"log/slog"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
//go:embed migrations/*.sql
|
|
var migrations embed.FS
|
|
|
|
type Migration struct {
|
|
Version int
|
|
Name string
|
|
}
|
|
|
|
func Migrate(logger *slog.Logger, db *pgxpool.Pool) {
|
|
logger.Info("Running migrations...")
|
|
migrationTableSql := `
|
|
CREATE TABLE IF NOT EXISTS migrations(
|
|
version SERIAL PRIMARY KEY,
|
|
name VARCHAR(50)
|
|
);`
|
|
|
|
_, err := db.Exec(context.Background(), migrationTableSql)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
files, err := migrations.ReadDir("migrations")
|
|
if err != nil {
|
|
log.Fatal("Failed to read migrations directory")
|
|
}
|
|
|
|
for _, file := range files {
|
|
var migration Migration
|
|
row := db.QueryRow(context.Background(), "SELECT * FROM migrations WHERE name = $1;", file.Name())
|
|
err = row.Scan(&migration.Version, &migration.Name)
|
|
if err == pgx.ErrNoRows {
|
|
logger.Info(fmt.Sprintf("Running migration: %s", file.Name()))
|
|
migrationSql, err := migrations.ReadFile(fmt.Sprintf("migrations/%s", file.Name()))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
_, err = db.Exec(context.Background(), string(migrationSql))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
_, err = db.Exec(context.Background(), "INSERT INTO migrations(name) VALUES($1);", file.Name())
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
logger.Info("Migrations completed")
|
|
}
|