mirror of https://github.com/status-im/migrate.git
132 lines
2.8 KiB
Go
132 lines
2.8 KiB
Go
// Package sqlite3 implements the Driver interface.
|
|
package sqlite3
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/mattes/migrate/driver"
|
|
"github.com/mattes/migrate/file"
|
|
"github.com/mattes/migrate/migrate/direction"
|
|
"github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
type Driver struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
const tableName = "schema_migration"
|
|
|
|
func (driver *Driver) Initialize(url string) error {
|
|
filename := strings.SplitN(url, "sqlite3://", 2)
|
|
if len(filename) != 2 {
|
|
return errors.New("invalid sqlite3:// scheme")
|
|
}
|
|
|
|
db, err := sql.Open("sqlite3", filename[1])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := db.Ping(); err != nil {
|
|
return err
|
|
}
|
|
driver.db = db
|
|
|
|
if err := driver.ensureVersionTableExists(); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (driver *Driver) Close() error {
|
|
if err := driver.db.Close(); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (driver *Driver) ensureVersionTableExists() error {
|
|
if _, err := driver.db.Exec("CREATE TABLE IF NOT EXISTS " + tableName + " (version INTEGER PRIMARY KEY AUTOINCREMENT);"); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (driver *Driver) FilenameExtension() string {
|
|
return "sql"
|
|
}
|
|
|
|
func (driver *Driver) Migrate(f file.File, pipe chan interface{}) {
|
|
defer close(pipe)
|
|
pipe <- f
|
|
|
|
tx, err := driver.db.Begin()
|
|
if err != nil {
|
|
pipe <- err
|
|
return
|
|
}
|
|
|
|
if f.Direction == direction.Up {
|
|
if _, err := tx.Exec("INSERT INTO "+tableName+" (version) VALUES (?)", f.Version); err != nil {
|
|
pipe <- err
|
|
if err := tx.Rollback(); err != nil {
|
|
pipe <- err
|
|
}
|
|
return
|
|
}
|
|
} else if f.Direction == direction.Down {
|
|
if _, err := tx.Exec("DELETE FROM "+tableName+" WHERE version=?", f.Version); err != nil {
|
|
pipe <- err
|
|
if err := tx.Rollback(); err != nil {
|
|
pipe <- err
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
if err := f.ReadContent(); err != nil {
|
|
pipe <- err
|
|
return
|
|
}
|
|
|
|
if _, err := tx.Exec(string(f.Content)); err != nil {
|
|
sqliteErr, isErr := err.(sqlite3.Error)
|
|
|
|
if isErr {
|
|
// The sqlite3 library only provides error codes, not position information. Output what we do know
|
|
pipe <- errors.New(fmt.Sprintf("SQLite Error (%s); Extended (%s)\nError: %s", sqliteErr.Code.Error(), sqliteErr.ExtendedCode.Error(), sqliteErr.Error()))
|
|
} else {
|
|
pipe <- errors.New(fmt.Sprintf("An error occurred: %s", err.Error()))
|
|
}
|
|
|
|
if err := tx.Rollback(); err != nil {
|
|
pipe <- err
|
|
}
|
|
return
|
|
}
|
|
|
|
if err := tx.Commit(); err != nil {
|
|
pipe <- err
|
|
return
|
|
}
|
|
}
|
|
|
|
func (driver *Driver) Version() (uint64, error) {
|
|
var version uint64
|
|
err := driver.db.QueryRow("SELECT version FROM " + tableName + " ORDER BY version DESC LIMIT 1").Scan(&version)
|
|
switch {
|
|
case err == sql.ErrNoRows:
|
|
return 0, nil
|
|
case err != nil:
|
|
return 0, err
|
|
default:
|
|
return version, nil
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
driver.RegisterDriver("sqlite3", &Driver{})
|
|
}
|