nikola/go_sql_schema_migrate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make package

Browse Source
This commit is contained in:
Nikola Petrov
2026-05-31 16:11:06 +02:00
parent 4898b6ab27
commit 7375d2d168
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
main.go → cmd/go-migrate/main.go
View File

@@ -7,7 +7,7 @@ import (
"log" "log"
"os" "os"
"git.petrovv.com/go-migrate/lib" "git.petrovv.com/go-migrate"
"github.com/jackc/pgx/v5/pgxpool" "github.com/jackc/pgx/v5/pgxpool"
"github.com/joho/godotenv" "github.com/joho/godotenv"
) )
@@ -44,13 +44,13 @@ func generateMigrations(schemaDir string) {
defer db.Close() defer db.Close()
ctx := context.Background() ctx := context.Background()
migrations, err := lib.GetMigrations(ctx, db, schemaDir) migrations, err := migrate.GetMigrations(ctx, db, schemaDir)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
if len(migrations) == 0 { if len(migrations) == 0 {
fmt.Println("✓ Database schema is up to date") fmt.Println("Database schema is up to date")
return return
} }
@@ -66,13 +66,13 @@ func applyMigrations(schemaDir string) {
defer db.Close() defer db.Close()
ctx := context.Background() ctx := context.Background()
migrations, err := lib.GetMigrations(ctx, db, schemaDir) migrations, err := migrate.GetMigrations(ctx, db, schemaDir)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
if len(migrations) == 0 { if len(migrations) == 0 {
fmt.Println("✓ Database schema is up to date") fmt.Println("Database schema is up to date")
return return
} }
@@ -85,7 +85,7 @@ func applyMigrations(schemaDir string) {
} }
} }
fmt.Println("✓ All migrations applied successfully") fmt.Println("All migrations applied successfully")
} }
func getDBConnection() *pgxpool.Pool { func getDBConnection() *pgxpool.Pool {

5
lib/lib.go → migrate.go
View File

@@ -1,4 +1,4 @@
package lib package migrate
import ( import (
"context" "context"
@@ -981,10 +981,13 @@ func parseForeignKey(def string) *ForeignKey {
return fk return fk
} }
// LoadDesiredSchema loads schema definitions from SQL files in the given directory
func LoadDesiredSchema(dir string) (*Schema, error) { func LoadDesiredSchema(dir string) (*Schema, error) {
return loadDesiredSchema(dir) return loadDesiredSchema(dir)
} }
// GetMigrations compares the current database schema with the desired schema
// and returns a list of SQL migration statements
func GetMigrations(ctx context.Context, db *pgxpool.Pool, schemaDir string) ([]string, error) { func GetMigrations(ctx context.Context, db *pgxpool.Pool, schemaDir string) ([]string, error) {
// Load current database schema // Load current database schema
currentSchema, err := GetCurrentSchema(ctx, db) currentSchema, err := GetCurrentSchema(ctx, db)