run migrations in a transaction

Author: Miguel Molina

generator/generator_test.go

@@ -70,11 +70,11 @@ func TestMigrationGeneratorGenerate(t *testing.T) {
 
 	content, err := ioutil.ReadFile(g.migrationFile(migrationUp, g.now()))
 	require.NoError(t, err)
-	require.Equal(t, expectedTable2+"\n\n", string(content))
+	require.Equal(t, "BEGIN;\n\n"+expectedTable2+"\n\nCOMMIT;\n", string(content))
 
 	content, err = ioutil.ReadFile(g.migrationFile(migrationDown, g.now()))
 	require.NoError(t, err)
-	require.Equal(t, "DROP TABLE table2;\n\n", string(content))
+	require.Equal(t, "BEGIN;\n\nDROP TABLE table2;\n\nCOMMIT;\n", string(content))
 
 	expected, err := migration.Lock.MarshalText()
 	require.NoError(t, err)

generator/migration.go

@@ -323,6 +323,7 @@ func (cs ChangeSet) sorted(dropIndex, createIndex map[string]*TableSchema) (Chan
 
 func (cs ChangeSet) MarshalText() ([]byte, error) {
 	var buf bytes.Buffer
+	buf.WriteString("BEGIN;\n\n")
 	for _, c := range cs {
 		bytes, err := c.MarshalText()
 		if err != nil {
@@ -331,6 +332,7 @@ func (cs ChangeSet) MarshalText() ([]byte, error) {
 		buf.Write(bytes)
 		buf.WriteRune('\n')
 	}
+	buf.WriteString("COMMIT;\n")
 	return buf.Bytes(), nil
 }
 

generator/migration_test.go

@@ -62,7 +62,7 @@ func TestChangeSet(t *testing.T) {
 			&DropTable{"foo"},
 			&DropColumn{"col", "table"},
 		},
-		"DROP TABLE foo;\n\nALTER TABLE table DROP COLUMN col;\n\n",
+		"BEGIN;\n\nDROP TABLE foo;\n\nALTER TABLE table DROP COLUMN col;\n\nCOMMIT;\n",
 	)
 }