Multiple statements per migration file

Makes it possible to run several queries in each file.
balboah · Nov 17, 2014 · 5f1ca5d · mattes · Nov 17, 2014 · balboah
1 parent d5f5eb6
commit 5f1ca5d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/driver/cassandra/cassandra.go b/driver/cassandra/cassandra.go
@@ -3,6 +3,7 @@ package cassandra
 
 import (
 	"net/url"
+	"strings"
 	"time"
 
 	"github.com/gocql/gocql"
@@ -120,7 +121,16 @@ func (driver *Driver) Migrate(f file.File, pipe chan interface{}) {
 		return
 	}
 
-	err = driver.session.Query(string(f.Content)).Exec()
+	for _, query := range strings.Split(string(f.Content), ";") {
+		query = strings.TrimSpace(query)
+		if len(query) == 0 {
+			continue
+		}
+
+		if err = driver.session.Query(query).Exec(); err != nil {
+			return
+		}
+	}
 }
 
 func (driver *Driver) Version() (uint64, error) {

diff --git a/driver/cassandra/cassandra_test.go b/driver/cassandra/cassandra_test.go
@@ -53,8 +53,11 @@ func TestMigrate(t *testing.T) {
 			Direction: direction.Up,
 			Content: []byte(`
                 CREATE TABLE yolo (
-                    id varint primary key
+                    id varint primary key,
+                    msg text
                 );
+
+				CREATE INDEX ON yolo (msg);
             `),
 		},
 		{