42 lines
846 B
Go
42 lines
846 B
Go
|
package migrate_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
|
||
|
|
_ "github.com/mattn/go-sqlite3"
|
||
|
|
"github.com/remind101/migrate"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Example() {
|
||
|
|
migrations := []migrate.Migration{
|
||
|
|
{
|
||
|
|
ID: 1,
|
||
|
|
Up: func(tx *sql.Tx) error {
|
||
|
|
_, err := tx.Exec("CREATE TABLE people (id int)")
|
||
|
|
return err
|
||
|
|
},
|
||
|
|
Down: func(tx *sql.Tx) error {
|
||
|
|
_, err := tx.Exec("DROP TABLE people")
|
||
|
|
return err
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ID: 2,
|
||
|
|
// For simple sql migrations, you can use the migrate.Queries
|
||
|
|
// helper.
|
||
|
|
Up: migrate.Queries([]string{
|
||
|
|
"ALTER TABLE people ADD COLUMN first_name text",
|
||
|
|
}),
|
||
|
|
Down: func(tx *sql.Tx) error {
|
||
|
|
// It's not possible to remove a column with
|
||
|
|
// sqlite.
|
||
|
|
_, err := tx.Exec("SELECT 1 FROM people")
|
||
|
|
return err
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
db, _ := sql.Open("sqlite3", ":memory:")
|
||
|
|
_ = migrate.Exec(db, migrate.Up, migrations...)
|
||
|
|
}
|