postgresql/migration.go

package postgresql

import (
	"database/sql"
	"fmt"
	"os"

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

func runMigration(db *sql.DB, server string) {
	// init migrate data
	driver, _ := postgres.WithInstance(db, &postgres.Config{})
	m, err := migrate.NewWithDatabaseInstance(
		fmt.Sprintf("file:///%s", getMigrationDirectoryPath(server)),
		"postgres",
		driver,
	)

	// up
	if err == nil {
		if err = m.Up(); err != nil && err.Error() != "no change" {
			fmt.Printf("[postgresql] run migration error: %s", err.Error())
			return
		}
		fmt.Printf("⚡️[postgres]: done migration \n")
	}
}

func getMigrationDirectoryPath(server string) string {
	migrationDir := fmt.Sprintf("/external/postgresql/%s/migrations/sql", server)

	dirname, err := os.Getwd()
	if err != nil {
		panic(err)
	}

	// Check path existed
	path := dirname + migrationDir
	if _, err = os.Stat(path); os.IsNotExist(err) {
		// Create if not existed
		err = os.Mkdir(path, 0755)
		if err != nil {
			panic(err)
		}
	}

	return path
}