split sql scripts

migrate.go

@@ -4,6 +4,7 @@ import (
 	"errors"
 	"io/ioutil"
 	"path/filepath"
+	"strings"
 )
 
 // NewMigrationHandler - create new migration handler.
@@ -95,15 +96,39 @@ func (m *MigrationHandler) runScript(scriptName string) error {
 		return err
 	}
 
-	// EXEC SCRIPT
+	// split script
+	scriptsQuery := strings.Split(string(fileBytes), ";")
+	for _, sqlQuery := range scriptsQuery {
+		// EXEC SCRIPT
+		err := m.runTx(sqlQuery)
+		if err != nil {
+			return err
+		}
+	}
+
+	// update version
+	sqlQuery := "INSERT INTO " + versionsTableName + " SET name=?"
+	_, err = m.Data.DBDriver.Exec(sqlQuery, scriptName)
+	if err != nil {
+		return errors.New("failed to exec query: " + err.Error())
+	}
+	return nil
+}
+
+func (m *MigrationHandler) runTx(sqlQuery string) error {
+	if sqlQuery == "" {
+		// skip empty script
+		return nil
+	}
+
 	// begin tx
 	tx, err := m.Data.DBDriver.Begin()
 	if err != nil {
 		return errors.New("failed to begin tx: " + err.Error())
 	}
 
 	// exec query
-	_, err = tx.Exec(string(fileBytes))
+	_, err = tx.Exec(sqlQuery)
 	if err != nil {
 		return errors.New("failed to exec script: " + err.Error())
 	}
@@ -114,13 +139,6 @@ func (m *MigrationHandler) runScript(scriptName string) error {
 		tx.Rollback()
 		return errors.New("failed to commit tx: " + err.Error())
 	}
-
-	// update version
-	sqlQuery := "INSERT INTO " + versionsTableName + " SET name=?"
-	_, err = m.Data.DBDriver.Exec(sqlQuery, scriptName)
-	if err != nil {
-		return errors.New("failed to exec query: " + err.Error())
-	}
 	return nil
 }