status-im
/
migrate
mirror of https://github.com/status-im/migrate.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Use counter for migration version information instead of rows

This commit is contained in:
Paul Bergeron 2014-08-26 19:43:05 -07:00
parent ca8cd2c56b
commit c7e6d54365
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
driver/cassandra/cassandra.go
View File

@ -15,6 +15,7 @@ type Driver struct {
}
const tableName = "schema_migrations"
const versionRow = 1
// Cassandra Driver URL format:
// cassandra://host:port/keyspace
@ -47,7 +48,7 @@ func (driver *Driver) Close() error {
}
func (driver *Driver) ensureVersionTableExists() error {
err := driver.session.Query("CREATE TABLE IF NOT EXISTS " + tableName + " (version bigint primary key);").Exec()
err := driver.session.Query("CREATE TABLE IF NOT EXISTS " + tableName + " (version counter, versionRow bigint primary key);").Exec()
if err != nil {
return err
}
@ -63,13 +64,13 @@ func (driver *Driver) Migrate(f file.File, pipe chan interface{}) {
pipe <- f
if f.Direction == direction.Up {
err := driver.session.Query("INSERT INTO "+tableName+" (version) VALUES (?)", f.Version).Exec()
err := driver.session.Query("UPDATE "+tableName+" SET version = version + 1 where versionRow = ?", versionRow).Exec()
if err != nil {
pipe <- err
return
}
} else if f.Direction == direction.Down {
err := driver.session.Query("DELETE FROM "+tableName+" WHERE version = ?", f.Version).Exec()
err := driver.session.Query("UPDATE "+tableName+" SET version = version - 1 where versionRow = ?", versionRow).Exec()
if err != nil {
pipe <- err
return
@ -91,6 +92,6 @@ func (driver *Driver) Migrate(f file.File, pipe chan interface{}) {
func (driver *Driver) Version() (uint64, error) {
var version int64
err := driver.session.Query("SELECT version FROM " + tableName + " ORDER BY version DESC LIMIT 1").Scan(&version)
err := driver.session.Query("SELECT version FROM "+tableName+" WHERE versionRow = ?", versionRow).Scan(&version)
return uint64(version), err
}