mirror of https://github.com/status-im/migrate.git
split sql statements by ;
This commit is contained in:
parent
2faaebc76a
commit
cc23e29eef
|
@ -7,15 +7,11 @@ import (
|
|||
"database/sql"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/go-sql-driver/mysql"
|
||||
"github.com/mattes/migrate/file"
|
||||
"github.com/mattes/migrate/migrate/direction"
|
||||
"regexp"
|
||||
"strconv"
|
||||
|
||||
// Use fork instead of github.com/go-sql-driver/mysql, because it
|
||||
// has clientMultiStatements enabled.
|
||||
// see https://github.com/go-sql-driver/mysql/issues/66
|
||||
"github.com/mattes/mysql"
|
||||
)
|
||||
|
||||
type Driver struct {
|
||||
|
@ -93,7 +89,12 @@ func (driver *Driver) Migrate(f file.File, pipe chan interface{}) {
|
|||
return
|
||||
}
|
||||
|
||||
if _, err := tx.Exec(string(f.Content)); err != nil {
|
||||
// TODO this is not good! unfortunately there is no mysql driver that
|
||||
// supports multiple statements per query.
|
||||
sqlStmts := bytes.Split(f.Content, ";")
|
||||
|
||||
for _, sqlStmt := range sqlStmts {
|
||||
if _, err := tx.Exec(string(sqlStmt)); err != nil {
|
||||
mysqlErr := err.(*mysql.MySQLError)
|
||||
|
||||
re, err := regexp.Compile(`at line ([0-9]+)$`)
|
||||
|
@ -140,6 +141,7 @@ func (driver *Driver) Migrate(f file.File, pipe chan interface{}) {
|
|||
}
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if err := tx.Commit(); err != nil {
|
||||
pipe <- err
|
||||
|
|
Loading…
Reference in New Issue