mirror of
https://github.com/status-im/migrate.git
synced 2025-02-24 08:48:09 +00:00
78 lines
2.2 KiB
Go
78 lines
2.2 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/driver/bash"
|
|
"github.com/mattes/migrate/driver/cassandra"
|
|
"github.com/mattes/migrate/driver/mysql"
|
|
"github.com/mattes/migrate/driver/postgres"
|
|
"github.com/mattes/migrate/driver/sqlite3"
|
|
"github.com/mattes/migrate/file"
|
|
)
|
|
|
|
var driverMap = map[string]Driver{
|
|
"postgres": &postgres.Driver{},
|
|
"mysql": &mysql.Driver{},
|
|
"bash": &bash.Driver{},
|
|
"cassandra": &cassandra.Driver{},
|
|
"sqlite3": &sqlite3.Driver{},
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
if d, found := driverMap[u.Scheme]; found {
|
|
verifyFilenameExtension(u.Scheme, d)
|
|
if err := d.Initialize(url); err != nil {
|
|
return nil, err
|
|
}
|
|
return d, nil
|
|
}
|
|
return nil, fmt.Errorf("Driver '%s' not found.", u.Scheme)
|
|
}
|
|
|
|
// verifyFilenameExtension panics if the drivers 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))
|
|
}
|
|
}
|