[pocketbase#53] fix migrations register in framework mode

aguung · Jul 10, 2022 · 72cb2d3 · 72cb2d3
1 parent 0739e90
commit 72cb2d3
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/migrations/1640988000_init.go b/migrations/1640988000_init.go
@@ -3,6 +3,8 @@ package migrations
 
 import (
 	"fmt"
+	"path/filepath"
+	"runtime"
 
 	"github.com/pocketbase/dbx"
 	"github.com/pocketbase/pocketbase/daos"
@@ -20,7 +22,14 @@ func Register(
 	down func(db dbx.Builder) error,
 	optFilename ...string,
 ) {
-	AppMigrations.Register(up, down, optFilename...)
+	var optFiles []string
+	if len(optFilename) > 0 {
+		optFiles = optFilename
+	} else {
+		_, path, _, _ := runtime.Caller(1)
+		optFiles = append(optFiles, filepath.Base(path))
+	}
+	AppMigrations.Register(up, down, optFiles...)
 }
 
 func init() {