golib/database/sqlmigrate/mymigrate/examples/todos/main.go

// Command todos demonstrates mymigrate with embedded SQL migrations.
//
// Usage:
//
//	export MY_URL='user:pass@tcp(host:3306)/dbname?multiStatements=true&parseTime=true'
//	go run . up        # apply all pending migrations
//	go run . up 1      # apply next 1 migration
//	go run . down      # roll back last migration
//	go run . down 2    # roll back last 2 migrations
//	go run . status    # show applied and pending
//	go run . reset     # roll back all migrations
package main

import (
	"context"
	"database/sql"
	"embed"
	"fmt"
	stdfs "io/fs"
	"os"
	"strconv"

	_ "github.com/go-sql-driver/mysql"

	"github.com/therootcompany/golib/database/sqlmigrate"
	"github.com/therootcompany/golib/database/sqlmigrate/mymigrate"
)

//go:embed migrations/*.sql
var migrationsFS embed.FS

func main() {
	if err := run(); err != nil {
		fmt.Fprintf(os.Stderr, "error: %v\n", err)
		os.Exit(1)
	}
}

func run() error {
	ctx := context.Background()

	myURL := os.Getenv("MY_URL")
	if myURL == "" {
		return fmt.Errorf("MY_URL environment variable is required\n  Format: user:pass@tcp(host:port)/dbname?multiStatements=true&parseTime=true")
	}

	db, err := sql.Open("mysql", myURL)
	if err != nil {
		return fmt.Errorf("opening database: %w", err)
	}
	defer db.Close()

	if err := db.PingContext(ctx); err != nil {
		return fmt.Errorf("connecting: %w", err)
	}

	runner, err := mymigrate.New(db)
	if err != nil {
		return err
	}

	subFS, err := stdfs.Sub(migrationsFS, "migrations")
	if err != nil {
		return fmt.Errorf("reading migrations: %w", err)
	}
	migrations, err := sqlmigrate.Collect(subFS)
	if err != nil {
		return fmt.Errorf("collecting migrations: %w", err)
	}

	args := os.Args[1:]
	if len(args) == 0 {
		args = []string{"status"}
	}

	switch args[0] {
	case "up":
		n := -1
		if len(args) > 1 {
			n, err = strconv.Atoi(args[1])
			if err != nil || n < 1 {
				return fmt.Errorf("up count must be a positive integer")
			}
		}
		applied, err := sqlmigrate.Up(ctx, runner, migrations, n)
		if err != nil {
			return err
		}
		if len(applied) == 0 {
			fmt.Println("Already up-to-date.")
		} else {
			for _, name := range applied {
				fmt.Printf("  applied: %s\n", name)
			}
		}

	case "down":
		n := 1
		if len(args) > 1 {
			n, err = strconv.Atoi(args[1])
			if err != nil || n < 1 {
				return fmt.Errorf("down count must be a positive integer")
			}
		}
		rolled, err := sqlmigrate.Down(ctx, runner, migrations, n)
		if err != nil {
			return err
		}
		if len(rolled) == 0 {
			fmt.Println("Nothing to roll back.")
		} else {
			for _, name := range rolled {
				fmt.Printf("  rolled back: %s\n", name)
			}
		}

	case "status":
		status, err := sqlmigrate.GetStatus(ctx, runner, migrations)
		if err != nil {
			return err
		}
		fmt.Printf("Applied: %d\n", len(status.Applied))
		for _, name := range status.Applied {
			fmt.Printf("  %s\n", name)
		}
		fmt.Printf("Pending: %d\n", len(status.Pending))
		for _, name := range status.Pending {
			fmt.Printf("  %s\n", name)
		}

	case "reset":
		rolled, err := sqlmigrate.Down(ctx, runner, migrations, -1)
		if err != nil {
			return err
		}
		if len(rolled) == 0 {
			fmt.Println("Nothing to reset.")
		} else {
			for _, name := range rolled {
				fmt.Printf("  rolled back: %s\n", name)
			}
		}

	default:
		return fmt.Errorf("unknown command %q (use: up, down, status, reset)", args[0])
	}

	return nil
}