From 46a0d19c7f3f3d85123000aa6f088e477b602a5e Mon Sep 17 00:00:00 2001
From: Jude Zhu <judezhu.hust@gmail.com>
Date: Tue, 9 Apr 2024 07:06:22 -0700
Subject: [PATCH] refactor load migrations

---
 pkg/spanner/migration.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/pkg/spanner/migration.go b/pkg/spanner/migration.go
index 2472e5e..f4b8260 100644
--- a/pkg/spanner/migration.go
+++ b/pkg/spanner/migration.go
@@ -85,10 +85,13 @@ func LoadMigrations(dir string) (Migrations, error) {
 		return nil, err
 	}
 
+	return LoadMigrationFiles(dir, files)
+}
+
+func LoadMigrationFiles(baseDir string, files []os.DirEntry) (Migrations, error) {
 	var migrations Migrations
 
 	versions := map[uint64]string{}
-
 	for _, f := range files {
 		if f.IsDir() {
 			continue
@@ -106,7 +109,7 @@ func LoadMigrations(dir string) (Migrations, error) {
 			continue
 		}
 
-		file, err := os.ReadFile(filepath.Join(dir, filename))
+		file, err := os.ReadFile(filepath.Join(baseDir, filename))
 		if err != nil {
 			continue
 		}