mirror of
https://github.com/status-im/migrate.git
synced 2025-02-24 08:48:09 +00:00
9754378aa5
We can't test driver.New methods because of circular imports.
That's probably why the initial code had a map[string]interface{} as
registry: it removes a dependency import.
I prefer the remove the registry and have a registry returning a real
Driver. It will ease the development later.
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
// Package driver holds the driver interface.
|
|
package driver
|
|
|
|
import (
|
|
"fmt"
|
|
neturl "net/url" // alias to allow `url string` func signature in New
|
|
|
|
"github.com/mattes/migrate/file"
|
|
)
|
|
|
|
// Driver is the interface type that needs to implemented by all drivers.
|
|
type Driver interface {
|
|
|
|
// Initialize is the first function to be called.
|
|
// Check the url string and open and verify any connection
|
|
// that has to be made.
|
|
Initialize(url string) error
|
|
|
|
// Close is the last function to be called.
|
|
// Close any open connection here.
|
|
Close() error
|
|
|
|
// FilenameExtension returns the extension of the migration files.
|
|
// The returned string must not begin with a dot.
|
|
FilenameExtension() string
|
|
|
|
// Migrate is the heart of the driver.
|
|
// It will receive a file which the driver should apply
|
|
// to its backend or whatever. The migration function should use
|
|
// the pipe channel to return any errors or other useful information.
|
|
Migrate(file file.File, pipe chan interface{})
|
|
|
|
// Version returns the current migration version.
|
|
Version() (uint64, error)
|
|
}
|
|
|
|
// New returns Driver and calls Initialize on it
|
|
func New(url string) (Driver, error) {
|
|
u, err := neturl.Parse(url)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
d := GetDriver(u.Scheme)
|
|
if d == nil {
|
|
return nil, fmt.Errorf("Driver '%s' not found.", u.Scheme)
|
|
}
|
|
verifyFilenameExtension(u.Scheme, d)
|
|
if err := d.Initialize(url); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return d, nil
|
|
}
|
|
|
|
// verifyFilenameExtension panics if the driver's filename extension
|
|
// is not correct or empty.
|
|
func verifyFilenameExtension(driverName string, d Driver) {
|
|
f := d.FilenameExtension()
|
|
if f == "" {
|
|
panic(fmt.Sprintf("%s.FilenameExtension() returns empty string.", driverName))
|
|
}
|
|
if f[0:1] == "." {
|
|
panic(fmt.Sprintf("%s.FilenameExtension() returned string must not start with a dot.", driverName))
|
|
}
|
|
}
|