fix source/file driver

golang-migrate · Sep 16, 2021 · 0775323 · 0775323
1 parent 14f6d3a
commit 0775323
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 64 deletions.
diff --git a/source/file/file.go b/source/file/file.go
@@ -6,12 +6,34 @@ import (
 	"path/filepath"
 
 	"github.com/golang-migrate/migrate/v4/source"
+	"github.com/golang-migrate/migrate/v4/source/iofs"
 )
 
 func init() {
 	source.Register("file", &File{})
 }
 
+type File struct {
+	iofs.PartialDriver
+	url  string
+	path string
+}
+
+func (f *File) Open(url string) (source.Driver, error) {
+	p, err := parseURL(url)
+	if err != nil {
+		return nil, err
+	}
+	nf := &File{
+		url:  url,
+		path: p,
+	}
+	if err := nf.Init(os.DirFS(p), "."); err != nil {
+		return nil, err
+	}
+	return nf, nil
+}
+
 func parseURL(url string) (string, error) {
 	u, err := nurl.Parse(url)
 	if err != nil {

diff --git a/source/file/file_go115.go b/source/file/file_go115.go
diff --git a/source/file/file_go116.go b/source/file/file_go116.go