darkkirb/harness-drone
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Integrate MigrationDriver to migrate.go

Browse source
This commit is contained in:
Nurahmadie 2014-02-15 20:16:54 +07:00
parent d2eed594ea
commit 54a9544044
1 changed files with 32 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
pkg/database/migrate/migrate.go
View file

@ -49,17 +49,41 @@ const deleteRevisionStmt = `
DELETE FROM migration where revision = ?
`
// Operation interface covers basic migration operations.
// Implementation details is specific for each database,
// see migrate/sqlite.go for implementation reference.
type Operation interface {
CreateTable(tableName string, args []string) (sql.Result, error)
RenameTable(tableName, newName string) (sql.Result, error)
DropTable(tableName string) (sql.Result, error)
AddColumn(tableName, columnSpec string) (sql.Result, error)
DropColumns(tableName string, columnsToDrop []string) (sql.Result, error)
RenameColumns(tableName string, columnChanges map[string]string) (sql.Result, error)
}
type Revision interface {
Up(tx *sql.Tx) error
Down(tx *sql.Tx) error
Up(op Operation) error
Down(op Operation) error
Revision() int64
}
type MigrationDriver struct {
Tx *sql.Tx
}
type Migration struct {
db *sql.DB
revs []Revision
}
var Driver func(tx *sql.Tx) Operation
func New(db *sql.DB) *Migration {
return &Migration{db: db}
}
@ -119,12 +143,14 @@ func (m *Migration) up(target, current int64) error {
return err
}
op := Driver(tx)
// loop through and execute revisions
for _, rev := range m.revs {
if rev.Revision() >= target {
current = rev.Revision()
// execute the revision Upgrade.
if err := rev.Up(tx); err != nil {
if err := rev.Up(op); err != nil {
log.Printf("Failed to upgrade to Revision Number %v\n", current)
log.Println(err)
return tx.Rollback()
@ -150,6 +176,8 @@ func (m *Migration) down(target, current int64) error {
return err
}
op := Driver(tx)
// reverse the list of revisions
revs := []Revision{}
for _, rev := range m.revs {
@ -162,7 +190,7 @@ func (m *Migration) down(target, current int64) error {
if rev.Revision() > target {
current = rev.Revision()
// execute the revision Upgrade.
if err := rev.Down(tx); err != nil {
if err := rev.Down(op); err != nil {
log.Printf("Failed to downgrade to Revision Number %v\n", current)
log.Println(err)
return tx.Rollback()