query-orchestration/internal/database/migrations.go

package database

import (
	"context"
	"database/sql"
	"fmt"
	"log"
	"os"
	"path"

	"github.com/golang-migrate/migrate/v4"
	_ "github.com/golang-migrate/migrate/v4/database/postgres"
	_ "github.com/golang-migrate/migrate/v4/source/file"
	_ "github.com/lib/pq"
)

func createDB() {
	driver := "postgres"
	conf := mustGetConfig()

	opts := ""
	if conf.disableSSL {
		opts += "sslmode=disable"
	}

	connStr := fmt.Sprintf("%s://%s:%s@%s:%d/?%s", driver, conf.user, conf.password, conf.host, conf.port.Int(), opts)

	db, err := sql.Open("postgres", connStr)
	if err != nil {
		log.Panicf("Error opening database: %v", err)
	}

	err = db.Ping()
	if err != nil {
		log.Panicf("Error pinging database: %v", err)
	}

	rs, err := db.Query(fmt.Sprintf("SELECT 'CREATE DATABASE %s' WHERE NOT EXISTS (SELECT FROM pg_database WHERE datname = '%s')", conf.name, conf.name))
	if err != nil {
		log.Panicf("Error creating database: %v", err)
	}

	if rs.Next() {
		log.Printf("Database created: %s", conf.name)
	} else {
		log.Printf("Database already exists: %s", conf.name)
	}
}

type MigrationConfig struct {
	BasePath string
}

func RunMigrations(ctx context.Context, config *MigrationConfig) {
	createDB()

	connStr := mustGetURI()

	basePath := config.BasePath
	if basePath == "" {
		basePath = os.Getenv("PWD")
	}

	migPath := "file://" + path.Join(basePath, "database/migrations")
	m, err := migrate.New(
		migPath,
		connStr.String(),
	)
	if err != nil {
		log.Panicf("failed to create migrate instance: %v", err)
	}

	err = m.Up()
	if err == migrate.ErrNoChange {
		log.Println("No changes required")
	} else if err != nil {
		log.Panicf("failed to apply migrations: %v", err)
	} else {
		log.Println("Migrations applied successfully!")
	}
}