hanzo/migrate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use httpfs.PartialDriver as main godoc_vfs source implementation

Browse Source
This commit is contained in:
Dale Hui
2020-01-02 00:10:22 -08:00
parent a4820755ab
commit 8261b52d33
1 changed files with 9 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/godoc_vfs/vfs.go
+9 -90
View File
@@ -7,15 +7,11 @@
package godoc_vfs
import (
"bytes"
"fmt"
"io"
"io/ioutil"
"os"
"path"
"github.com/golang-migrate/migrate/v4/source"
"github.com/golang-migrate/migrate/v4/source/httpfs"
"golang.org/x/tools/godoc/vfs"
vfs_httpfs "golang.org/x/tools/godoc/vfs/httpfs"
)
func init() {
@@ -25,9 +21,9 @@ func init() {
// VFS is an implementation of driver that returns migrations from a virtual
// file system.
type VFS struct {
migrations *source.Migrations
fs vfs.FileSystem
path string
httpfs.PartialDriver
fs vfs.FileSystem
path string
}
// Open implements the source.Driver interface for VFS.
@@ -48,90 +44,13 @@ func WithInstance(fs vfs.FileSystem, searchPath string) (source.Driver, error) {
}
bn := &VFS{
fs: fs,
path: searchPath,
migrations: source.NewMigrations(),
fs: fs,
path: searchPath,
}
files, err := fs.ReadDir(searchPath)
if err != nil {
if err := bn.Init(vfs_httpfs.New(fs), searchPath); err != nil {
return nil, err
}
for _, fi := range files {
m, err := source.DefaultParse(fi.Name())
if err != nil {
continue // ignore files that we can't parse
}
if !bn.migrations.Append(m) {
return nil, fmt.Errorf("unable to parse file %v", fi)
}
}
return bn, nil
}
// Close implements the source.Driver interface for VFS.
// It is a no-op and should not be used.
func (b *VFS) Close() error {
return nil
}
// First returns the first migration verion found in the file system.
// If no version is available os.ErrNotExist is returned.
func (b *VFS) First() (version uint, err error) {
v, ok := b.migrations.First()
if !ok {
return 0, &os.PathError{Op: "first", Path: "<vfs>://" + b.path, Err: os.ErrNotExist}
}
return v, nil
}
// Prev returns the previous version available to the driver.
// If no previous version is available os.ErrNotExist is returned.
func (b *VFS) Prev(version uint) (prevVersion uint, err error) {
v, ok := b.migrations.Prev(version)
if !ok {
return 0, &os.PathError{Op: fmt.Sprintf("prev for version %v", version), Path: "<vfs>://" + b.path, Err: os.ErrNotExist}
}
return v, nil
}
// Prev returns the next version available to the driver.
// If no previous version is available os.ErrNotExist is returned.
func (b *VFS) Next(version uint) (nextVersion uint, err error) {
v, ok := b.migrations.Next(version)
if !ok {
return 0, &os.PathError{Op: fmt.Sprintf("next for version %v", version), Path: "<vfs>://" + b.path, Err: os.ErrNotExist}
}
return v, nil
}
// ReadUp returns the up migration body and an identifier that helps with
// finding this migration in the source.
// If there is no up migration available for this version it returns
// os.ErrNotExist.
func (b *VFS) ReadUp(version uint) (r io.ReadCloser, identifier string, err error) {
if m, ok := b.migrations.Up(version); ok {
body, err := vfs.ReadFile(b.fs, path.Join(b.path, m.Raw))
if err != nil {
return nil, "", err
}
return ioutil.NopCloser(bytes.NewReader(body)), m.Identifier, nil
}
return nil, "", &os.PathError{Op: fmt.Sprintf("read version %v", version), Path: "<vfs>://" + b.path, Err: os.ErrNotExist}
}
// ReadDown returns the down migration body and an identifier that helps with
// finding this migration in the source.
func (b *VFS) ReadDown(version uint) (r io.ReadCloser, identifier string, err error) {
if m, ok := b.migrations.Down(version); ok {
body, err := vfs.ReadFile(b.fs, path.Join(b.path, m.Raw))
if err != nil {
return nil, "", err
}
return ioutil.NopCloser(bytes.NewReader(body)), m.Identifier, nil
}
return nil, "", &os.PathError{Op: fmt.Sprintf("read version %v", version), Path: "<vfs>://" + b.path, Err: os.ErrNotExist}
}